def test_soft_deletes_are_not_retrieved_from_builder_helpers(self): self.create_users() count = 0 query = SoftDeletesTestUser.query() for users in query.chunk(2): count += len(users) self.assertEqual(1, count) query = SoftDeletesTestUser.query() self.assertEqual(1, len(query.lists("email"))) Paginator.current_page_resolver(lambda: 1) query = SoftDeletesTestUser.query() self.assertEqual(1, len(query.paginate(2).items)) Paginator.current_page_resolver(lambda: 1) query = SoftDeletesTestUser.query() self.assertEqual(1, len(query.simple_paginate(2).items)) self.assertEqual( 0, SoftDeletesTestUser.where("email", "*****@*****.**").increment("id") ) self.assertEqual( 0, SoftDeletesTestUser.where("email", "*****@*****.**").decrement("id") )
def test_returns_relevant_context(self): p = Paginator(['item3', 'item4', 'item5'], 2, 2) self.assertEqual(2, p.current_page) self.assertTrue(p.has_pages()) self.assertTrue(p.has_more_pages()) self.assertEqual(['item3', 'item4'], p.items) self.assertEqual(2, p.per_page) self.assertEqual(3, p.next_page) self.assertEqual(1, p.previous_page) self.assertEqual(3, p.first_item) self.assertEqual(4, p.last_item) self.assertEqual('item4', p[1])
def init_app(self, app): if 'ORATOR_DATABASES' not in app.config: raise RuntimeError('Missing "ORATOR_DATABASES" configuration') self._config = app.config['ORATOR_DATABASES'] self.Model.set_connection_resolver(self) # Setting current page resolver def current_page_resolver(): return int(request.args.get('page', 1)) Paginator.current_page_resolver(current_page_resolver) self.init_commands(app)
def init_app(self, app): if 'ORATOR_DATABASES' not in app.config: raise RuntimeError('Missing "ORATOR_DATABASES" configuration') # Register request hooks self.register_handlers(app) # Getting config databases self._config = app.config['ORATOR_DATABASES'] # Initializing database manager self._db = self._manager_class(self._config) self.Model.set_connection_resolver(self._db) # Setting current page resolver Paginator.current_page_resolver(self._current_page_resolver) # Setting commands self.init_commands()
def test_soft_deletes_are_not_retrieved_from_builder_helpers(self): self.create_users() count = 0 query = SoftDeletesTestUser.query() for users in query.chunk(2): count += len(users) self.assertEqual(1, count) query = SoftDeletesTestUser.query() self.assertEqual(1, len(query.lists('email'))) Paginator.current_page_resolver(lambda: 1) query = SoftDeletesTestUser.query() self.assertEqual(1, len(query.paginate(2).items)) Paginator.current_page_resolver(lambda: 1) query = SoftDeletesTestUser.query() self.assertEqual(1, len(query.simple_paginate(2).items)) self.assertEqual(0, SoftDeletesTestUser.where('email', '*****@*****.**').increment('id')) self.assertEqual(0, SoftDeletesTestUser.where('email', '*****@*****.**').decrement('id'))
def test_current_page_resolver(self): def current_page_resolver(): return 7 Paginator.current_page_resolver(current_page_resolver) self.assertEqual(7, Paginator.resolve_current_page())