From 65a065c4eef5ec689ef51f670393df78629594da Mon Sep 17 00:00:00 2001
From: "Langhammer, Jens" <jens.langhammer@haufe-lexware.com>
Date: Mon, 14 Oct 2019 16:08:24 +0200
Subject: [PATCH] policy(minor): add unittests for policy engine

---
 .coveragerc                            |  1 +
 .gitlab-ci.yml                         |  4 +-
 passbook/policies/engine.py            |  5 +-
 passbook/policies/tests/__init__.py    |  0
 passbook/policies/tests/test_engine.py | 64 ++++++++++++++++++++++++++
 5 files changed, 70 insertions(+), 4 deletions(-)
 create mode 100644 passbook/policies/tests/__init__.py
 create mode 100644 passbook/policies/tests/test_engine.py

diff --git a/.coveragerc b/.coveragerc
index 552696206..e6fcbfa97 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -21,6 +21,7 @@ exclude_lines =
     def __str__
     def __repr__
     if self\.debug
+    if TYPE_CHECKING
 
     # Don't complain if tests don't hit defensive assertion code:
     raise AssertionError
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 34907e25e..a894e196a 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -80,9 +80,9 @@ pylint:
   - redis:latest
 coverage:
   script:
-    - coverage run manage.py test
+    - coverage run --concurrency=multiprocessing manage.py test
+    - coverage combine
     - coverage report
-    - coverage html
   stage: test
   services:
   - postgres:latest
diff --git a/passbook/policies/engine.py b/passbook/policies/engine.py
index 3d9cbf69e..813b37a32 100644
--- a/passbook/policies/engine.py
+++ b/passbook/policies/engine.py
@@ -1,7 +1,7 @@
 """passbook policy engine"""
 from multiprocessing import Pipe
 from multiprocessing.connection import Connection
-from typing import List, Tuple
+from typing import List, Optional, Tuple
 
 from django.core.cache import cache
 from django.http import HttpRequest
@@ -19,13 +19,14 @@ class PolicyProcessInfo:
 
     process: PolicyProcess
     connection: Connection
-    result: PolicyResult
+    result: Optional[PolicyResult]
     policy: Policy
 
     def __init__(self, process: PolicyProcess, connection: Connection, policy: Policy):
         self.process = process
         self.connection = connection
         self.policy = policy
+        self.result = None
 
 class PolicyEngine:
     """Orchestrate policy checking, launch tasks and return result"""
diff --git a/passbook/policies/tests/__init__.py b/passbook/policies/tests/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/passbook/policies/tests/test_engine.py b/passbook/policies/tests/test_engine.py
new file mode 100644
index 000000000..0a8f16202
--- /dev/null
+++ b/passbook/policies/tests/test_engine.py
@@ -0,0 +1,64 @@
+"""policy engine tests"""
+from django.core.cache import cache
+from django.test import TestCase
+
+from passbook.core.models import DebugPolicy, Policy, User
+from passbook.policies.engine import PolicyEngine
+
+
+class PolicyTestEngine(TestCase):
+    """PolicyEngine tests"""
+
+    def setUp(self):
+        cache.clear()
+        self.user = User.objects.create_user(
+            username="policyuser")
+        self.policy_false = DebugPolicy.objects.create(
+            result=False,
+            wait_min=0,
+            wait_max=1)
+        self.policy_true = DebugPolicy.objects.create(
+            result=True,
+            wait_min=0,
+            wait_max=1)
+        self.policy_negate = DebugPolicy.objects.create(
+            negate=True,
+            result=True,
+            wait_min=0,
+            wait_max=1)
+        self.policy_raises = Policy.objects.create(
+            name='raises')
+
+    def test_engine_empty(self):
+        """Ensure empty policy list passes"""
+        engine = PolicyEngine([], self.user)
+        self.assertEqual(engine.build().passing, True)
+
+    def test_engine_without_user(self):
+        """Ensure engine raises error if no user is set"""
+        with self.assertRaises(ValueError):
+            PolicyEngine([]).build()
+
+    def test_engine(self):
+        """Ensure all policies passes (Mix of false and true -> false)"""
+        engine = PolicyEngine(DebugPolicy.objects.filter(negate__exact=False), self.user)
+        self.assertEqual(engine.build().passing, False)
+
+    def test_engine_negate(self):
+        """Test negate flag"""
+        engine = PolicyEngine(DebugPolicy.objects.filter(negate__exact=True), self.user)
+        self.assertEqual(engine.build().passing, False)
+
+    def test_engine_policy_error(self):
+        """Test negate flag"""
+        engine = PolicyEngine(Policy.objects.filter(name='raises'), self.user)
+        self.assertEqual(engine.build().passing, False)
+
+    def test_engine_cache(self):
+        """Ensure empty policy list passes"""
+        engine = PolicyEngine(DebugPolicy.objects.filter(negate__exact=False), self.user)
+        self.assertEqual(len(cache.keys('policy_*')), 0)
+        self.assertEqual(engine.build().passing, False)
+        self.assertEqual(len(cache.keys('policy_*')), 2)
+        self.assertEqual(engine.build().passing, False)
+        self.assertEqual(len(cache.keys('policy_*')), 2)