def test_force_delete_extension(self): scope = SoftDeletingScope() builder = Builder(None) scope.extend(builder) callback = builder.get_macro("force_delete") query = flexmock(QueryBuilder(None, None, None)) given_builder = Builder(query) query.should_receive("delete").once() callback(given_builder)
def test_force_delete_extension(self): scope = SoftDeletingScope() builder = Builder(None) scope.extend(builder) callback = builder.get_macro('force_delete') query = flexmock(QueryBuilder(None, None, None)) given_builder = Builder(query) query.should_receive('delete').once() callback(given_builder)
def test_with_trashed_extension(self): scope = flexmock(SoftDeletingScope()) builder = Builder(None) scope.extend(builder) callback = builder.get_macro('with_trashed') query = flexmock(QueryBuilder(None, None, None)) given_builder = Builder(query) model = flexmock(ModelStub()) given_builder.set_model(model) scope.should_receive('remove').once().with_args(given_builder, model) result = callback(given_builder) self.assertEqual(given_builder, result)
def test_restore_extension(self): scope = SoftDeletingScope() builder = Builder(None) scope.extend(builder) callback = builder.get_macro("restore") query = flexmock(QueryBuilder(None, None, None)) builder_mock = flexmock(BuilderWithTrashedStub) given_builder = BuilderWithTrashedStub(query) builder_mock.should_receive("with_trashed").once() builder_mock.should_receive("get_model").once().and_return(ModelStub()) builder_mock.should_receive("update").once().with_args({"deleted_at": None}) callback(given_builder)
def test_restore_extension(self): scope = SoftDeletingScope() builder = Builder(None) scope.extend(builder) callback = builder.get_macro('restore') query = flexmock(QueryBuilder(None, None, None)) builder_mock = flexmock(BuilderWithTrashedStub) given_builder = BuilderWithTrashedStub(query) builder_mock.should_receive('with_trashed').once() builder_mock.should_receive('get_model').once().and_return(ModelStub()) builder_mock.should_receive('update').once().with_args({'deleted_at': None}) callback(given_builder)
def test_with_trashed_extension(self): scope = flexmock(SoftDeletingScope()) builder = Builder(None) scope.extend(builder) callback = builder.get_macro("with_trashed") query = flexmock(QueryBuilder(None, None, None)) given_builder = Builder(query) model = flexmock(ModelStub()) given_builder.set_model(model) given_builder.remove_global_scope = mock.MagicMock(return_value=given_builder) result = callback(given_builder) self.assertEqual(given_builder, result) given_builder.remove_global_scope.assert_called_with(scope)
def test_with_trashed_extension(self): scope = flexmock(SoftDeletingScope()) builder = Builder(None) scope.extend(builder) callback = builder.get_macro('with_trashed') query = flexmock(QueryBuilder(None, None, None)) given_builder = Builder(query) model = flexmock(ModelStub()) given_builder.set_model(model) given_builder.remove_global_scope = mock.MagicMock(return_value=given_builder) result = callback(given_builder) self.assertEqual(given_builder, result) given_builder.remove_global_scope.assert_called_with(scope)