def test_handle_model_generation(self, mock_set):
mock_set.return_value = ('Foo1', True)
model, auth_model = generators.handle_model_generation(
'foo', '/stories')
mock_set.assert_called_once_with('foo', 'Story')
assert model == 'Foo1'
assert auth_model
def test_handle_model_generation_value_err(self, mock_set):
mock_set.side_effect = ValueError('strange error')
with pytest.raises(ValueError) as ex:
generators.handle_model_generation('foo', '/stories')
assert str(ex.value) == 'Story: strange error'
mock_set.assert_called_once_with('foo', 'Story')