General structure of operation and MBOX writing.

So far, only unit test for the latter.

1 files changed, 16 insertions, 0 deletions

diff --git a/test/test_unit.py b/test/test_unit.py
index 4674529..e79c635 100644
--- a/test/test_unit.py
+++ b/test/test_unit.py
@@ -1,3 +1,4 @@
+import pickle
 import unittest
 import gg_scrapper
 
@@ -16,5 +17,20 @@ class TestGGScrapper(unittest.TestCase):
         obs_URL = gg_scrapper.Group.do_redirect(ORIG_URL)
         self.assertEqual(obs_URL, EXP_URL)
 
+
+class TestMBOX(unittest.TestCase):
+    def test_create_mbox(self):
+        '''Create a mbox file from (pickled) Group
+        '''
+        group_file_name = 'test/group.pickle'
+        with open(group_file_name, 'r', encoding='utf8') as group_f:
+            group = pickle.load(group_f)
+
+        mbx = gg_scrapper.MBOX()
+        mbx.format_mbox(group)
+
+        with open('test/generated_mbox.mbx') as exp_f:
+            self.assertEqual(exp_f.read(), mbx.mbox_string)
+
 if __name__ == '__main__':
     unittest.main()