def test_get_object_from_string(get_model): BaseTestDataRestView.get_object('app.model:1') get_model.assert_called_once_with('app', 'model') get_model().objects.get.assert_called_once_with(pk=1)
def test_get_object_model(): model = Mock(**{'objects.get.return_value': 'object'}) assert_equal(BaseTestDataRestView.get_object(1, model), 'object') model.objects.get.assert_called_once_with(pk=1)