Reorganize the server testing directory (#1705)

2026-09-30 06:58:12 +08:00 · 2020-08-05 08:31:02 -07:00
parent 550847f763
commit cdae4f9f10
133 changed files with 72 additions and 69 deletions
@@ -0,0 +1,147 @@
+import unittest
+
+import requests
+
+from server.common.app_config import AppConfig
+from server.test import FIXTURES_ROOT, test_server
+
+
+class AuthTest(unittest.TestCase):
+    def setUp(self):
+        self.dataset_dataroot = FIXTURES_ROOT
+
+    def test_auth_none(self):
+        c = AppConfig()
+        c.update_server_config(
+            authentication__type=None, multi_dataset__dataroot=self.dataset_dataroot
+        )
+        c.update_default_dataset_config(user_annotations__enable=False)
+
+        c.complete_config()
+
+        with test_server(app_config=c) as server:
+            session = requests.Session()
+            r = session.get(f"{server}/d/pbmc3k.cxg/api/v0.2/config")
+            data_config = r.json()
+            assert "authentication" not in data_config["config"]
+
+    def test_auth_session(self):
+        c = AppConfig()
+        c.update_server_config(
+            authentication__type="session", multi_dataset__dataroot=self.dataset_dataroot
+        )
+        c.update_default_dataset_config(user_annotations__enable=True)
+        c.complete_config()
+
+        with test_server(app_config=c) as server:
+            session = requests.Session()
+            r = session.get(f"{server}/d/pbmc3k.cxg/api/v0.2/config")
+            data_config = r.json()
+            assert data_config["config"]["authentication"]["is_authenticated"]
+            assert not data_config["config"]["authentication"]["requires_client_login"]
+            assert data_config["config"]["authentication"]["username"] == "anonymous"
+
+    def test_auth_test(self):
+        c = AppConfig()
+        c.update_server_config(authentication__type="test")
+        c.update_server_config(
+            multi_dataset__dataroot=dict(
+                a1=dict(dataroot=self.dataset_dataroot, base_url="auth"),
+                a2=dict(dataroot=self.dataset_dataroot, base_url="no-auth"),
+            )
+        )
+
+        # specialize the configs
+        c.add_dataroot_config("a1", app__authentication_enable=True, user_annotations__enable=True)
+        c.add_dataroot_config("a2", app__authentication_enable=False, user_annotations__enable=False)
+
+        c.complete_config()
+
+        with test_server(app_config=c) as server:
+            session = requests.Session()
+
+            # auth datasets
+            r = session.get(f"{server}/auth/pbmc3k.cxg/api/v0.2/config")
+            data_config = r.json()
+            assert not data_config["config"]["authentication"]["is_authenticated"]
+            assert data_config["config"]["authentication"]["requires_client_login"]
+            assert data_config["config"]["authentication"]["username"] is None
+            assert data_config["config"]["parameters"]["annotations"]
+
+            login_uri = data_config["config"]["authentication"]["login"]
+            logout_uri = data_config["config"]["authentication"]["logout"]
+
+            assert login_uri == "/login?dataset=auth/pbmc3k.cxg"
+            assert logout_uri == "/logout?dataset=auth/pbmc3k.cxg"
+
+            r = session.get(f"{server}/{login_uri}")
+            # check that the login redirect worked
+            assert r.history[0].status_code == 302
+            assert r.url == f"{server}/auth/pbmc3k.cxg/"
+
+            r = session.get(f"{server}/auth/pbmc3k.cxg/api/v0.2/config")
+            data_config = r.json()
+            assert data_config["config"]["authentication"]["is_authenticated"]
+            assert data_config["config"]["authentication"]["username"] == "test_account"
+            assert data_config["config"]["parameters"]["annotations"]
+
+            r = session.get(f"{server}/{logout_uri}")
+            # check that the logout redirect worked
+            assert r.history[0].status_code == 302
+            assert r.url == f"{server}/auth/pbmc3k.cxg/"
+            r = session.get(f"{server}/auth/pbmc3k.cxg/api/v0.2/config")
+            data_config = r.json()
+            assert not data_config["config"]["authentication"]["is_authenticated"]
+            assert data_config["config"]["authentication"]["username"] is None
+            assert data_config["config"]["parameters"]["annotations"]
+
+            # no-auth datasets
+            r = session.get(f"{server}/no-auth/pbmc3k.cxg/api/v0.2/config")
+            data_config = r.json()
+            assert "authentication" not in data_config["config"]
+            assert not data_config["config"]["parameters"]["annotations"]
+
+    def test_auth_test_single(self):
+        c = AppConfig()
+        c.update_server_config(
+            authentication__type="test",
+            single_dataset__datapath=f"{self.dataset_dataroot}/pbmc3k.cxg")
+
+        c.complete_config()
+
+        with test_server(app_config=c) as server:
+            session = requests.Session()
+
+            r = session.get(f"{server}/api/v0.2/config")
+            data_config = r.json()
+            assert not data_config["config"]["authentication"]["is_authenticated"]
+            assert data_config["config"]["authentication"]["requires_client_login"]
+            assert data_config["config"]["authentication"]["username"] is None
+            assert data_config["config"]["parameters"]["annotations"]
+
+            login_uri = data_config["config"]["authentication"]["login"]
+            logout_uri = data_config["config"]["authentication"]["logout"]
+
+            assert login_uri == "/login"
+            assert logout_uri == "/logout"
+
+            r = session.get(f"{server}/{login_uri}")
+            # check that the login redirect worked
+            assert r.history[0].status_code == 302
+            assert r.url == f"{server}/"
+
+            r = session.get(f"{server}/api/v0.2/config")
+            data_config = r.json()
+            assert data_config["config"]["authentication"]["is_authenticated"]
+            assert data_config["config"]["authentication"]["username"] == "test_account"
+            assert data_config["config"]["parameters"]["annotations"]
+
+            r = session.get(f"{server}/{logout_uri}")
+            # check that the logout redirect worked
+            assert r.history[0].status_code == 302
+            assert r.url == f"{server}/"
+            r = session.get(f"{server}/api/v0.2/config")
+            data_config = r.json()
+            assert not data_config["config"]["authentication"]["is_authenticated"]
+            assert data_config["config"]["authentication"]["username"] is None
+            assert data_config["config"]["parameters"]["annotations"]