fixed testing script
This commit is contained in:
27
test_app.py
Normal file
27
test_app.py
Normal file
@@ -0,0 +1,27 @@
|
||||
import pytest
|
||||
import os
|
||||
from app import app
|
||||
from models import db, Feature
|
||||
|
||||
@pytest.fixture
|
||||
def client():
|
||||
app.config['TESTING'] = True
|
||||
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
|
||||
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
||||
db.init_app(app) # Re-init DB with test config
|
||||
|
||||
with app.test_client() as client:
|
||||
with app.app_context():
|
||||
db.create_all()
|
||||
yield client
|
||||
db.drop_all()
|
||||
|
||||
def test_index(client):
|
||||
response = client.get('/')
|
||||
assert response.status_code == 200
|
||||
assert b'Professional Demo Site' in response.data
|
||||
|
||||
def test_seed_db(client): # Add fixture here
|
||||
with app.app_context():
|
||||
Feature.seed_db()
|
||||
assert Feature.query.count() == 3
|
||||
Reference in New Issue
Block a user