energylink/pkg/cmd/grafana-cli/commands/commandstest/fake_api_client.go

package commandstest

import (
	"github.com/grafana/grafana/pkg/cmd/grafana-cli/models"
)

type FakeGrafanaComClient struct {
	GetPluginFunc      func(pluginId, repoUrl string) (models.Plugin, error)
	DownloadFileFunc   func(pluginName, filePath, url string, checksum string) (content []byte, err error)
	ListAllPluginsFunc func(repoUrl string) (models.PluginRepo, error)
}

func (client *FakeGrafanaComClient) GetPlugin(pluginId, repoUrl string) (models.Plugin, error) {
	if client.GetPluginFunc != nil {
		return client.GetPluginFunc(pluginId, repoUrl)
	}

	return models.Plugin{}, nil
}

func (client *FakeGrafanaComClient) DownloadFile(pluginName, filePath, url string, checksum string) (content []byte, err error) {
	if client.DownloadFileFunc != nil {
		return client.DownloadFileFunc(pluginName, filePath, url, checksum)
	}

	return make([]byte, 0), nil
}

func (client *FakeGrafanaComClient) ListAllPlugins(repoUrl string) (models.PluginRepo, error) {
	if client.ListAllPluginsFunc != nil {
		return client.ListAllPluginsFunc(repoUrl)
	}
	return models.PluginRepo{}, nil
}