dashboard: add tests

circle/dashboard/tests/test_views.py

@@ -32,6 +32,7 @@ from ..views import VmRenewView
 from storage.models import Disk
 from storage.models import Disk
 from firewall.models import Vlan, Host, VlanGroup
 from firewall.models import Vlan, Host, VlanGroup
 from mock import Mock, patch
 from mock import Mock, patch
+from django_sshkey.models import UserKey
 
 
 import django.conf
 import django.conf
 settings = django.conf.settings.FIREWALL_SETTINGS
 settings = django.conf.settings.FIREWALL_SETTINGS
@@ -1959,3 +1960,61 @@ class VmListTest(LoginMixin, TestCase):
             's': "A:B:C:D:"
             's': "A:B:C:D:"
         })
         })
         self.assertEqual(200, resp.status_code)
         self.assertEqual(200, resp.status_code)
+
+
+class SshKeyTest(LoginMixin, TestCase):
+    def setUp(self):
+        self.u1 = User.objects.create(username='user1')
+        self.u1.set_password('password')
+        self.u1.save()
+        self.u2 = User.objects.create(username='user2')
+        self.u2.set_password('password')
+        self.u2.save()
+        self.k1 = UserKey(key='ssh-rsa AAAAB3NzaC1yc2EC asd', user=self.u1)
+        self.k1.save()
+
+    def tearDown(self):
+        super(SshKeyTest, self).tearDown()
+        self.k1.delete()
+        self.u1.delete()
+
+    def test_permitted_edit(self):
+        c = Client()
+        self.login(c, self.u1)
+
+        resp = c.post("/dashboard/sshkey/1/",
+                      {'key': 'ssh-rsa AAAAB3NzaC1yc2EC'})
+        self.assertEqual(UserKey.objects.get(id=1).user, self.u1)
+        self.assertEqual(200, resp.status_code)
+
+    def test_unpermitted_edit(self):
+        c = Client()
+        self.login(c, self.u2)
+
+        resp = c.post("/dashboard/sshkey/1/",
+                      {'key': 'ssh-rsa AAAAB3NzaC1yc2EC'})
+        self.assertEqual(UserKey.objects.get(id=1).user, self.u1)
+        self.assertEqual(403, resp.status_code)
+
+    def test_permitted_add(self):
+        c = Client()
+        self.login(c, self.u1)
+
+        resp = c.post("/dashboard/sshkey/create/",
+                      {'name': 'asd', 'key': 'ssh-rsa AAAAB3NzaC1yc2EC'})
+        self.assertEqual(UserKey.objects.get(id=2).user, self.u1)
+        self.assertEqual(302, resp.status_code)
+
+    def test_permitted_delete(self):
+        c = Client()
+        self.login(c, self.u1)
+
+        resp = c.post("/dashboard/sshkey/delete/1/")
+        self.assertEqual(302, resp.status_code)
+
+    def test_unpermitted_delete(self):
+        c = Client()
+        self.login(c, self.u2)
+
+        resp = c.post("/dashboard/sshkey/delete/1/")
+        self.assertEqual(403, resp.status_code)