def setUp(self):
self.work_state_1 = WorkerState(TEST_WORKER_ID_1, 10)
self.work_state_2 = WorkerState(TEST_WORKER_ID_2)
argparser = ParlaiParser(False, False)
argparser.add_parlai_data_path()
argparser.add_mturk_args()
self.opt = argparser.parse_args([])
self.opt['task'] = 'unittest'
self.opt['assignment_duration_in_seconds'] = 6
mturk_agent_ids = ['mturk_agent_1']
self.mturk_manager = MTurkManager(opt=self.opt, mturk_agent_ids=mturk_agent_ids)
self.worker_manager = WorkerManager(self.mturk_manager, self.opt)
class TestWorkerState(unittest.TestCase):
"""
Various unit tests for the WorkerState class.
"""
def setUp(self):
self.work_state_1 = WorkerState(TEST_WORKER_ID_1, 10)
self.work_state_2 = WorkerState(TEST_WORKER_ID_2)
argparser = ParlaiParser(False, False)
argparser.add_parlai_data_path()
argparser.add_mturk_args()
self.opt = argparser.parse_args(print_args=False)
self.opt['task'] = 'unittest'
self.opt['assignment_duration_in_seconds'] = 6
mturk_agent_ids = ['mturk_agent_1']
self.mturk_manager = MTurkManager(opt=self.opt,
mturk_agent_ids=mturk_agent_ids)
self.worker_manager = WorkerManager(self.mturk_manager, self.opt)
def tearDown(self):
self.mturk_manager.shutdown()
def test_worker_state_init(self):
"""
Test proper initialization of worker states.
"""
self.assertEqual(self.work_state_1.worker_id, TEST_WORKER_ID_1)
self.assertEqual(self.work_state_2.worker_id, TEST_WORKER_ID_2)
self.assertEqual(self.work_state_1.disconnects, 10)
self.assertEqual(self.work_state_2.disconnects, 0)
def test_worker_state_agent_management(self):
"""
Test public state management methods of worker_state.
"""
agent_1 = MTurkAgent(
self.opt,
self.mturk_manager,
TEST_HIT_ID_1,
TEST_ASSIGNMENT_ID_1,
TEST_WORKER_ID_1,
)
agent_2 = MTurkAgent(
self.opt,
self.mturk_manager,
TEST_HIT_ID_2,
TEST_ASSIGNMENT_ID_2,
TEST_WORKER_ID_1,
)
agent_3 = MTurkAgent(
self.opt,
self.mturk_manager,
TEST_HIT_ID_3,
TEST_ASSIGNMENT_ID_3,
TEST_WORKER_ID_3,
)
self.assertEqual(self.work_state_1.active_conversation_count(), 0)
self.work_state_1.add_agent(agent_1)
self.assertEqual(self.work_state_1.active_conversation_count(), 1)
self.work_state_1.add_agent(agent_2)
self.assertEqual(self.work_state_1.active_conversation_count(), 2)
with self.assertRaises(AssertionError):
self.work_state_1.add_agent(agent_3)
self.assertEqual(self.work_state_1.active_conversation_count(), 2)
self.assertEqual(self.work_state_1.completed_assignments(), 0)
self.assertTrue(self.work_state_1.has_assignment(
agent_1.assignment_id))
self.assertTrue(self.work_state_1.has_assignment(
agent_2.assignment_id))
self.assertFalse(
self.work_state_1.has_assignment(agent_3.assignment_id))
self.assertEqual(
agent_1,
self.work_state_1.get_agent_for_assignment(agent_1.assignment_id))
self.assertEqual(
agent_2,
self.work_state_1.get_agent_for_assignment(agent_2.assignment_id))
self.assertIsNone(
self.work_state_1.get_agent_for_assignment(agent_3.assignment_id))
agent_1.set_status(AssignState.STATUS_DONE)
self.assertEqual(self.work_state_1.active_conversation_count(), 1)
self.assertEqual(self.work_state_1.completed_assignments(), 1)
agent_2.set_status(AssignState.STATUS_DISCONNECT)
self.assertEqual(self.work_state_1.active_conversation_count(), 0)
self.assertEqual(self.work_state_1.completed_assignments(), 1)
def test_manager_alive_makes_state(self):
test_worker = self.worker_manager.worker_alive(TEST_WORKER_ID_1)
self.assertIsInstance(test_worker, WorkerState)
self.assertEqual(test_worker.worker_id, TEST_WORKER_ID_1)
self.assertNotEqual(test_worker, self.work_state_1)