From 860c443fe99dbb932dfdc946620423aa4357109c Mon Sep 17 00:00:00 2001
From: dpcollins-google <40498610+dpcollins-google@users.noreply.github.com>
Date: Wed, 7 Oct 2020 13:26:05 -0400
Subject: [PATCH] fix: update pubsub_context to correctly modify proto map
 fields (#38)

---
 google/cloud/pubsublite/internal/wire/pubsub_context.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/google/cloud/pubsublite/internal/wire/pubsub_context.py b/google/cloud/pubsublite/internal/wire/pubsub_context.py
index c3964eaa..f64968f1 100644
--- a/google/cloud/pubsublite/internal/wire/pubsub_context.py
+++ b/google/cloud/pubsublite/internal/wire/pubsub_context.py
@@ -23,11 +23,11 @@ def _version() -> _Semver:
 def pubsub_context(framework: Optional[str] = None) -> Mapping[str, str]:
     """Construct the pubsub context mapping for the given framework."""
     context = struct_pb2.Struct()
-    context.fields["language"] = struct_pb2.Value(string_value="PYTHON")
+    context.fields["language"].string_value = "PYTHON"
     if framework:
-        context.fields["framework"] = struct_pb2.Value(string_value=framework)
+        context.fields["framework"].string_value = framework
     version = _version()
-    context.fields["major_version"] = struct_pb2.Value(number_value=version.major)
-    context.fields["minor_version"] = struct_pb2.Value(number_value=version.minor)
+    context.fields["major_version"].number_value = version.major
+    context.fields["minor_version"].number_value = version.minor
     encoded = b64encode(context.SerializeToString()).decode("utf-8")
     return {"x-goog-pubsub-context": encoded}