Basic and scruffy Subrip read support.
[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::ifstream;
34 using std::getline;
35
36 boost::filesystem::path private_test;
37
38 struct TestConfig
39 {
40         TestConfig()
41         {
42                 if (boost::unit_test::framework::master_test_suite().argc >= 2) {
43                         private_test = boost::unit_test::framework::master_test_suite().argv[1];
44                 } else {
45                         BOOST_TEST_MESSAGE ("Private data libsub-test-private not found; some tests will not run");
46                 }
47
48                 sub::make_iso6937_tables ();
49         }
50 };
51
52 BOOST_GLOBAL_FIXTURE (TestConfig);
53
54 void
55 check_text (boost::filesystem::path a, boost::filesystem::path b)
56 {
57         if (!boost::filesystem::exists (a)) {
58                 cerr << "File not found: " << a << "\n";
59         }
60
61         if (!boost::filesystem::exists (b)) {
62                 cerr << "File not found: " << b << "\n";
63         }
64         
65         BOOST_CHECK (boost::filesystem::exists (a));
66         BOOST_CHECK (boost::filesystem::exists (b));
67         
68         ifstream p (a.c_str ());
69         ifstream q (b.c_str ());
70
71         string x;
72         string y;
73         while (p.good() && q.good()) {
74                 getline (p, x);
75                 getline (q, y);
76                 BOOST_CHECK_EQUAL (x, y);
77         }
78
79         BOOST_CHECK (p.good() == false);
80         BOOST_CHECK (q.good() == false);
81 }
82
83 void
84 check_file (boost::filesystem::path ref, boost::filesystem::path check)
85 {
86         uintmax_t N = boost::filesystem::file_size (ref);
87         BOOST_CHECK_EQUAL (N, boost::filesystem::file_size (check));
88         FILE* ref_file = fopen (ref.string().c_str(), "rb");
89         BOOST_CHECK (ref_file);
90         FILE* check_file = fopen (check.string().c_str(), "rb");
91         BOOST_CHECK (check_file);
92         
93         int const buffer_size = 65536;
94         uint8_t* ref_buffer = new uint8_t[buffer_size];
95         uint8_t* check_buffer = new uint8_t[buffer_size];
96
97         while (N) {
98                 uintmax_t this_time = min (uintmax_t (buffer_size), N);
99                 size_t r = fread (ref_buffer, 1, this_time, ref_file);
100                 BOOST_CHECK_EQUAL (r, this_time);
101                 r = fread (check_buffer, 1, this_time, check_file);
102                 BOOST_CHECK_EQUAL (r, this_time);
103
104                 BOOST_CHECK_EQUAL (memcmp (ref_buffer, check_buffer, this_time), 0);
105                 N -= this_time;
106         }
107
108         delete[] ref_buffer;
109         delete[] check_buffer;
110
111         fclose (ref_file);
112         fclose (check_file);
113 }