|
1 | 1 | from sonic_py_common.security_cipher import security_cipher |
| 2 | +from .mock_swsscommon import ConfigDBConnector |
2 | 3 |
|
3 | | -temp = security_cipher() |
4 | 4 |
|
5 | 5 | class TestSecurityCipher(object): |
6 | 6 | 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 |
10 | 12 |
|
11 | 13 | 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" |
17 | 21 |
|
18 | 22 |
|
0 commit comments