|
|
@@ -0,0 +1,33 @@
|
|
|
+from django.test import TestCase
|
|
|
+from secrets.forms import UserKeyForm
|
|
|
+from secrets.models import UserKey
|
|
|
+from utilities.testing import create_test_user
|
|
|
+from .constants import PUBLIC_KEY, SSH_PUBLIC_KEY
|
|
|
+
|
|
|
+
|
|
|
+class UserKeyFormTestCase(TestCase):
|
|
|
+
|
|
|
+ def setUp(self):
|
|
|
+ user = create_test_user(
|
|
|
+ permissions=[
|
|
|
+ 'secrets.view_secretrole',
|
|
|
+ 'secrets.add_secretrole',
|
|
|
+ ]
|
|
|
+ )
|
|
|
+ self.userkey = UserKey(user=user)
|
|
|
+
|
|
|
+ def test_upload_rsakey(self):
|
|
|
+ form = UserKeyForm(
|
|
|
+ data={'public_key': PUBLIC_KEY},
|
|
|
+ instance=self.userkey,
|
|
|
+ )
|
|
|
+ self.assertTrue(form.is_valid())
|
|
|
+ self.assertTrue(form.save())
|
|
|
+
|
|
|
+ def test_upload_sshkey(self):
|
|
|
+ form = UserKeyForm(
|
|
|
+ data={'public_key': SSH_PUBLIC_KEY},
|
|
|
+ instance=self.userkey,
|
|
|
+ )
|
|
|
+ print(form.is_valid())
|
|
|
+ self.assertFalse(form.is_valid())
|