Skip to content

Commit 105c84f

Browse files
committed
Add test for Encrypted
1 parent 9961209 commit 105c84f

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

tests/test_types.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
from sqlorm.types import SQLType
2+
from sqlorm.types.encrypted import Encrypted
3+
import hashlib
24

35

46
def test_type():
@@ -16,4 +18,12 @@ def test_type():
1618
class T:
1719
__sqltype__ = "customtype"
1820

19-
assert SQLType.from_pytype(T).sql_type == "customtype"
21+
assert SQLType.from_pytype(T).sql_type == "customtype"
22+
23+
24+
def test_encrypted():
25+
key = hashlib.md5(b"key").digest()
26+
t = Encrypted(key)
27+
assert t.sql_type == "text"
28+
assert t.dumper("value") != "value"
29+
assert t.loader(t.dumper("value")) == "value"

0 commit comments

Comments
 (0)