add more tests first key flow

This commit is contained in:
Cayo Puigdefabregas 2024-03-06 19:46:27 +01:00
parent df26696c99
commit 67c804d640
1 changed files with 70 additions and 2 deletions

View File

@ -22,7 +22,59 @@ PILOTS = [
"e-operator-claim",
]
class AdminDashboardViewTest(TestCase):
class KeyFirstTimeTest(TestCase):
def setUp(self):
cache.set("KEY_DIDS", '')
self.user = User.objects.create_user(
email='user1@example.org',
password='testpass12',
)
self.admin_user = User.objects.create_superuser(
email='adminuser@example.org',
password='adminpass12')
self.org = Organization.objects.create(name="testserver", main=True)
settings.DOMAIN = self.org.name
settings.ENABLE_EMAIL = False
def set_cache(self):
cache.set("KEY_DIDS", '1234', None)
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 test_user_without_key(self):
cache.set("KEY_DIDS", '')
self.user_login()
response = self.client.get(reverse('idhub:user_dashboard'))
self.assertEqual(response.status_code, 302)
self.assertEqual(response.url, reverse('idhub:user_waiting'))
def test_admin_without_key(self):
cache.set("KEY_DIDS", '')
self.admin_login()
response = self.client.get(reverse('idhub:admin_dashboard'))
self.assertEqual(response.status_code, 302)
self.assertEqual(response.url, reverse('idhub:encryption_key'))
def test_admin_addfirst_key(self):
self.admin_login()
response = self.client.get(reverse('idhub:encryption_key'))
self.assertEqual(response.status_code, 200)
data = {
"key": 1
}
response = self.client.post(reverse('idhub:encryption_key'), data=data)
self.assertEqual(response.status_code, 302)
self.assertEqual(response.url, reverse('idhub:admin_dashboard'))
cache.set("KEY_DIDS", '')
class CredentialsViewTest(TestCase):
def setUp(self):
cache.set("KEY_DIDS", '1234', None)
@ -99,7 +151,8 @@ class AdminDashboardViewTest(TestCase):
return data
def test_create_did_web(self):
def test_admin_create_did_key(self):
self.admin_login()
url = reverse('idhub:admin_dids_new')
data = {"label": "Default", "type": DID.Types.KEY.value}
response = self.client.get(url)
@ -112,6 +165,20 @@ class AdminDashboardViewTest(TestCase):
response = self.client.get(response.url)
self.assertIn("DID created successfully", response.content.decode('utf-8'))
def test_user_create_did_key(self):
self.user_login()
url = reverse('idhub:user_dids_new')
data = {"label": "Default", "type": DID.Types.KEY.value}
response = self.client.get(url)
self.assertEqual(response.status_code, 200)
response = self.client.post(url, data=data)
self.assertEqual(response.status_code, 302)
self.assertEqual(response.url, reverse('idhub:user_dids'))
response = self.client.get(response.url)
self.assertIn("DID created successfully", response.content.decode('utf-8'))
def _upload_data_membership(self, fileschema):
did = self.create_did()
schema = Schemas.objects.get(file_schema__contains=fileschema)
@ -141,6 +208,7 @@ class AdminDashboardViewTest(TestCase):
self.assertIn("successfully", response.content.decode('utf-8'))
def test_upload_data(self):
self.admin_login()
for p in PILOTS:
self._upload_data_membership(p)