Update default encryption key (temporalio#24)

lorensr · web-flow · commit 1a414e2d4f94 · 2022-11-23T21:20:21.000-06:00
diff --git a/encryption/codec.py b/encryption/codec.py
@@ -6,8 +6,8 @@
 from temporalio.api.common.v1 import Payload
 from temporalio.converter import PayloadCodec
 
-default_key = base64.b64decode(b"MkUb3RVdHQuOTedqETZW7ra2GkZqpBRmYWRACUospMc=")
-default_key_id = "my-key"
+default_key = b"test-key-test-key-test-key-test!"
+default_key_id = "test-key-id"
 
 
 class EncryptionCodec(PayloadCodec):
@@ -43,7 +43,9 @@ async def decode(self, payloads: Iterable[Payload]) -> List[Payload]:
             # Confirm our key ID is the same
             key_id = p.metadata.get("encryption-key-id", b"").decode()
             if key_id != self.key_id:
-                raise ValueError(f"Unrecognized key ID {key_id}")
+                raise ValueError(
+                    f"Unrecognized key ID {key_id}. Current key ID is {self.key_id}."
+                )
             # Decrypt and append
             ret.append(Payload.FromString(self.decrypt(p.data)))
         return ret
diff --git a/encryption/starter.py b/encryption/starter.py
@@ -23,7 +23,7 @@ async def main():
         GreetingWorkflow.run,
         "Temporal",
         id=f"encryption-workflow-id",
-        task_queue="encryption-task-queue",
+        task_queue="encryption",
     )
     print(f"Workflow result: {result}")
 
diff --git a/encryption/worker.py b/encryption/worker.py
@@ -9,7 +9,7 @@
 from encryption.codec import EncryptionCodec
 
 
-@workflow.defn
+@workflow.defn(name="Workflow")
 class GreetingWorkflow:
     @workflow.run
     async def run(self, name: str) -> str:

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ async def main():`
`23`	`23`	`GreetingWorkflow.run,`
`24`	`24`	`"Temporal",`
`25`	`25`	`id=f"encryption-workflow-id",`
`26`		`- task_queue="encryption-task-queue",`
	`26`	`+ task_queue="encryption",`
`27`	`27`	`)`
`28`	`28`	`print(f"Workflow result: {result}")`
`29`	`29`