mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
33 lines
904 B
Python
33 lines
904 B
Python
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,
|
|
)
|
|
self.assertFalse(form.is_valid())
|