summaryrefslogtreecommitdiff
path: root/test/tpm_test/crypto_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/tpm_test/crypto_test.py')
-rw-r--r--test/tpm_test/crypto_test.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/test/tpm_test/crypto_test.py b/test/tpm_test/crypto_test.py
index 147b5f301b..7fa4ea6146 100644
--- a/test/tpm_test/crypto_test.py
+++ b/test/tpm_test/crypto_test.py
@@ -10,11 +10,9 @@ from __future__ import print_function
import struct
import xml.etree.ElementTree as ET
+import subcmd
import utils
-# Extension 'cryptography' subcommand codes:
-AES = 0
-
# Basic crypto operations
DECRYPT = 0
ENCRYPT = 1
@@ -99,7 +97,7 @@ class CryptoD(object):
self.submodes = submodes
SUPPORTED_MODES = {
- 'AES': CryptoD(AES, {
+ 'AES': CryptoD(subcmd.AES, {
'ECB': 0,
'CTR': 1,
'CBC': 2,