def test_process_inserts(self): models.delete_tables() models.create_tables() new_playgrounds, revision_group = data.process_changes( 'tests/data/test_inserts.json') self.assertEqual(len(new_playgrounds), 1) playground = Playground.select().where( Playground.id == new_playgrounds[0].id)[0] self.assertEqual(playground.name, 'NEW NAME') revisions = Revision.select().where( Revision.revision_group == revision_group) self.assertEqual(revisions.count(), 1) revision = revisions[0] self.assertEqual(revision.playground.id, playground.id) log = revision.get_log() self.assertEqual(len(log), 1) self.assertEqual(log[0]['field'], 'name') self.assertEqual(log[0]['from'], '') self.assertEqual(log[0]['to'], 'NEW NAME') headers = revision.get_headers() self.assertEqual(headers['content_length'], '18') self.assertEqual(headers['host'], 'localhost') cookies = revision.get_cookies()
def test_process_inserts(self): models.delete_tables() models.create_tables() new_playgrounds, revision_group = data.process_changes('tests/data/test_inserts.json') self.assertEqual(len(new_playgrounds), 1) playground = Playground.select().where(Playground.id == new_playgrounds[0].id)[0] self.assertEqual(playground.name, 'NEW NAME') revisions = Revision.select().where(Revision.revision_group == revision_group) self.assertEqual(revisions.count(), 1) revision = revisions[0] self.assertEqual(revision.playground.id, playground.id) log = revision.get_log() self.assertEqual(len(log), 1) self.assertEqual(log[0]['field'], 'name') self.assertEqual(log[0]['from'], '') self.assertEqual(log[0]['to'], 'NEW NAME') headers = revision.get_headers() self.assertEqual(headers['content_length'], '18') self.assertEqual(headers['host'], 'localhost') cookies = revision.get_cookies()
def test_add_playground(self): models.delete_tables() models.create_tables() response = self.client.post(url_for('insert_playground'), data={ 'name': 'NEW PLAYGROUND' }) self.assertEqual(response.status_code, 302) redirect_url = app_config.S3_BASE_URL self.assertEqual(response.headers['Location'].split('?')[0], redirect_url + '/search.html') with open('data/changes.json') as f: inserts = json.load(f) self.assertEqual(len(inserts), 1) self.assertEqual(inserts[0]['action'], 'insert') self.assertEqual(inserts[0]['playground']['name'], 'NEW PLAYGROUND')
def test_add_playground(self): models.delete_tables() models.create_tables() response = self.client.post(url_for('insert_playground'), data={'name': 'NEW PLAYGROUND'}) self.assertEqual(response.status_code, 302) redirect_url = app_config.S3_BASE_URL self.assertEqual(response.headers['Location'].split('?')[0], redirect_url + '/search.html') with open('data/changes.json') as f: inserts = json.load(f) self.assertEqual(len(inserts), 1) self.assertEqual(inserts[0]['action'], 'insert') self.assertEqual(inserts[0]['playground']['name'], 'NEW PLAYGROUND')
def delete_tables(): """ Delete all database tables. """ models.delete_tables()
def load_test_playgrounds(): models.delete_tables() models.create_tables() data.load_playgrounds(TEST_PLAYGROUNDS_CSV)