def test_manage(self):
manager = TaskManager()
self._create_employees(10)
self._create_tasks(100*10)
manager.manage()
self._test_load_diff()
def test_create_task(self):
data = {'name': 'anytank', 'complexity': 2}
manager = TaskManager()
manager.create_task(**data)
self.assertEqual(len(manager.tasks), 1)
self.assertEqual(manager.tasks[0].name, data['name'])
self.assertEqual(manager.tasks[0].complexity, data['complexity'])
def test_create_employee(self):
data = {'name': 'anyemployee', 'cp_per_time': 2}
manager = TaskManager()
manager.create_empoyee(**data)
self.assertEqual(len(manager.employees), 1)
self.assertEqual(manager.employees[0].name, data['name'])
self.assertEqual(manager.employees[0].cp_per_time, data['cp_per_time'])
def _create_employees(self, count):
manager = TaskManager()
for i in range(count):
manager.create_empoyee('employee_%d' % i,
randint(1, self.max_complexity))
def _create_tasks(self, count):
manager = TaskManager()
for i in range(count):
manager.create_task('task_%d' % i,
randint(1, self.max_complexity))
