Add more tests for the Key model. Remove the key_id field and made it a property derived from the fingerprint.

This commit is contained in:
Roberto Rosario
2016-03-23 11:58:08 -04:00
parent f82b2000c3
commit cd077cb076
10 changed files with 65 additions and 65 deletions

View File

@@ -4,7 +4,10 @@ from django.test import TestCase
from ..models import Key
from .literals import TEST_KEY_DATA, TEST_KEY_FINGERPRINT, TEST_KEY_ID
from .literals import (
TEST_KEY_DATA, TEST_KEY_FINGERPRINT, TEST_SEARCH_FINGERPRINT,
TEST_SEARCH_UID
)
class KeyTestCase(TestCase):
@@ -12,5 +15,19 @@ class KeyTestCase(TestCase):
# Creating a Key instance is analogous to importing a key
key = Key.objects.create(key_data=TEST_KEY_DATA)
self.assertEqual(key.key_id, TEST_KEY_ID)
self.assertEqual(key.fingerprint, TEST_KEY_FINGERPRINT)
def test_key_search(self):
search_results = Key.objects.search(query=TEST_SEARCH_UID)
self.assertTrue(
TEST_SEARCH_FINGERPRINT in [
key_stub.fingerprint for key_stub in search_results
]
)
def test_key_receive(self):
Key.objects.receive_key(key_id=TEST_SEARCH_FINGERPRINT)
self.assertEqual(Key.objects.all().count(), 1)
self.assertEqual(Key.objects.first().fingerprint, TEST_SEARCH_FINGERPRINT)