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())
