Template-ize collect so that any container can be used.
[libsub.git] / test / test.cc
1 /*
2     Copyright (C) 2014 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #define BOOST_TEST_DYN_LINK
21 #define BOOST_TEST_MODULE libsub_test
22 #include <boost/test/unit_test.hpp>
23 #include <boost/filesystem.hpp>
24 #include <fstream>
25 #include <string>
26 #include "iso6937_tables.h"
27
28 using std::string;
29 using std::cerr;
30 using std::cout;
31 using std::min;
32 using std::max;
33 using std::hex;
34 using std::ifstream;
35 using std::getline;
36 using std::stringstream;
37
38 boost::filesystem::path private_test;
39
40 struct TestConfig
41 {
42         TestConfig()
43         {
44                 if (boost::unit_test::framework::master_test_suite().argc >= 2) {
45                         private_test = boost::unit_test::framework::master_test_suite().argv[1];
46                 } else {
47                         BOOST_TEST_MESSAGE ("Private data libsub-test-private not found; some tests will not run");
48                 }
49
50                 sub::make_iso6937_tables ();
51         }
52 };
53
54 BOOST_GLOBAL_FIXTURE (TestConfig);
55
56 void
57 check_text (boost::filesystem::path a, boost::filesystem::path b)
58 {
59         if (!boost::filesystem::exists (a)) {
60                 cerr << "File not found: " << a << "\n";
61         }
62
63         if (!boost::filesystem::exists (b)) {
64                 cerr << "File not found: " << b << "\n";
65         }
66         
67         BOOST_CHECK (boost::filesystem::exists (a));
68         BOOST_CHECK (boost::filesystem::exists (b));
69         
70         ifstream p (a.c_str ());
71         ifstream q (b.c_str ());
72
73         string x;
74         string y;
75         while (p.good() && q.good()) {
76                 getline (p, x);
77                 getline (q, y);
78                 BOOST_CHECK_EQUAL (x, y);
79         }
80
81         BOOST_CHECK (p.good() == false);
82         BOOST_CHECK (q.good() == false);
83 }
84
85 void
86 check_file (boost::filesystem::path ref, boost::filesystem::path check)
87 {
88         uintmax_t N = boost::filesystem::file_size (ref);
89         BOOST_CHECK_EQUAL (N, boost::filesystem::file_size (check));
90         FILE* ref_file = fopen (ref.string().c_str(), "rb");
91         BOOST_CHECK (ref_file);
92         FILE* check_file = fopen (check.string().c_str(), "rb");
93         BOOST_CHECK (check_file);
94         
95         int const buffer_size = 65536;
96         uint8_t* ref_buffer = new uint8_t[buffer_size];
97         uint8_t* check_buffer = new uint8_t[buffer_size];
98
99         uintmax_t offset = 0;
100         while (offset < N) {
101                 uintmax_t this_time = min (uintmax_t (buffer_size), N - offset);
102                 size_t r = fread (ref_buffer, 1, this_time, ref_file);
103                 BOOST_CHECK_EQUAL (r, this_time);
104                 r = fread (check_buffer, 1, this_time, check_file);
105                 BOOST_CHECK_EQUAL (r, this_time);
106
107                 for (uintmax_t i = 0; i < this_time; ++i) {
108                         stringstream s;
109                         s << "Files differ at offset " << (offset + i) << "; reference is " << hex << ((int) ref_buffer[i]) << ", check is " << ((int) check_buffer[i]);
110                         BOOST_CHECK_MESSAGE (ref_buffer[i] == check_buffer[i], s.str ());
111                 }
112
113                 offset += this_time;
114         }
115
116         delete[] ref_buffer;
117         delete[] check_buffer;
118
119         fclose (ref_file);
120         fclose (check_file);
121 }