[PATCH 2/2] Test DM4 import

[Miika Turkia]

Signed-off-by: Miika Turkia <miika.turkia at gmail.com>
---
 tests/testparse.cpp | 27 +++++++++++++++++++++++++++
 tests/testparse.h   |  2 ++
 2 files changed, 29 insertions(+)

diff --git a/tests/testparse.cpp b/tests/testparse.cpp
index 2bedc88..b8d1e98 100644
--- a/tests/testparse.cpp
+++ b/tests/testparse.cpp
@@ -1,6 +1,7 @@
 #include "testparse.h"
 #include "dive.h"
 #include "file.h"
+#include "divelist.h"
 #include <QTextStream>
 
 void TestParse::testParseCSV()
@@ -57,6 +58,32 @@ void TestParse::testParseCompareOutput()
 	QString readin = orgS.readAll();
 	QString written = outS.readAll();
 	QCOMPARE(readin, written);
+	clear_dive_file_data();
+}
+
+void TestParse::testParseDM4()
+{
+	sqlite3 *handle;
+
+	QCOMPARE(sqlite3_open(SUBSURFACE_SOURCE "/dives/TestDiveDM4.db", &handle), 0);
+	QCOMPARE(parse_dm4_buffer(handle, 0, 0, 0, &dive_table), 0);
+
+	sqlite3_close(handle);
+}
+
+void TestParse::testParseCompareDM4Output()
+{
+	QCOMPARE(save_dives("./testsd4out.ssrf"), 0);
+	QFile org(SUBSURFACE_SOURCE "/dives/TestDiveDM4.xml");
+	org.open(QFile::ReadOnly);
+	QFile out("./testdm4out.ssrf");
+	out.open(QFile::ReadOnly);
+	QTextStream orgS(&org);
+	QTextStream outS(&out);
+	QString readin = orgS.readAll();
+	QString written = outS.readAll();
+	QCOMPARE(readin, written);
+	clear_dive_file_data();
 }
 
 QTEST_MAIN(TestParse)
diff --git a/tests/testparse.h b/tests/testparse.h
index 5e1dc03..23a7c49 100644
--- a/tests/testparse.h
+++ b/tests/testparse.h
@@ -11,6 +11,8 @@ private slots:
 	void testParseV2NoQuestion();
 	void testParseV3();
 	void testParseCompareOutput();
+	void testParseDM4();
+	void testParseCompareDM4Output();
 };
 
 #endif
-- 
2.1.4