commit for publication

2026-09-17 14:41:22 +01:00
commit 0bf030f3c7
26 changed files with 2506 additions and 0 deletions
@@ -0,0 +1,293 @@
+
+import pytest
+import autotest_ui
+import flask
+
+
+@pytest.fixture
+def client():
+    autotest_ui.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:'
+    autotest_ui.app.config['TESTING'] = True
+
+
+    with autotest_ui.app.test_client() as client:
+        with autotest_ui.app.app_context():
+            autotest_ui.db.create_all()
+            yield client
+            autotest_ui.db.drop_all()
+
+
+
+# database helper functions
+
+def setup_create_user_admin(client):
+    with autotest_ui.app.app_context():
+        admin_user = autotest_ui.User(
+            username="admin",
+            email="admin@localhost",
+            password="admin",
+            user_role_value=autotest_ui.UserRoles.ADMIN
+        )
+        autotest_ui.db.session.add(admin_user)
+        autotest_ui.db.session.commit()
+
+
+
+
+def init_database(client):
+    setup_create_user_admin(client)
+
+
+# login helper function
+
+def login_as_admin(client):
+    return client.post("/signin", data={
+        "username": "admin",
+        "password": "admin"
+    }, follow_redirects=True)
+
+
+
+
+
+# test functions
+
+
+class Test_home:
+
+    def test_page_loads(self, client):
+        init_database(client)
+
+        response_logged_out = client.get("/")
+        assert response_logged_out.status_code == 200
+
+        login_as_admin(client)
+        response_logged_in = client.get("/")
+        assert response_logged_in.status_code == 200
+
+
+class Test_signin:
+
+    def test_signin(self, client):
+        init_database(client)
+
+        # ensure the user page doesn't load when logged out
+        response = client.get("/account")
+        assert response.status_code == 302
+        assert "/signin" in response.headers.get("Location")
+
+        # ensure the page loads
+        response_logged_out = client.get("/signin")
+        assert response_logged_out.status_code == 200
+
+        # actually login
+        login_as_admin(client)
+
+        # check the account page loads
+        response = client.get("/account")
+        assert response.status_code == 200
+        
+        # signing out redirects to signin page
+        response = client.get("/signout")
+        assert response.status_code == 302
+        assert "/signin" in response.headers.get("Location")
+
+        # ensure the uses is actually logged out
+        response = client.get("/account")
+        assert response.status_code == 302
+        assert "/signin" in response.headers.get("Location")
+
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 0, f"Unexpected errors found: {error_messages}"
+
+
+
+
+
+
+
+class Test_product_setup:
+
+    def test_page_loads(self, client):
+        init_database(client)
+        login_as_admin(client)
+        response = client.get("/product-setup")
+        assert response.status_code == 200
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 0, f"Unexpected errors found: {error_messages}"
+
+
+    def test_add_family(self, client):
+        init_database(client)
+        login_as_admin(client)
+        
+        response = client.post("/product-setup/add-family", data={
+            "family_name": "DPX-S"
+        })
+        assert response.status_code == 302
+
+        response = client.post("/product-setup/add-family", data={
+            "family_name": "DPX-M"
+        })
+        assert response.status_code == 302
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 0, f"Unexpected errors found: {error_messages}"
+
+        response = client.post("/product-setup/add-family", data={
+            "family_name": ""
+        })
+        assert response.status_code == 302
+
+        response = client.post("/product-setup/add-family")
+        assert response.status_code == 302
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 2, f"Expected 2 errors. Errors found: {error_messages}"
+
+        with autotest_ui.app.app_context():
+            inserted_family = autotest_ui.BoardFamily.query.filter_by(
+                board_family_prefix="DPX-S"
+            ).first()
+
+            assert inserted_family is not None, "The record was not found in the database"
+            assert inserted_family.board_family_prefix == "DPX-S"
+
+
+    def test_add_board(self, client):
+
+        init_database(client)
+        login_as_admin(client)
+
+        response = client.post("/product-setup/add-family", data={
+            "family_name": "DPX-S"
+        })
+        assert response.status_code == 302
+
+        with autotest_ui.app.app_context():
+            inserted_boards = autotest_ui.BoardFamily.query.filter_by(
+                board_family_prefix="DPX-S"
+            ).first()
+
+
+        response = client.post("/product-setup/add-board", data={
+            "board_name": "450",
+            "board_family_id": inserted_boards.board_family_id
+        })
+        assert response.status_code == 302
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 0, f"Unexpected errors found: {error_messages}"
+
+        
+
+
+
+
+class Test_operating_system:
+
+    def test_page_loads(self, client):
+        init_database(client)
+        login_as_admin(client)
+
+        response = client.get("/operating-system-setup")
+        assert response.status_code == 200
+    
+
+    def test_add_os(self, client):
+        init_database(client)
+        login_as_admin(client)
+
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Windows 10" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Windows 11" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Ubuntu 22" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Ubuntu 24" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Ubuntu 26" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Fedora 43" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Fedora 44" })
+        
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 0, f"Unexpected errors found: {error_messages}"
+
+        # test invalid records
+        client.post("/operating-system-setup/add-os", data={ "os_name": "" })
+        client.post("/operating-system-setup/add-os")
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Test-ing!£" })
+        client.post("/operating-system-setup/add-os", data={ "os_name": "Windows 11" })
+
+
+        with autotest_ui.app.app_context():
+            with client.session_transaction() as session:
+                flashed_messages = flask.get_flashed_messages(with_categories=True)
+                error_messages = [msg for cat, msg in flashed_messages if cat == "error"]
+                assert len(error_messages) == 4, f"Expected 4 errors. Errors found: {error_messages}"
+
+
+        with autotest_ui.app.app_context():
+            inserted_os = autotest_ui.OperatingSystem.query.all()
+            assert len(inserted_os) == 7
+
+
+
+class Test_projects:
+    
+    def test_project(self, client):
+
+        init_database(client)
+        login_as_admin(client)
+
+        response = client.get("/projects")
+        assert response.status_code == 200
+
+
+        client.post("/projects/add-project", data={
+            "project_name": "DirectPCI"
+        })
+
+        with autotest_ui.app.app_context():
+            inserted_project = autotest_ui.Project.query.all()
+            assert len(inserted_project) == 1
+        
+
+        
+        client.post("/projects/add-version", data={
+            "project_id": inserted_project[0].project_id,
+            "major": 1,
+            "minor": 2,
+            "micro": 0
+        })
+
+        client.post("/projects/add-version", data={
+            "project_id": inserted_project[0].project_id,
+            "major": 1,
+            "minor": 3,
+            "micro": 0
+        })
+
+
+        with autotest_ui.app.app_context():
+            inserted_project_version = autotest_ui.ProjectVersion.query.all()
+            assert len(inserted_project_version) == 2
+
+