def setUp(self): """Before each test, set up a blank database""" # Create MongoDB test database connection = pymongo.Connection('localhost', 27017) db = connection['flaskr_test'] connection.close() flaskr.app.config['TESTING'] = True flaskr.connect_db('flaskr_test') self.app = flaskr.app.test_client()
def add_entry(): if not session.get('logged_in'): abort(401) print request.form['title'] print request.form['text'] g.db=flaskr.connect_db() g.db.execute('insert into entries (title, text) values (?, ?)', [request.form['title'], request.form['text']]) g.db.commit() flash('New entry was successfully posted') return redirect(url_for('show_entries'))
def test_database_setup(self): """ Testing that table info returns three rows, as the SQL statement should return one row for each table column. """ con = flaskr.connect_db() cur = con.execute('PRAGMA table_info(entries);') rows = cur.fetchall() self.assertEquals(len(rows), 3)
def test_write_entry(self): expected = ("My Title","My Text") with self.app.test_request_context('/'): self.app.preprocess_request() flaskr.write_entry(*expected) con = flaskr.connect_db() cur = con.execute("select * from entries;") rows = cur.fetchall() self.assertEquals(len(rows), 1) for val in expected: self.assertTrue(val in rows[0])
def test_write_entry(self): """ Test the write_entry() method of flaskr. """ expected = ("My Title", "My Text") with self.app.test_request_context('/'): self.app.preprocess_request() # Insert our test data into the test database. flaskr.write_entry(*expected) con = flaskr.connect_db() # Check for presence of test data. cur = con.execute("select * from entries;") rows = cur.fetchall() self.assertEquals(len(rows), 1) for val in expected: # Actually check the values against expected. self.assertTrue(val in rows[0])
def test_db_setup(self): con = flaskr.connect_db() cur = con.execute('PRAGMA table_info(entries)') rows = cur.fetchall() self.assertEquals(len(rows), 3)
def before_request(): g.db = connect_db()
def init_db(): with closing(connect_db()) as db: with app.open_resource("schema.sql") as f: db.cursor().executescript(f.read()) db.commit()
def init_db(): with closing(connect_db()) as db: with app.open_resource('schema.sql') as f: #cursor(游标),用来执行数据库命令。可以excute数据命令、exeutescript执行脚本。 db.cursor().executescript(f.read()) db.commit()
def setUp(self): """Before each test, set up a blank database""" self.db_fd, flaskr.app.config['DATABASE'] = tempfile.mkstemp() self.app = flaskr.app.test_client() flaskr.connect_db()
def test_database_setup(self): con = flaskr.connect_db() cur = con.execute('PRAGMA table_info(entries);') rows = cur.fetchall() self.assertTrue(len(rows) == 3)
def show_entries(): g.db=flaskr.connect_db() cur = g.db.execute('select title, text from entries order by id desc') entries = [dict(title=row[0], text=row[1]) for row in cur.fetchall()] return render_template('show_entries.html', entries=entries)
def setUp(self): self.db_fd, flaskr.app.config["DATABASE"] = tempfile.mkstemp() flaskr.app.config["TESTING"] = True self.app = flaskr.app.test_client() flaskr.connect_db()