Improve testing for Unmarshal AuthProperties

Author: calvinmclean

model/auth_test.go

@@ -95,29 +95,68 @@ func TestAuthDefinitionsStructLevelValidation(t *testing.T) {
 }
 
 func TestUnmarshalJSONMultipleAuthProperties(t *testing.T) {
-	a1JSON := `{
+	t.Run("BearerAuthProperties", func(t *testing.T) {
+		a1JSON := `{
+		"name": "a1",
+		"scheme": "bearer",
+		"properties": {
+			"token": "token1"
+		}
+	}`
+		a2JSON := `{
+		"name": "a2",
+		"scheme": "bearer",
+		"properties": {
+			"token": "token2"
+		}
+	}`
+
+		var a1 Auth
+		err := json.Unmarshal([]byte(a1JSON), &a1)
+		assert.NoError(t, err)
+
+		var a2 Auth
+		err = json.Unmarshal([]byte(a2JSON), &a2)
+		assert.NoError(t, err)
+
+		a1Properties := a1.Properties.(*BearerAuthProperties)
+		a2Properties := a2.Properties.(*BearerAuthProperties)
+
+		assert.Equal(t, "token1", a1Properties.Token)
+		assert.Equal(t, "token2", a2Properties.Token)
+		assert.NotEqual(t, a1Properties, a2Properties)
+	})
+
+	t.Run("OAuth2AuthProperties", func(t *testing.T) {
+		a1JSON := `{
 	"name": "a1",
-	"scheme": "bearer",
+	"scheme": "oauth2",
 	"properties": {
-		"token": "token1"
+		"clientSecret": "secret1"
 	}
 }`
-	a2JSON := `{
+
+		a2JSON := `{
 	"name": "a2",
-	"scheme": "bearer",
+	"scheme": "oauth2",
 	"properties": {
-		"token": "token2"
+		"clientSecret": "secret2"
 	}
 }`
 
-	var a1 Auth
-	err := json.Unmarshal([]byte(a1JSON), &a1)
-	assert.NoError(t, err)
+		var a1 Auth
+		err := json.Unmarshal([]byte(a1JSON), &a1)
+		assert.NoError(t, err)
+
+		var a2 Auth
+		err = json.Unmarshal([]byte(a2JSON), &a2)
+		assert.NoError(t, err)
 
-	var a2 Auth
-	err = json.Unmarshal([]byte(a2JSON), &a2)
-	assert.NoError(t, err)
+		a1Properties := a1.Properties.(*OAuth2AuthProperties)
+		a2Properties := a2.Properties.(*OAuth2AuthProperties)
 
-	assert.Equal(t, "token1", a1.Properties.(*BearerAuthProperties).Token)
-	assert.Equal(t, "token2", a2.Properties.(*BearerAuthProperties).Token)
+		assert.Equal(t, "secret1", a1Properties.ClientSecret)
+		assert.Equal(t, "secret2", a2Properties.ClientSecret)
+		assert.NotEqual(t, a1Properties, a2Properties)
+	})
 }