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