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()
def before_request():
g.db = connect_db()
