def test_encrypt_decrypt(self): raw = '900:123456789:DEPARTMENT OF DEFENSE,USER:33' token = VistaToken.encrypt(raw) self.assertEqual(88, len(token)) expected = TokenParams( site_id='900', fed_id='123456789', username='******', duz='33' ) params = VistaToken.decrypt(token) self.assertEqual(expected, params)
def _set_vista_params(self): import json with open("controllers/config.json") as cfg_file: data = json.load(cfg_file) self.host = data['vista']['host'] self.port = data['vista']['port'] token = data['vista']['token'] vparams = VistaToken.decrypt(token) if vparams.site_id != data['vista']['site_id']: raise RpcException('Invalid vparams') self.user = { "fed_id": vparams.fed_id, "user_name": vparams.username, "source_name": data['vista']['site_name'], "source_id": vparams.site_id, "uid": vparams.duz, "phone": 'No phone' } self.cps = data['control_points']