class TestLoadBalancer(TestCase):
def setUp(self) -> None:
self.load_balancer = LoadBalancer(4, 2)
def test_create_a_server_should_return_a_server_instance(self):
server = self.load_balancer.create_a_server()
self.assertIsInstance(server, Server)
def test_calculate_user_should_return_one_when_exist_only_one_server_with_one_running_task(self):
self.load_balancer.assign_task_to_server(1)
expected = 1
result = self.load_balancer.calculate_user()
self.assertEqual(expected, result)
def test_run_tasks_should_reduce_ttask_user_attribute_by_one(self):
self.load_balancer.assign_task_to_server(1)
self.load_balancer.run_tasks()
expected = 3
result = self.load_balancer.servers[0].users[0].ttask
self.assertEqual(expected, result)
def test_output_should_return_two_comma_two_when_exist_four_active_users(self):
self.load_balancer.assign_task_to_server(4)
expected = "2, 2"
result = self.load_balancer.output()
self.assertEqual(expected, result)
def test_assign_task_to_server_should_create_a_server_for_a_task_when_there_is_no_server_running(self):
self.load_balancer.assign_task_to_server(1)
expected = 1
result = len(self.load_balancer.servers)
self.assertEqual(expected, result)
def test_assign_task_to_server_should_not_create_a_server_for_a_task_when_there_is_an_available_server_running(self):
self.load_balancer.assign_task_to_server(1)
self.load_balancer.assign_task_to_server(1)
expected = 1
result = len(self.load_balancer.servers)
self.assertEqual(expected, result)
def test_assign_task_to_server_should_create_a_new_instance_when_there_is_a_server_running_but_not_available(self):
self.load_balancer.assign_task_to_server(2)
self.load_balancer.assign_task_to_server(1)
expected = 2
result = len(self.load_balancer.servers)
self.assertEqual(expected, result)
tasks = [int(i.rstrip("\n")) for i in filereader]
TTASK = int(tasks[0])
if TTASK < 1 or TTASK > 10:
print(ttask_invalid_mesage)
sys.exit()
UMAX = int(tasks[1])
if UMAX < 1 or UMAX > 10:
print(umax_invalid_mesage)
sys.exit()
load_balancer = LoadBalancer(TTASK, UMAX)
tasks = tasks[2:]
with open("output.txt", "w") as output_file:
for i in tasks:
load_balancer.run_tasks()
load_balancer.assign_task_to_server(i)
load_balancer.calculate_user()
output_file.write(load_balancer.output())
output_file.write("\n")
while len(load_balancer.servers):
load_balancer.run_tasks()
load_balancer.calculate_user()
output_file.write(load_balancer.output())
output_file.write("\n")
output_file.write(str(load_balancer.total))
