add test_decode_readme, add test_generate_new_readme (#50)

* add test_generate_new_readme

* add test_decode_readme

tests/test_main.py

@@ -2,12 +2,14 @@
 Tests for the main.py
 '''
 import unittest
+import datetime
+import base64
 import os
 
 try:
     # For travis build which uses
     # python -m unittest discover
-    from main import make_graph
+    from main import make_graph, generate_new_readme, decode_readme
 except Exception as e:
     print("Error: missing 'main.py'\nTrying ablsolute import...")
 
@@ -62,6 +64,31 @@ class TestMain(unittest.TestCase):
             for j, graph in enumerate(graphs):
                 test(percents[j], blocks[i], graph)
 
+    def test_generate_new_readme(self):
+        '''Tests generate_new_readme method from main.py'''
+        dummy_readme = '''My Readme Start
+        <!--START_SECTION:waka-->
+        <!--END_SECTION:waka-->
+        My Readme End'''
+        dummy_stats = '''```text
+        Python     24 hrs 15 mins  █████████████████████████   100.00 %
+        ```'''
+        expected_generated_readme = '''My Readme Start
+        <!--START_SECTION:waka-->\n```text
+        Python     24 hrs 15 mins  █████████████████████████   100.00 %
+        ```\n<!--END_SECTION:waka-->
+        My Readme End
+        '''
+        expected_generated_readme = expected_generated_readme.strip()
+        actual_generated_readme = generate_new_readme(dummy_stats, dummy_readme)
+        self.assertEqual(actual_generated_readme, expected_generated_readme)
+
+    def test_decode_readme(self):
+        '''Tests decode_readme method from main.py'''
+        dummy_data = base64.b64encode(bytes('Some Data From GitHub', 'utf-8'))
+        expected_result = 'Some Data From GitHub'
+        actual_result = decode_readme(dummy_data)
+        self.assertEqual(actual_result, expected_result)
 
 if __name__ == '__main__':
     if __package__ is None:
@@ -70,9 +97,9 @@ if __name__ == '__main__':
         # python test/test_main.py
         sys.path.append(os.path.dirname(
         os.path.dirname(os.path.abspath(__file__))))
-        from main import make_graph
+        from main import make_graph, generate_new_readme, decode_readme
     else:
         # Later on if WakaReadme is implemetaion as package
         # python -m tests/test_main
-        from ..main import make_graph
+        from ..main import make_graph, generate_new_readme, decode_readme
     unittest.main()