Skip to content

Commit cb51987

Browse files
committed
Fixed AUT issues
1 parent 519dc8d commit cb51987

File tree

2 files changed

+25
-9
lines changed

2 files changed

+25
-9
lines changed

src/sonic-py-common/tests/mock_swsscommon.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,15 @@ def connect(self, db):
1414

1515
def get(self, db, table, field):
1616
return self.data.get(field, "N/A")
17+
18+
19+
class ConfigDBConnector:
20+
def __init__(self):
21+
self.CONFIG_DB = 'CONFIG_DB'
22+
self.data = {"key_encrypt": "True"}
23+
24+
def connect(self, db):
25+
pass
26+
27+
def get(self, db, table, field):
28+
return self.data.get(field, "N/A")
Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,22 @@
11
from sonic_py_common.security_cipher import security_cipher
2+
from .mock_swsscommon import ConfigDBConnector
23

3-
temp = security_cipher()
44

55
class TestSecurityCipher(object):
66
def test_passkey_encryption(self):
7-
encrypt, err = temp.encrypt_passkey("TACPLUS", "passkey1", "TEST1")
8-
assert encrypt != "passkey1"
9-
assert err == None
7+
with mock.patch("sonic_py_common.security_cipher.ConfigDBConnector", new=ConfigDBConnector):
8+
temp = security_cipher()
9+
encrypt, err = temp.encrypt_passkey("TACPLUS", "passkey1", "TEST1")
10+
assert encrypt != "passkey1"
11+
assert err == None
1012

1113
def test_passkey_decryption(self):
12-
encrypt, err = temp.encrypt_passkey("RADIUS", "passkey2", "TEST2")
13-
assert err == None
14-
decrypt, err = temp.decrypt_passkey("RADIUS", encrypt)
15-
assert err == None
16-
assert decrypt == "passkey2"
14+
with mock.patch("sonic_py_common.security_cipher.ConfigDBConnector", new=ConfigDBConnector):
15+
temp = security_cipher()
16+
encrypt, err = temp.encrypt_passkey("RADIUS", "passkey2", "TEST2")
17+
assert err == None
18+
decrypt, err = temp.decrypt_passkey("RADIUS", encrypt)
19+
assert err == None
20+
assert decrypt == "passkey2"
1721

1822

0 commit comments

Comments
 (0)