1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
import unittest
from keyczar import keyczar
from django.db import connection
from django.conf import settings
from django.core.management import call_command
from django.db.models import loading
from django_extensions.tests.models import Secret
from django_extensions.db.fields.encrypted import EncryptedTextField, EncryptedCharField
class EncryptedFieldsTestCase(unittest.TestCase):
def __init__(self, *args, **kwargs):
self.crypt = keyczar.Crypter.Read(settings.ENCRYPTED_FIELD_KEYS_DIR)
super(EncryptedFieldsTestCase, self).__init__(*args, **kwargs)
def setUp(self):
self.old_installed_apps = settings.INSTALLED_APPS
settings.INSTALLED_APPS.append('django_extensions.tests')
loading.cache.loaded = False
call_command('syncdb', verbosity=0)
def tearDown(self):
settings.INSTALLED_APPS = self.old_installed_apps
def testCharFieldCreate(self):
test_val = "Test Secret"
secret = Secret.objects.create(name=test_val)
cursor = connection.cursor()
query = "SELECT name FROM %s WHERE id = %d" % (Secret._meta.db_table, secret.id)
cursor.execute(query)
db_val, = cursor.fetchone()
decrypted_val = self.crypt.Decrypt(db_val[len(EncryptedCharField.prefix):])
self.assertEqual(test_val, decrypted_val)
def testCharFieldRead(self):
test_val = "Test Secret"
secret = Secret.objects.create(name=test_val)
retrieved_secret = Secret.objects.get(id=secret.id)
self.assertEqual(test_val, retrieved_secret.name)
def testTextFieldCreate(self):
test_val = "Test Secret"
secret = Secret.objects.create(text=test_val)
cursor = connection.cursor()
query = "SELECT text FROM %s WHERE id = %d" % (Secret._meta.db_table, secret.id)
cursor.execute(query)
db_val, = cursor.fetchone()
decrypted_val = self.crypt.Decrypt(db_val[len(EncryptedCharField.prefix):])
self.assertEqual(test_val, decrypted_val)
def testTextFieldRead(self):
test_val = "Test Secret"
secret = Secret.objects.create(text=test_val)
retrieved_secret = Secret.objects.get(id=secret.id)
self.assertEqual(test_val, retrieved_secret.text)
|
