import unittest
from fastapi.testclient import TestClient
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from pick_student.database import Base
from pick_student.login import get_password_hash
from pick_student.main import get_db
from run import app  # 从 run.py 导入主 app
from pick_student.crud import create_teacher

# 设置测试数据库
SQLALCHEMY_DATABASE_URL = "sqlite:///./test_test.db"  # 使用不同的数据库文件名
engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# 创建所有表
Base.metadata.create_all(bind=engine)

# 重写 get_db 函数用于测试
def override_get_db():
    db = TestingSessionLocal()
    try:
        yield db
    except Exception as e:
        db.close()
        raise e  # 重新抛出异常
    finally:
        db.close()  # 确保数据库连接被正确关闭

# 覆盖 app 中的依赖项
app.dependency_overrides[get_db] = override_get_db

# 创建测试客户端
client = TestClient(app)

class TestAuthRoutes(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        # 在测试开始时设置初始数据库状态
        db = TestingSessionLocal()
        hashed_password = get_password_hash("password")  # 假设已经生成一个哈希密码
        create_teacher(db, user_name="test_teacher", password_hash=hashed_password)
        db.close()

    def test_login_success(self):
        # 测试登录成功的情况
        response = client.post(
            "/teacher/jwt/token",
            data={"username": "test_teacher", "password": "password"},
        )
        print(response.json())
        self.assertEqual(response.status_code, 200)
        data = response.json()
        self.assertIn("access_token", data)
        self.assertIn("token_type", data)

    def test_login_failure(self):
        # 测试登录失败的情况
        response = client.post(
            "/teacher/jwt/token",
            data={"username": "wrong_teacher", "password": "wrong_password"},
        )
        self.assertEqual(response.status_code, 401)
        data = response.json()
        self.assertEqual(data["detail"], "Incorrect username or password")

    def test_register_user(self):
        # 测试注册新用户
        response = client.post(
            "/teacher/jwt/register",
            data={"username": "new_teacher", "password": "new_password"},
        )
        print(response.json())
        self.assertEqual(response.status_code, 200)
        data = response.json()
        self.assertIn("access_token", data)
        self.assertIn("token_type", data)

    @classmethod
    def tearDownClass(cls):
        # 清理测试数据库
        Base.metadata.drop_all(bind=engine)

if __name__ == "__main__":
    unittest.main()