def test_add_arguments(self):
c = crd.Command()
parser = MagicMock()
c.add_arguments(parser)
parser.add_argument.assert_called_once_with(
'--number',
dest='number',
help="how many would you like to create",
default=100
)
def test_handle_default_100(self):
com = crd.Command()
with patch.object(crd.PatientGenerator, 'make') as maker:
com.handle(number=None)
self.assertEqual(100, maker.call_count)
def test_handle(self):
com = crd.Command()
with patch.object(crd.PatientGenerator, 'make') as maker:
com.handle(number='23')
self.assertEqual(23, maker.call_count)