Changeset View
Changeset View
Standalone View
Standalone View
test/util/rpcauth-test.py
| Show All 11 Lines | |||||
| import sys | import sys | ||||
| import unittest | import unittest | ||||
| class TestRPCAuth(unittest.TestCase): | class TestRPCAuth(unittest.TestCase): | ||||
| def setUp(self): | def setUp(self): | ||||
| config = configparser.ConfigParser() | config = configparser.ConfigParser() | ||||
| config_path = os.path.abspath( | config_path = os.path.abspath( | ||||
| os.path.join(os.sep, os.path.abspath(os.path.dirname(__file__)), | os.path.join( | ||||
| "../config.ini")) | os.sep, os.path.abspath(os.path.dirname(__file__)), "../config.ini" | ||||
| ) | |||||
| ) | |||||
| with open(config_path, encoding="utf8") as config_file: | with open(config_path, encoding="utf8") as config_file: | ||||
| config.read_file(config_file) | config.read_file(config_file) | ||||
| sys.path.insert(0, os.path.dirname(config['environment']['RPCAUTH'])) | sys.path.insert(0, os.path.dirname(config["environment"]["RPCAUTH"])) | ||||
| self.rpcauth = importlib.import_module('rpcauth') | self.rpcauth = importlib.import_module("rpcauth") | ||||
| def test_generate_salt(self): | def test_generate_salt(self): | ||||
| for i in range(16, 32 + 1): | for i in range(16, 32 + 1): | ||||
| self.assertEqual(len(self.rpcauth.generate_salt(i)), i * 2) | self.assertEqual(len(self.rpcauth.generate_salt(i)), i * 2) | ||||
| def test_generate_password(self): | def test_generate_password(self): | ||||
| password = self.rpcauth.generate_password() | password = self.rpcauth.generate_password() | ||||
| expected_password = base64.urlsafe_b64encode( | expected_password = base64.urlsafe_b64encode( | ||||
| base64.urlsafe_b64decode(password)).decode('utf-8') | base64.urlsafe_b64decode(password) | ||||
| ).decode("utf-8") | |||||
| self.assertEqual(expected_password, password) | self.assertEqual(expected_password, password) | ||||
| def test_check_password_hmac(self): | def test_check_password_hmac(self): | ||||
| salt = self.rpcauth.generate_salt(16) | salt = self.rpcauth.generate_salt(16) | ||||
| password = self.rpcauth.generate_password() | password = self.rpcauth.generate_password() | ||||
| password_hmac = self.rpcauth.password_to_hmac(salt, password) | password_hmac = self.rpcauth.password_to_hmac(salt, password) | ||||
| m = hmac.new(bytearray(salt, 'utf-8'), | m = hmac.new(bytearray(salt, "utf-8"), bytearray(password, "utf-8"), "SHA256") | ||||
| bytearray(password, 'utf-8'), 'SHA256') | |||||
| expected_password_hmac = m.hexdigest() | expected_password_hmac = m.hexdigest() | ||||
| self.assertEqual(expected_password_hmac, password_hmac) | self.assertEqual(expected_password_hmac, password_hmac) | ||||
| if __name__ == '__main__': | if __name__ == "__main__": | ||||
| unittest.main() | unittest.main() | ||||