class doServerManager_handle_serverTests(unittest.TestCase): def setUp(self): self.manager = ServerManager() def test_handle_server_somedir(self): result = self.manager.handle_server('somedir') self.assertEqual(result, ServerStatus.ONLINE) def test_handle_server_otherdir(self): result = self.manager.handle_server('otherdir') self.assertEqual(result, ServerStatus.OFFLINE)