wip: migrating hub

Author: justinmerrell

tests/modules/test_lan.py

@@ -14,6 +14,9 @@ class TestLan(unittest.TestCase):
     ''' Tests for the lan module '''
 
     def test_monitor_network(self):
+        '''
+        Confirm that the network monitor is running.
+        '''
         with patch('modules.rec_lan.test_network') as mocked_test_network:
             mocked_test_network.return_value = 0
 

tests/test_updater.py

@@ -16,21 +16,23 @@ class TestUpdater(unittest.TestCase):
     '''Collection of tests.'''
 
     def test_get_current_versions(self):
-        '''Verify that the current versions are returned.'''
+        '''
+        Verify that the current versions are returned.
+        '''
         system_json = StringIO("""{
-                        "serial": "536780dfe639468e8e23fc568006950d",
-                        "timezone": "America/New_York",
-                        "CurrentVersion": "0_0_0",
-                        "HUBid": 40,
-                        "Token": "5a12ff36eed2f0647a48af62e635eb8cfd4c5979",
-                        "facility": "3b9fdc97-9649-4c80-8b48-10df647bd032"
-                    }""")
+                            "serial": "536780dfe639468e8e23fc568006950d",
+                            "timezone": "America/New_York",
+                            "CurrentVersion": "0_0_0",
+                            "HUBid": 40,
+                            "Token": "5a12ff36eed2f0647a48af62e635eb8cfd4c5979",
+                            "facility": "3b9fdc97-9649-4c80-8b48-10df647bd032"
+                        }""")
 
         with patch('updater.open') as mock_open:
             mock_open.side_effect = [system_json, system_json]
-            updater.current_hub_version()
-            # hub_updater.update_version_name()
+            self.assertEqual(updater.current_hub_version(), "0_0_0")
             mock_open.assert_called()
 
+
 if __name__ == '__main__':
     unittest.main()