def test_run_wrong_ace_format(self, mock_eng, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace="asdasdasdasd", models="User,Story") with pytest.raises(ValueError) as ex: obj.run() assert "--ace" in str(ex.value) mock_eng.get_document_cls.assert_has_calls([call("User"), call("Story")])
def test_run_wrong_ace_format(self, mock_eng, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace='asdasdasdasd', models='User,Story') with pytest.raises(ValueError) as ex: obj.run() assert '--ace' in str(ex.value) mock_eng.get_document_cls.assert_has_calls( [call('User'), call('Story')])
def test_run(self, mock_six, mock_count, mock_eng, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace="{}", models="User") model = Mock(__name__="Foo") mock_eng.get_document_cls.return_value = model mock_count.return_value = {model: 123} obj.run() mock_six.print_.assert_has_calls([call("Model,Count"), call("Foo,123")]) mock_count.assert_called_once_with(ace={}, models=[model])
def test_run_none_count(self, mock_six, mock_count, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace='{}', models='') mock_count.return_value = {Mock(__name__='Foo'): None} obj.run() mock_six.print_.assert_has_calls([ call('Model,Count'), call('Foo,Not es-based'), ])
def test_run(self, mock_six, mock_count, mock_eng, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace='{}', models='User') model = Mock(__name__='Foo') mock_eng.get_document_cls.return_value = model mock_count.return_value = {model: 123} obj.run() mock_six.print_.assert_has_calls([ call('Model,Count'), call('Foo,123'), ]) mock_count.assert_called_once_with(ace={}, models=[model])
def test_run_none_count(self, mock_six, mock_count, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace="{}", models="") mock_count.return_value = {Mock(__name__="Foo"): None} obj.run() mock_six.print_.assert_has_calls([call("Model,Count"), call("Foo,Not es-based")])
def test_run_no_models(self, mock_count, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace="{}", models=None) mock_count.return_value = {Mock(__name__="Foo"): 1} obj.run() mock_count.assert_called_once_with(ace={}, models=None)
def test_run_no_models(self, mock_count, mock_boot, mock_parse): obj = CountACECommand() obj.options = Mock(ace='{}', models=None) mock_count.return_value = {Mock(__name__='Foo'): 1} obj.run() mock_count.assert_called_once_with(ace={}, models=None)