new: added option ContextOptionIdempotencyKey

PaloAltoNetworks · Oct 3, 2019 · 58719fc · 58719fc
1 parent 56bc2de
commit 58719fc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/options.go b/options.go
@@ -183,3 +183,10 @@ func ContextOptionRetryRatio(r int64) ContextOption {
 		c.retryRatio = r
 	}
 }
+
+// ContextOptionIdempotencyKey sets a custom idempotency key.
+func ContextOptionIdempotencyKey(key string) ContextOption {
+	return func(c *mcontext) {
+		c.idempotencyKey = key
+	}
+}
diff --git a/options_test.go b/options_test.go
@@ -140,4 +140,9 @@ func TestManipulate_ContextOption(t *testing.T) {
 		ContextOptionRetryRatio(42)(mctx.(*mcontext))
 		So(mctx.RetryRatio(), ShouldEqual, 42)
 	})
+
+	Convey("Calling ContextOptionIdempotencyKey should work", t, func() {
+		ContextOptionIdempotencyKey("42")(mctx.(*mcontext))
+		So(mctx.(*mcontext).idempotencyKey, ShouldEqual, "42")
+	})
 }