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_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)