chore: Create tests

Author: jackylamhk

poetry.lock

@@ -9,6 +9,9 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = ">3.9"
 
+[tool.poetry.group.dev.dependencies]
+pytest = "^7.4.3"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

pyproject.toml

@@ -0,0 +1,27 @@
+import pytest
+from http_csp.csp import CSP
+
+
+def test_csp_parsing_clean():
+    policy = CSP("base-uri self; child-src self http://localhost;")
+    assert policy.base_uri == ["self"]
+    assert policy.child_src == ["self", "http://localhost"]
+
+
+def test_csp_parsing_unclean():
+    policy = CSP("   base_uri self;child_src  self  http://localhost ;  ")
+    assert policy.base_uri == ["self"]
+    assert policy.child_src == ["self", "http://localhost"]
+
+
+def test_csp_parsing_error():
+    with pytest.raises(ValueError):
+        CSP("non-valid option")
+
+
+def test_csp_generation():
+    policy = CSP()
+    policy.base_uri = ["self"]
+    policy.child_src = ["self", "http://localhost"]
+    generated_policy = policy.generate()
+    assert generated_policy == "base-uri self; child-src self http://localhost;"