def init_data(): """Fish data for project""" db.drop_all() db.create_all() user = User(username='******', email='*****@*****.**', password='******') user.save()
def init_data(): """Fish data for project""" db.drop_all() db.create_all() admin = User(username=app.config['ADMIN_USERNAME'], email=app.config['ADMIN_EMAIL'], password=app.config['ADMIN_PASSWORD']) admin.save()
def init_data(): """Fish data for project""" db.drop_all() db.create_all() admin = User( username=app.config["ADMIN_USERNAME"], email=app.config["ADMIN_EMAIL"], password=app.config["ADMIN_PASSWORD"] ) admin.save()
class KitTestCase(TestCase): def create_app(self): return AppFactory(TestingConfig).get_app(__name__) def setUp(self): db.create_all() self.user = User(username='******', email='*****@*****.**', password='******') self.user.save() def tearDown(self): db.session.remove() db.drop_all() def assertContains(self, response, text, count=None, status_code=200, msg_prefix=''): """ Asserts that a response indicates that some content was retrieved successfully, (i.e., the HTTP status code was as expected), and that ``text`` occurs ``count`` times in the content of the response. If ``count`` is None, the count doesn't matter - the assertion is true if the text occurs at least once in the response. """ if msg_prefix: msg_prefix += ": " self.assertEqual( response.status_code, status_code, msg_prefix + ( "Couldn't retrieve content: Response code was %d " "(expected %wd)" % (response.status_code, status_code) ) ) real_count = response.data.count(text.encode()) if count is not None: self.assertEqual( real_count, count, msg_prefix + ( "Found %d instances of '%s' in response " "(expected %d)" % (real_count, text, count) ) ) else: self.assertTrue( real_count != 0, msg_prefix + "Couldn't find '%s' in response" % text )
class KitTestCase(TestCase): def create_app(self): return AppFactory(TestingConfig).get_app(__name__) def setUp(self): db.create_all() self.user = User(username='******', email='*****@*****.**', password='******') self.user.save() def tearDown(self): db.session.remove() db.drop_all() def assertContains(self, response, text, count=None, status_code=200, msg_prefix=''): """ Asserts that a response indicates that some content was retrieved successfully, (i.e., the HTTP status code was as expected), and that ``text`` occurs ``count`` times in the content of the response. If ``count`` is None, the count doesn't matter - the assertion is true if the text occurs at least once in the response. """ if msg_prefix: msg_prefix += ": " self.assertEqual( response.status_code, status_code, msg_prefix + "Couldn't retrieve content: Response code was %d" " (expected %d)" % (response.status_code, status_code)) real_count = response.data.count(text) if count is not None: self.assertEqual( real_count, count, msg_prefix + "Found %d instances of '%s' in response" " (expected %d)" % (real_count, text, count)) else: self.assertTrue( real_count != 0, msg_prefix + "Couldn't find '%s' in response" % text)
class KitTestCase(TestCase): username = '******' email = '*****@*****.**' password = '******' login_url = '/login' logout_url = '/logout' def create_app(self): return AppFactory(TestingConfig).get_app(__name__) def setUp(self): db.create_all() self.user = User(username=self.username, email=self.email, password=self.password) self.user.save() def tearDown(self): db.session.remove() db.drop_all() def assertContains(self, response, text, count=None, status_code=200, msg_prefix=''): """ Asserts that a response indicates that some content was retrieved successfully, (i.e., the HTTP status code was as expected), and that ``text`` occurs ``count`` times in the content of the response. If ``count`` is None, the count doesn't matter - the assertion is true if the text occurs at least once in the response. """ if msg_prefix: msg_prefix += ": " self.assertEqual( response.status_code, status_code, msg_prefix + "Couldn't retrieve content: Response code was %d" " (expected %d)" % (response.status_code, status_code)) real_count = response.data.count(text) if count is not None: self.assertEqual( real_count, count, msg_prefix + "Found %d instances of '%s' in response" " (expected %d)" % (real_count, text, count)) else: self.assertTrue( real_count != 0, msg_prefix + "Couldn't find '%s' in response" % text) def login(self): return self.client.post(self.login_url, data=dict(email=self.email, password=self.password), follow_redirects=True) def logout(self): return self.client.post(self.logout_url, follow_redirects=True) def flash_messages(self, response): from StringIO import StringIO stri = StringIO(response.data) div = None while True: nl = stri.readline() if nl == '': return None if '<div class="l-75-c flash-messages">' in nl: div = nl break while True: nl = stri.readline() if nl == '': return None div += nl if '</div>' in nl: break return div