cost-model/test/costmodel_test.go

package costmodel_test

// Mocks can be regenerated with something like GO111MODULE=on mockgen -destination ./test/mocks/mock_provider.go  -package mocks github.com/kubecost/cost-model/cloud Provider

import (
	"net/url"
	"testing"
	"time"

	"github.com/golang/mock/gomock"
	//costModel "github.com/kubecost/cost-model/costmodel"
	"github.com/kubecost/test/mocks"
	v1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	//testclient "k8s.io/client-go/kubernetes/fake"
	fakecontroller "k8s.io/client-go/tools/cache/testing"
)

func TestCostModel(t *testing.T) {
	ctrl := gomock.NewController(t)

	defer ctrl.Finish()

	u, _ := url.Parse("http://localhost:9003")
	cli := mocks.NewMockClient(ctrl)
	cli.EXPECT().URL(gomock.Any(), gomock.Any()).AnyTimes().Return(u)
	cli.EXPECT().Do(gomock.Any(), gomock.Any()).AnyTimes()

	//clientset := testclient.NewSimpleClientset()
	//provider := mocks.NewMockProvider(ctrl)

	fc := fakecontroller.NewFakeControllerSource()
	//cm := costModel.NewCostModel(fc)
	fc.Add(&v1.Pod{
		ObjectMeta: metav1.ObjectMeta{
			Name: "foo",
		},
	})
	time.Sleep(100 * time.Millisecond)
	/*
		c, err := cm.ComputeCostData(cli, clientset, provider, "1d", "", "")
		if err != nil {
			panic(err)
		}
		for _, costs := range c {
			assert.Equal(t, "foo", costs.PodName)
		}
	*/
}