コード例 #1
0
 def test_run_wrong_to_ace_format(self, mock_eng, mock_boot, mock_parse):
     obj = UpdateACECommand()
     obj.options = Mock(from_ace="{}", to_ace="asdasdasdasd", models="User,Story")
     with pytest.raises(ValueError) as ex:
         obj.run()
     assert "--to_ace" in str(ex.value)
     mock_eng.get_document_cls.assert_has_calls([call("User"), call("Story")])
コード例 #2
0
 def test_run(self, mock_count, mock_eng, mock_boot, mock_parse):
     obj = UpdateACECommand()
     obj.options = Mock(from_ace='{"a": 1}', to_ace='{"b": 2}', models="User")
     model = Mock(__name__="Foo")
     mock_eng.get_document_cls.return_value = model
     mock_count.return_value = {model: 123}
     obj.run()
     mock_count.assert_called_once_with(from_ace={"a": 1}, to_ace={"b": 2}, models=[model])
コード例 #3
0
 def test_run_wrong_to_ace_format(self, mock_eng, mock_boot, mock_parse):
     obj = UpdateACECommand()
     obj.options = Mock(from_ace='{}',
                        to_ace='asdasdasdasd',
                        models='User,Story')
     with pytest.raises(ValueError) as ex:
         obj.run()
     assert '--to_ace' in str(ex.value)
     mock_eng.get_document_cls.assert_has_calls(
         [call('User'), call('Story')])
コード例 #4
0
 def test_run(self, mock_count, mock_eng, mock_boot, mock_parse):
     obj = UpdateACECommand()
     obj.options = Mock(from_ace='{"a": 1}',
                        to_ace='{"b": 2}',
                        models='User')
     model = Mock(__name__='Foo')
     mock_eng.get_document_cls.return_value = model
     mock_count.return_value = {model: 123}
     obj.run()
     mock_count.assert_called_once_with(from_ace={"a": 1},
                                        to_ace={"b": 2},
                                        models=[model])
コード例 #5
0
 def test_run_no_models(self, mock_count, mock_boot, mock_parse):
     obj = UpdateACECommand()
     obj.options = Mock(to_ace="{}", from_ace="{}", models=None)
     mock_count.return_value = {Mock(__name__="Foo"): 1}
     obj.run()
     mock_count.assert_called_once_with(to_ace={}, from_ace={}, models=None)
コード例 #6
0
 def test_run_no_models(self, mock_count, mock_boot, mock_parse):
     obj = UpdateACECommand()
     obj.options = Mock(to_ace='{}', from_ace='{}', models=None)
     mock_count.return_value = {Mock(__name__='Foo'): 1}
     obj.run()
     mock_count.assert_called_once_with(to_ace={}, from_ace={}, models=None)