class TestServerClock(unittest.TestCase): def setUp(self): self.TTASK = 4 self.UMAX = 2 self.server = Server(self.TTASK, self.UMAX) def test_server_clock(self): """Checks if server clock influences tasks.""" self.server.add_task() self.server.clock() self.assertEqual(self.server.task_list[0].missing_ttask, 3) def test_server_clock_remove_tasks(self): self.server.add_task() self.server.clock() self.server.add_task() self.server.clock() self.assertEqual(self.server.task_list[0].missing_ttask, 2) self.assertEqual(self.server.task_list[1].missing_ttask, 3) self.server.clock() self.assertEqual(self.server.task_list[0].missing_ttask, 1) self.assertEqual(self.server.task_list[1].missing_ttask, 2) self.server.clock() self.assertEqual(self.server.task_list[0].missing_ttask, 1) self.server.add_task() self.server.clock() self.assertEqual(self.server.task_list[0].missing_ttask, 3)
class TestServer(unittest.TestCase): def setUp(self): self.TTASK = 5 self.UMAX = 8 self.server = Server(self.TTASK, self.UMAX) def test_server_attributes(self): self.assertEqual(self.server.ttask, self.TTASK) self.assertEqual(self.server.umax, self.UMAX) def test_server_add_task(self): self.assertEqual(0, len(self.server.task_list)) self.server.add_task() self.assertEqual(1, len(self.server.task_list)) def test_server_is_working(self): self.assertFalse(self.server.is_working()) self.server.add_task() self.assertTrue(self.server.is_working()) def test_server_task_umax_limit(self): """Tests the server UMAX limits.""" for i in range(self.UMAX): self.server.add_task() self.assertRaises(Exception, self.server.add_task) def test_server_task_ttask_limit(self): """Tests the server TTASK limits.""" with self.assertRaises(AssertionError): Server(11, 10) def test_server_task_finish(self): """Tests if server is finishing tasks.""" self.server.add_task() for i in range(self.TTASK): self.server.clock() self.assertFalse(self.server.is_working()) def test_server_have_slots(self): self.assertEqual(self.UMAX, self.server.available_slots()) for i in range(self.UMAX): self.server.add_task() self.assertEqual(0, self.server.available_slots())