work

databricks · May 17, 2024 · ceac218 · ceac218
1 parent d8939e1
commit ceac218
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/useragent/user_agent_test.go b/useragent/user_agent_test.go
@@ -74,3 +74,11 @@ func TestFromContext_Custom(t *testing.T) {
 func TestDefaultsAreValid(t *testing.T) {
 	WithProduct(productName, productVersion)
 }
+
+func TestMultiplePartners(t *testing.T) {
+	WithPartner("partner1")
+	WithPartner("partner2")
+	userAgent := FromContext(context.Background())
+	assert.Contains(t, userAgent, "partner/partner1")
+	assert.Contains(t, userAgent, "partner/partner2")
+}