def test_task_header_agent_bind(self): # Setup sim = task_simulator.TaskSimulator() task = sim.add_task_state('2', responses.STATE_FINISHED) task.tags = [tags.action_tag(tags.ACTION_AGENT_BIND)] # Test self.command.task_header(task) # Verify self.assertEqual(self.prompt.get_write_tags(), ['agent-bind-header'])
def test_task_header_delete_binding(self): # Setup sim = task_simulator.TaskSimulator() task = sim.add_task_state('3', responses.STATE_FINISHED) task.tags = [tags.action_tag(tags.ACTION_DELETE_BINDING)] # Test self.command.task_header(task) # Verify self.assertEqual(self.prompt.get_write_tags(), ['delete-header'])