From 2e872fa1044bb71358896154aafe7a8b19807d59 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@usody.com>
Date: Tue, 5 Mar 2024 10:49:14 +0100
Subject: [PATCH] test user credentials

---
 idhub/tests/test_credentials.py | 80 +++++++++++++++++++++++++--------
 1 file changed, 62 insertions(+), 18 deletions(-)

diff --git a/idhub/tests/test_credentials.py b/idhub/tests/test_credentials.py
index c6a76ab..63c5514 100644
--- a/idhub/tests/test_credentials.py
+++ b/idhub/tests/test_credentials.py
@@ -2,24 +2,32 @@ import os
 import json
 
 from pathlib import Path
-from django.test import TestCase, RequestFactory
+from django.test import TestCase
 from django.core.files.uploadedfile import SimpleUploadedFile
 from django.core.cache import cache
 from django.urls import reverse
 from django.conf import settings
 
 from idhub_auth.models import User
-from idhub.models import DID, Schemas
+from idhub.models import DID, Schemas, VerificableCredential
 from oidc4vp.models import Organization
-from idhub.admin.views import PeopleListView
 
 
+
+PILOTS = [
+    "course-credential",
+    "federation-membership",
+    "membership-card",
+    "financial-vulnerability",
+    "e-operator-claim",
+]
+
 class AdminDashboardViewTest(TestCase):
 
     def setUp(self):
         cache.set("KEY_DIDS", '1234', None)
         self.user = User.objects.create_user(
-            email='normaluser@example.org',
+            email='user1@example.org',
             password='testpass12',
         )
         self.user.accept_gdpr=True
@@ -32,12 +40,19 @@ class AdminDashboardViewTest(TestCase):
         self.admin_user.save()
         self.org = Organization.objects.create(name="testserver", main=True)
 
-        self.client.login(email='adminuser@example.org',
-                          password='adminpass12')
         settings.DOMAIN = self.org.name
         settings.ENABLE_EMAIL = False
+
+        self.admin_login()
         self.create_schemas()
 
+    def user_login(self):
+        self.client.login(email='user1@example.org',
+                          password='testpass12')
+    def admin_login(self):
+        self.client.login(email='adminuser@example.org',
+                          password='adminpass12')
+        
     def create_did(self, label="Default"):
         did = DID.objects.create(label=label, type=DID.Types.WEB.value)
         did.set_did()
@@ -51,18 +66,15 @@ class AdminDashboardViewTest(TestCase):
                 continue
             self._create_schemas(x)
 
+        s = Schemas.objects
+        for p in PILOTS:
+            f = "{}.json".format(p)
+            assert s.filter(file_schema = f).exists()
+
     def _create_schemas(self, file_name):
         data = self.open_file(file_name)
-        try:
-            ldata = json.loads(data)
-            assert credtools.validate_schema(ldata)
-            dname = ldata.get('name')
-            title = ldata.get('title')
-            assert dname
-            assert title
-        except Exception:
-            title = ''
-            _name = ''
+        ldata = json.loads(data)
+        title = ldata.get('title')
 
         _name = json.dumps(ldata.get('name', ''))
         _description = json.dumps(ldata.get('description', ''))
@@ -94,8 +106,7 @@ class AdminDashboardViewTest(TestCase):
         response = self.client.get(response.url)
         assert "DID created successfully" in response.content.decode('utf-8')
 
-    def test_upload_data_membership(self):
-        fileschema = "membership-card"
+    def _upload_data_membership(self, fileschema):
         did = self.create_did()
         schema = Schemas.objects.get(file_schema__contains=fileschema)
         url = reverse('idhub:admin_import_add')
@@ -122,3 +133,36 @@ class AdminDashboardViewTest(TestCase):
         assert response.url == reverse('idhub:admin_import')
         response = self.client.get(response.url)
         assert "successfully" in response.content.decode('utf-8')
+
+    def test_upload_data(self):
+        for p in PILOTS:
+            self._upload_data_membership(p)
+
+    def test_user_require_credentail(self):
+        p = PILOTS[0]
+        self._upload_data_membership(p)
+        schema = Schemas.objects.get(file_schema__contains=p)
+        cred = VerificableCredential.objects.get(schema=schema)
+        url = reverse('idhub:user_credentials_request')
+        did = DID.objects.create(
+            label="Default",
+            type=DID.Types.WEB.value,
+            user=self.user
+        )
+
+        self.user_login()
+        response = self.client.get(url)
+        assert response.status_code == 200
+
+        data = {
+            "did": did.did,
+            "credential": cred.id,
+        }
+        
+        response = self.client.post(url, data=data)
+        import pdb; pdb.set_trace()
+
+        assert response.status_code == 302
+        assert response.url == reverse('idhub:user_credentials')
+        response = self.client.get(response.url)
+        assert "successfully" in response.content.decode('utf-8')