FluxCD MCP Server

// Copyright 2024 Stefan Prodan.
// SPDX-License-Identifier: AGPL-3.0

package entitlement

import (
	"testing"

	. "github.com/onsi/gomega"
)

func TestNewClient_DefaultVendor(t *testing.T) {
	g := NewWithT(t)

	// Unset the environment variable
	t.Setenv(MarketplaceTypeEnvKey, "")

	client, err := NewClient()
	g.Expect(err).ToNot(HaveOccurred())
	g.Expect(client).ToNot(BeNil())
	g.Expect(client.GetVendor()).To(Equal(DefaultVendor))
}

func TestNewClient_UnsupportedVendor(t *testing.T) {
	g := NewWithT(t)

	// Set the environment variable to an unsupported value
	t.Setenv(MarketplaceTypeEnvKey, "unsupported")

	client, err := NewClient()
	g.Expect(err).To(HaveOccurred())
	g.Expect(client).To(BeNil())
}