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)
