Start adding tests for the Snyk module

lib/snyk/self.go

@@ -15,11 +15,7 @@ const experimentalVersion = "2023-04-28~experimental"
 type selfDocument struct {
 	Data struct {
 		Attributes users.User `json:"attributes,omitempty"`
-		ID         string     `json:"id,omitempty"`
-		Type       string     `json:"type,omitempty"`
 	}
-	Jsonapi users.JsonApi `json:"jsonapi,omitempty"`
-	Links   users.Links   `json:"links,omitempty"`
 }
 
 func getSnykOrg(auth *securityprovider.SecurityProviderApiKey) (*uuid.UUID, error) {

lib/snyk/self_test.go

@@ -0,0 +1,38 @@
+package snyk
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/deepmap/oapi-codegen/pkg/securityprovider"
+	"github.com/google/uuid"
+	"github.com/jarcoal/httpmock"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetSnykOrg(t *testing.T) {
+	expectedOrg := uuid.New()
+	auth, _ := securityprovider.NewSecurityProviderApiKey("header", "name", "value")
+
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+	httpmock.RegisterResponder("GET", "https://api.snyk.io/rest/self",
+		func(req *http.Request) (*http.Response, error) {
+			jsonBody := `{
+				"data": {
+					"attributes": {
+						"default_org_context": "` + expectedOrg.String() + `"
+					}
+				}
+			}`
+			resp := httpmock.NewStringResponse(200, jsonBody)
+			resp.Header.Set("Content-Type", "application/json")
+			return resp, nil
+		},
+	)
+
+	actualOrg, err := getSnykOrg(auth)
+	assert.NoError(t, err)
+	assert.Equal(t, expectedOrg, *actualOrg)
+}
+