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_get_last_batch_number_returns_max_batch(self): repo = self.get_repository() connection = flexmock(Connection(None)) query = flexmock(QueryBuilder(connection, None, None)) repo.get_connection_resolver().should_receive('connection').with_args( None).and_return(connection) repo.get_connection().should_receive('table').once().with_args( 'migrations').and_return(query) query.should_receive('max').and_return(1) self.assertEqual(1, repo.get_last_batch_number())
def test_apply_scope_to_a_builder(self): scope = SoftDeletingScope() query = flexmock(QueryBuilder(None, None, None)) builder = Builder(query) model = flexmock(ModelStub()) model.should_receive('get_qualified_deleted_at_column').once( ).and_return('table.deleted_at') builder.get_query().should_receive('where_null').once().with_args( 'table.deleted_at') scope.apply(builder, model)
def test_log_inserts_record_into_migration_table(self): repo = self.get_repository() connection = flexmock(Connection(None)) query = flexmock(QueryBuilder(connection, None, None)) repo.get_connection_resolver().should_receive('connection').with_args( None).and_return(connection) repo.get_connection().should_receive('table').once().with_args( 'migrations').and_return(query) query.should_receive('insert').once().with_args(migration='bar', batch=1) repo.log('bar', 1)
def test_get_ran_migrations_list_migrations_by_package(self): repo = self.get_repository() connection = flexmock(Connection(None)) query = flexmock(QueryBuilder(connection, None, None)) repo.get_connection_resolver().should_receive('connection').with_args( None).and_return(connection) repo.get_connection().should_receive('table').once().with_args( 'migrations').and_return(query) query.should_receive('lists').once().with_args('migration').and_return( 'bar') self.assertEqual('bar', repo.get_ran())
def test_delete_sets_soft_deleted_column(self): model = flexmock(SoftDeleteModelStub()) model.set_exists(True) builder = flexmock(Builder) query_builder = flexmock(QueryBuilder(None, None, None)) query = Builder(query_builder) model.should_receive('new_query').and_return(query) builder.should_receive('where').once().with_args('id', 1).and_return(query) builder.should_receive('update').once().with_args({'deleted_at': t}) model.delete() self.assertIsInstance(model.deleted_at, datetime.datetime)
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_scope_can_remove_deleted_at_constraint(self): scope = SoftDeletingScope() query = flexmock(QueryBuilder(None, None, None)) model = flexmock(ModelStub()) builder = Builder(query) builder.set_model(model) model.should_receive('get_qualified_deleted_at_column').once( ).and_return('table.deleted_at') query.wheres = [{ 'type': 'null', 'column': 'foo' }, { 'type': 'null', 'column': 'table.deleted_at' }] scope.remove(builder, model) self.assertEqual(query.wheres, [{'type': 'null', 'column': 'foo'}])
def test_get_last_migrations(self): resolver_mock = flexmock(DatabaseManager) resolver_mock.should_receive('connection').and_return(None) resolver = flexmock(resolver_mock({})) repo = flexmock(DatabaseMigrationRepository(resolver, 'migrations')) connection = flexmock(Connection(None)) query = flexmock(QueryBuilder(connection, None, None)) repo.should_receive('get_last_batch_number').and_return(1) repo.get_connection_resolver().should_receive('connection').with_args( None).and_return(connection) repo.get_connection().should_receive('table').once().with_args( 'migrations').and_return(query) query.should_receive('where').once().with_args('batch', 1).and_return(query) query.should_receive('order_by').once().with_args( 'migration', 'desc').and_return(query) query.should_receive('get').once().and_return('foo') self.assertEqual('foo', repo.get_last())
def test_delete_removes_migration_from_table(self): repo = self.get_repository() connection = flexmock(Connection(None)) query = flexmock(QueryBuilder(connection, None, None)) repo.get_connection_resolver().should_receive('connection').with_args( None).and_return(connection) repo.get_connection().should_receive('table').once().with_args( 'migrations').and_return(query) query.should_receive('where').once().with_args('migration', 'foo').and_return(query) query.should_receive('delete').once() class Migration(object): migration = 'foo' def __getitem__(self, item): return self.migration repo.delete(Migration())