From 02099a01350a41a99ec400e9b29df08a01d88979 Mon Sep 17 00:00:00 2001
From: Paul Baltescu <pauldb89@gmail.com>
Date: Tue, 4 Jun 2013 23:17:57 +0100
Subject: Serialize data structures.

---
 extractor/alignment_test.cc | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

(limited to 'extractor/alignment_test.cc')

diff --git a/extractor/alignment_test.cc b/extractor/alignment_test.cc
index 43c37ebd..1b8ff531 100644
--- a/extractor/alignment_test.cc
+++ b/extractor/alignment_test.cc
@@ -1,12 +1,16 @@
 #include <gtest/gtest.h>
 
-#include <memory>
+#include <sstream>
 #include <string>
 
+#include <boost/archive/binary_iarchive.hpp>
+#include <boost/archive/binary_oarchive.hpp>
+
 #include "alignment.h"
 
 using namespace std;
 using namespace ::testing;
+namespace ar = boost::archive;
 
 namespace extractor {
 namespace {
@@ -14,19 +18,31 @@ namespace {
 class AlignmentTest : public Test {
  protected:
   virtual void SetUp() {
-    alignment = make_shared<Alignment>("sample_alignment.txt");
+    alignment = Alignment("sample_alignment.txt");
   }
 
-  shared_ptr<Alignment> alignment;
+  Alignment alignment;
 };
 
 TEST_F(AlignmentTest, TestGetLinks) {
   vector<pair<int, int>> expected_links = {
     make_pair(0, 0), make_pair(1, 1), make_pair(2, 2)
   };
-  EXPECT_EQ(expected_links, alignment->GetLinks(0));
+  EXPECT_EQ(expected_links, alignment.GetLinks(0));
   expected_links = {make_pair(1, 0), make_pair(2, 1)};
-  EXPECT_EQ(expected_links, alignment->GetLinks(1));
+  EXPECT_EQ(expected_links, alignment.GetLinks(1));
+}
+
+TEST_F(AlignmentTest, TestSerialization) {
+  stringstream stream(ios_base::binary | ios_base::out | ios_base::in);
+  ar::binary_oarchive output_stream(stream, ar::no_header);
+  output_stream << alignment;
+
+  Alignment alignment_copy;
+  ar::binary_iarchive input_stream(stream, ar::no_header);
+  input_stream >> alignment_copy;
+
+  EXPECT_EQ(alignment, alignment_copy);
 }
 
 } // namespace
-- 
cgit v1.2.3