[fix]:测试增添数据库保护

Author: indexYk

articles/27.pdf

@@ -10,11 +10,13 @@
 print(sys.path)
 
 from app.main import app
+from app.db.session import SessionLocal
 
 client = TestClient(app)
 # 初始化全局变量headers
 Headers = {}
 def setup_headers():
+    
     global Headers
     # 模拟登录获取令牌
     login_response = client.post("/public/login", json={
@@ -30,40 +32,48 @@ def setup_headers():
 
 
 def test_article_case1():
-    global Headers
-    # 确保headers已经设置
-    if not Headers:
-        setup_headers()
+    db = SessionLocal()
+    db.begin() 
+    try:
+        global Headers
+        # 确保headers已经设置
+        if not Headers:
+            setup_headers()
 
-    #创建文件夹
-    create_folder_response = client.post("/article/selfCreateFolder", headers=Headers, json={
-        "folder_name" : "测试文件夹1" 
-    })
-    assert create_folder_response.status_code == 200
+        #创建文件夹
+        create_folder_response = client.post("/article/selfCreateFolder", headers=Headers, json={
+            "folder_name" : "测试文件夹1" 
+        })
+        assert create_folder_response.status_code == 200
 
-    #获取文件夹列表
-    get_folder_list_response = client.get("/article/getSelfFolders", headers=Headers)
-    assert get_folder_list_response.status_code == 200
-    folders = get_folder_list_response.json().get('result')
-    folder_id = None
-    for folder in folders:
-        if folder.get('folder_name') == '测试文件夹1':
-            folder_id = folder.get('folder_id')
-            break
+        #获取文件夹列表
+        get_folder_list_response = client.get("/article/getSelfFolders", headers=Headers)
+        assert get_folder_list_response.status_code == 200
+        folders = get_folder_list_response.json().get('result')
+        folder_id = None
+        for folder in folders:
+            if folder.get('folder_name') == '测试文件夹1':
+                folder_id = folder.get('folder_id')
+                break
 
-    assert folder_id is not None
+        assert folder_id is not None
 
-    #获取该文件夹下的文件列表
-    get_file_list_response = client.get(f"/article/getArticlesInFolder", headers=Headers,
-                                          params={"folder_id": folder_id})
-    assert get_file_list_response.status_code == 200
-    files = get_file_list_response.json().get('result')
-    if len(files) == 0:
-        #向该文件夹下上传文件  
+        #获取该文件夹下的文件列表
+        get_file_list_response = client.get(f"/article/getArticlesInFolder", headers=Headers,
+                                            params={"folder_id": folder_id})
+        assert get_file_list_response.status_code == 200
+        files = get_file_list_response.json().get('result')
+        if len(files) == 0:
+            #向该文件夹下上传文件  
 
-        upload_file_response = client.post("/article/uploadToSelfFolder", params={"folder_id": folder_id},headers=Headers, files={
-        "article": os.path.join(project_root, "tests/in/test.pdf")})
-        assert upload_file_response.status_code == 200
+            upload_file_response = client.post("/article/uploadToSelfFolder", params={"folder_id": folder_id},headers=Headers, files={
+            "article": os.path.join(project_root, "tests/in/test.pdf")})
+            assert upload_file_response.status_code == 200
+    except Exception as e:
+        db.rollback()
+        raise e
+    finally:
+        db.close()
 
 
 

articles/28.pdf

@@ -10,6 +10,7 @@
 print(sys.path)
 
 from app.main import app
+from app.db.session import SessionLocal
 
 client = TestClient(app)
 # 初始化全局变量headers
@@ -32,68 +33,84 @@ def setup_headers():
 
 # 创建后删除笔记
 def test_note_case1():
-    global Headers
-    # 确保headers已经设置
-    if not Headers:
-        setup_headers()
-
-
-    # 创建笔记
-    create_response = client.post("/notes", json={
-        "article_id": 1,
-        "content": "<p> 12 <p>",
-        "title": "test"
-    }, headers=Headers)
-    assert create_response.status_code == 200
-    note_id = create_response.json().get('note_id')
-    if note_id is None:
-        raise ValueError("Failed to get note_id from create note response")
-
-    # 删除笔记
-    delete_response = client.delete(f"/notes/{note_id}", headers=Headers)
-    assert delete_response.status_code == 200
-
-    # 再次删除不存在的笔记
-    double_delete_response = client.delete(f"/notes/{note_id}", headers=Headers)
-    assert double_delete_response.status_code != 200
+    db = SessionLocal()
+    db.begin() 
+    try:
+        global Headers
+        # 确保headers已经设置
+        if not Headers:
+            setup_headers()
+
+
+        # 创建笔记
+        create_response = client.post("/notes", json={
+            "article_id": 1,
+            "content": "<p> 12 <p>",
+            "title": "test"
+        }, headers=Headers)
+        assert create_response.status_code == 200
+        note_id = create_response.json().get('note_id')
+        if note_id is None:
+            raise ValueError("Failed to get note_id from create note response")
+
+        # 删除笔记
+        delete_response = client.delete(f"/notes/{note_id}", headers=Headers)
+        assert delete_response.status_code == 200
+
+        # 再次删除不存在的笔记
+        double_delete_response = client.delete(f"/notes/{note_id}", headers=Headers)
+        assert double_delete_response.status_code != 200
+    except Exception as e:
+        db.rollback()
+        raise e
+    finally:
+        db.close()
 
 
 def test_note_case2():
-    global Headers
-    if not Headers:
-        setup_headers()
-
-    # 创建笔记
-    create_response = client.post("/notes", json={
-        "article_id": 1,
-        "content": "<p> 12 <p>",
-        "title": "test"
-    }, headers=Headers)
-    assert create_response.status_code == 200
-    note_id = create_response.json().get('note_id')
-    if note_id is None:
-        raise ValueError("Failed to get note_id from create note response")
-
-    print(note_id)
-    # 更新笔记
-    update_response = client.put(f"/notes/{note_id}", params={
-        "content": "<p> 123 <p>",
-        "title": "test2"
-    }, headers=Headers)
-    print(update_response.json())
-    assert update_response.status_code == 200
-
-    
-
-    # 获取笔记
-    get_response = client.get(f"/notes",params={
-        "id": note_id
-    }, headers=Headers)
-    assert get_response.status_code == 200
-    assert get_response.json().get('notes')[0].get('id') == note_id
-    assert get_response.json().get('notes')[0].get('content') == "<p> 123 <p>"
-    
-
-    #删除笔记
-    delete_response = client.delete(f"/notes/{note_id}", headers=Headers)
-    assert delete_response.status_code == 200
+    db = SessionLocal()
+    db.begin() 
+    try:
+        global Headers
+        if not Headers:
+            setup_headers()
+
+        # 创建笔记
+        create_response = client.post("/notes", json={
+            "article_id": 1,
+            "content": "<p> 12 <p>",
+            "title": "test"
+        }, headers=Headers)
+        assert create_response.status_code == 200
+        note_id = create_response.json().get('note_id')
+        if note_id is None:
+            raise ValueError("Failed to get note_id from create note response")
+
+        print(note_id)
+        # 更新笔记
+        update_response = client.put(f"/notes/{note_id}", params={
+            "content": "<p> 123 <p>",
+            "title": "test2"
+        }, headers=Headers)
+        print(update_response.json())
+        assert update_response.status_code == 200
+
+        
+
+        # 获取笔记
+        get_response = client.get(f"/notes",params={
+            "id": note_id
+        }, headers=Headers)
+        assert get_response.status_code == 200
+        assert get_response.json().get('notes')[0].get('id') == note_id
+        assert get_response.json().get('notes')[0].get('content') == "<p> 123 <p>"
+        
+
+        #删除笔记
+        delete_response = client.delete(f"/notes/{note_id}", headers=Headers)
+        assert delete_response.status_code == 200
+    except Exception as e:
+        db.rollback()
+        raise e
+    finally:
+        db.close()

articles/29.pdf

@@ -10,6 +10,7 @@
 print(sys.path)
 
 from app.main import app
+from app.db.session import SessionLocal
 
 client = TestClient(app)
 # 初始化全局变量headers
@@ -31,68 +32,84 @@ def setup_headers():
 
 
 def test_user_case1():
-    global Headers
-    if not Headers:
-        setup_headers()
-    # 修改信息
-    save_response = client.get("/user", headers=Headers)
-    assert save_response.status_code == 200
-    save_data = save_response.json()
-    
-    advise_response = client.put("/user", data={"username":"李国庆test",
-                                                 "address": "北京市海淀区中关村",
-                                                 "university": "北京大学",
-    }, headers=Headers)
-    assert advise_response.status_code == 200
+    db = SessionLocal()
+    db.begin() 
+    try:
+        global Headers
+        if not Headers:
+            setup_headers()
+        # 修改信息
+        save_response = client.get("/user", headers=Headers)
+        assert save_response.status_code == 200
+        save_data = save_response.json()
+        
+        advise_response = client.put("/user", data={"username":"李国庆test",
+                                                    "address": "北京市海淀区中关村",
+                                                    "university": "北京大学",
+        }, headers=Headers)
+        assert advise_response.status_code == 200
 
-    # 获取修改后的信息
-    adviser_response = client.get("/user", headers=Headers)
-    assert adviser_response.status_code == 200
-    assert adviser_response.json().get('address') == "北京市海淀区中关村"
-    assert adviser_response.json().get('university') == "北京大学"
-    assert adviser_response.json().get('username') == "李国庆test"
+        # 获取修改后的信息
+        adviser_response = client.get("/user", headers=Headers)
+        assert adviser_response.status_code == 200
+        assert adviser_response.json().get('address') == "北京市海淀区中关村"
+        assert adviser_response.json().get('university') == "北京大学"
+        assert adviser_response.json().get('username') == "李国庆test"
 
-    restore_response = client.put("/user", data={
-        "username": save_data.get('username'),
-        "address": save_data.get('address'),
-        "university": save_data.get('university')
-    }, headers=Headers)
-    assert restore_response.status_code == 200
+        restore_response = client.put("/user", data={
+            "username": save_data.get('username'),
+            "address": save_data.get('address'),
+            "university": save_data.get('university')
+        }, headers=Headers)
+        assert restore_response.status_code == 200
+    except Exception as e:
+        db.rollback()
+        raise e
+    finally:
+        db.close()
 
 def test_user_case2():
-    global Headers
-    if not Headers:
-        setup_headers()
-    
-    # 修改密码
-    password_response = client.post("/user/password", json={
-        "old_password": "123456",
-        "new_password": "654321"
-        }, headers=Headers)
-    assert password_response.status_code == 200
-    # 验证密码是否修改成功
-    login_response = client.post("/public/login", json={
-        "email": "[email protected]",
-        "password": "123456"
-    })
-    assert login_response.status_code == 401
-    login_response = client.post("/public/login", json={
-        "email": "[email protected]",
-        "password": "654321"
-    })
-    assert login_response.status_code == 200
-    # 还原密码
-    restore_response = client.post("/user/password", json={
-        "old_password": "654321",
-        "new_password": "123456"
-        }, headers=Headers)
-    assert restore_response.status_code == 200
-    # 验证密码是否还原成功
-    login_response = client.post("/public/login", json={
-        "email": "[email protected]",
-        "password": "123456"
-    })
-    assert login_response.status_code == 200
+    db = SessionLocal()
+    db.begin() 
+    try:
+        global Headers
+        if not Headers:
+            setup_headers()
+        
+        # 修改密码
+        password_response = client.post("/user/password", json={
+            "old_password": "123456",
+            "new_password": "654321"
+            }, headers=Headers)
+        assert password_response.status_code == 200
+        # 验证密码是否修改成功
+        login_response = client.post("/public/login", json={
+            "email": "[email protected]",
+            "password": "123456"
+        })
+        assert login_response.status_code == 401
+        login_response = client.post("/public/login", json={
+            "email": "[email protected]",
+            "password": "654321"
+        })
+        assert login_response.status_code == 200
+        # 还原密码
+        restore_response = client.post("/user/password", json={
+            "old_password": "654321",
+            "new_password": "123456"
+            }, headers=Headers)
+        assert restore_response.status_code == 200
+        # 验证密码是否还原成功
+        login_response = client.post("/public/login", json={
+            "email": "[email protected]",
+            "password": "123456"
+        })
+        assert login_response.status_code == 200
+    except Exception as e:
+        db.rollback()
+        raise e
+    finally:
+        db.close()