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