From b9ee6abebd52a8446498bd1c208651653ec88e27 Mon Sep 17 00:00:00 2001
From: FreneticScribbler <aj@aronajones.com>
Date: Sat, 8 Jan 2022 17:36:28 +0000
Subject: [PATCH] Slightly more testing

---
 training/tests/pages.py     | 16 ++++++++++++++++
 training/tests/test_unit.py | 12 ++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/training/tests/pages.py b/training/tests/pages.py
index e69de29b..a657b827 100644
--- a/training/tests/pages.py
+++ b/training/tests/pages.py
@@ -0,0 +1,16 @@
+from django.urls import reverse
+from pypom import Region
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support import expected_conditions as EC
+
+from PyRIGS.tests import regions
+from PyRIGS.tests.pages import BasePage, FormPage
+
+class TraineeDetail(BasePage):
+    URL_TEMPLATE = 'trainee/{pk}'
+
+    _name_selector = (By.XPATH, '//h2')
+
+    @property
+    def page_name(self):
+        return self.find_element(*self._name_selector).text
diff --git a/training/tests/test_unit.py b/training/tests/test_unit.py
index 06b8b531..00664c41 100644
--- a/training/tests/test_unit.py
+++ b/training/tests/test_unit.py
@@ -23,3 +23,15 @@ def test_add_requirement(admin_client, level):
     url = reverse('add_requirement', kwargs={'pk': level.pk})
     response = admin_client.post(url)
     assertContains(response, level.pk)
+
+
+def test_trainee_detail(admin_client, trainee, admin_user):
+    url = reverse('trainee_detail', kwargs={'pk': admin_user.pk})
+    response = admin_client.get(url)
+    assertContains(response, "Your Training Record")
+    assertContains(response, "No qualifications in any levels")
+
+    url = reverse('trainee_detail', kwargs={'pk': trainee.pk})
+    response = admin_client.get(url)
+    assertNotContains(response, "Your")
+    assertContains(response, "{}'s Training Record".format(trainee.first_name + " " + trainee.last_name))