def test_getAuthenticateRequest(self): reqdata = AuthenticateRequestData( authenticateRequests=[{ 'keyHandle': 'a' }, { 'keyHandle': 'b' }]) response = SignResponse(keyHandle='b') self.assertEqual({'keyHandle': 'b'}, reqdata.getAuthenticateRequest(response)) self.assertTrue( isinstance(reqdata.getAuthenticateRequest(response), SignRequest))
def test_authenticateRequests(self): reqdata = AuthenticateRequestData(authenticateRequests=[{}, { 'a': 1 }, { 'a': 1, 'b': 2 }]) self.assertEqual([{}, { 'a': 1 }, { 'a': 1, 'b': 2 }], reqdata.authenticateRequests) self.assertTrue( isinstance(reqdata.authenticateRequests[0], SignRequest))
def start_authenticate(devices, challenge=None): sign_requests = [u2f_v2.start_authenticate(d, challenge or rand_bytes(32)) for d in devices] return AuthenticateRequestData(authenticateRequests=sign_requests)