client_test.go

package sdk

import (
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestConnect(t *testing.T) {
	t.Run("a new client is created", func(t *testing.T) {
		client := Connect("https://api.vulncheck.com", "vulncheck_token")
		if client == nil {
			t.Error("client is nil")
		}

		assert.NotNil(t, client)
	})
}

func TestSetAuthHeader(t *testing.T) {
	t.Run("auth header is set", func(t *testing.T) {
		client := Connect("https://api.vulncheck.com", "vulncheck_token")
		req := httptest.NewRequest("GET", "/index", nil)
		client.SetAuthHeader(req)

		assert.Equal(t, "Bearer vulncheck_token", req.Header.Get("Authorization"))
	})
}

func TestGetUrl(t *testing.T) {
	t.Run("url is set", func(t *testing.T) {
		client := Connect("https://api.vulncheck.com", "vulncheck_token")
		assert.Equal(t, "https://api.vulncheck.com", client.GetUrl())
	})
}