package gkeheterogeneous

import (
	"testing"

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

func TestNewGetComputeClassRequest(t *testing.T) {
	t.Parallel()

	req := newGetComputeClassRequest("12345", "n2-standard-8", "n2d-standard-8")
	require.Equal(t, "12345", req.GetClusterId())
	require.Equal(t, "n2-standard-8", req.GetBaseInstanceType())
	require.Equal(t, "n2d-standard-8", req.GetHeterogeneousInstanceType())
	require.Empty(t, req.GetName())
}

func TestNewGetComputeClassRequestWithSameTypes(t *testing.T) {
	t.Parallel()

	req := newGetComputeClassRequest("99999", "n2-custom-8-16384", "n2d-custom-8-16384")
	require.Equal(t, "99999", req.GetClusterId())
	require.Equal(t, "n2-custom-8-16384", req.GetBaseInstanceType())
	require.Equal(t, "n2d-custom-8-16384", req.GetHeterogeneousInstanceType())
}