def keyset_to_json(stub: testing_api_pb2_grpc.KeysetStub, keyset: bytes) -> Text: request = testing_api_pb2.KeysetToJsonRequest(keyset=keyset) response = stub.ToJson(request) if response.err: raise tink.TinkError(response.err) return response.json_keyset
def test_generate_to_from_json(self): keyset_servicer = services.KeysetServicer() template = aead.aead_key_templates.AES128_GCM.SerializeToString() gen_request = testing_api_pb2.KeysetGenerateRequest(template=template) gen_response = keyset_servicer.Generate(gen_request, self._ctx) self.assertEqual(gen_response.WhichOneof('result'), 'keyset') keyset = gen_response.keyset tojson_request = testing_api_pb2.KeysetToJsonRequest(keyset=keyset) tojson_response = keyset_servicer.ToJson(tojson_request, self._ctx) self.assertEqual(tojson_response.WhichOneof('result'), 'json_keyset') json_keyset = tojson_response.json_keyset fromjson_request = testing_api_pb2.KeysetFromJsonRequest( json_keyset=json_keyset) fromjson_response = keyset_servicer.FromJson(fromjson_request, self._ctx) self.assertEqual(fromjson_response.WhichOneof('result'), 'keyset') self.assertEqual(fromjson_response.keyset, keyset)
def test_to_json_fail(self): keyset_servicer = services.KeysetServicer() request = testing_api_pb2.KeysetToJsonRequest(keyset=b'bad keyset') response = keyset_servicer.ToJson(request, self._ctx) self.assertEqual(response.WhichOneof('result'), 'err') self.assertNotEmpty(response.err)