def test_handle_calls_handle_label(self): namespace = argparse.Namespace(labels=['foo', 'bar']) command = LabelCommand() command.handle_label = mock.Mock() command.handle(namespace) self.assertEqual(command.handle_label.call_args_list, [ arg('foo', namespace), arg('bar', namespace), ])
def test_handle_raise_if_handle_label_not_implemented(self): command = LabelCommand() with self.assertRaises(NotImplementedError): command.handle(argparse.Namespace(labels=['foo']))