energylink/pkg/plugins/dashboard_importer_test.go

package plugins

import (
	"testing"

	"github.com/grafana/grafana/pkg/bus"
	"github.com/grafana/grafana/pkg/components/dynmap"
	m "github.com/grafana/grafana/pkg/models"
	"github.com/grafana/grafana/pkg/setting"
	. "github.com/smartystreets/goconvey/convey"
	"gopkg.in/ini.v1"
)

func TestDashboardImport(t *testing.T) {

	Convey("When importing plugin dashboard", t, func() {
		setting.Cfg = ini.Empty()
		sec, _ := setting.Cfg.NewSection("plugin.test-app")
		sec.NewKey("path", "../../tests/test-app")
		err := Init()

		So(err, ShouldBeNil)

		var importedDash *m.Dashboard
		bus.AddHandler("test", func(cmd *m.SaveDashboardCommand) error {
			importedDash = cmd.GetDashboardModel()
			cmd.Result = importedDash
			return nil
		})

		cmd := ImportDashboardCommand{
			PluginId: "test-app",
			Path:     "dashboards/connections.json",
			OrgId:    1,
			UserId:   1,
			Inputs: []ImportDashboardInput{
				{Name: "*", Type: "datasource"},
			},
		}

		err = ImportDashboard(&cmd)
		So(err, ShouldBeNil)

		Convey("should install dashboard", func() {
			So(importedDash, ShouldNotBeNil)

			dashData := dynmap.NewFromMap(importedDash.Data)
			So(dashData.String(), ShouldEqual, "")

			rows := importedDash.Data["rows"].([]interface{})
			row1 := rows[0].(map[string]interface{})
			panels := row1["panels"].([]interface{})
			panel := panels[0].(map[string]interface{})

			So(panel["datasource"], ShouldEqual, "graphite")
			So(importedDash.Data["__inputs"], ShouldBeNil)
		})
	})

	Convey("When evaling dashboard template", t, func() {
		template, _ := dynmap.NewObjectFromBytes([]byte(`{
      "__inputs": {
        "graphite": {
          "type": "datasource"
        }
      },
      "test": {
        "prop": "$__graphite"
      }
    }`))

		evaluator := &DashTemplateEvaluator{
			template: template,
			inputs: []ImportDashboardInput{
				{Name: "*", Type: "datasource", Value: "my-server"},
			},
		}

		res, err := evaluator.Eval()
		So(err, ShouldBeNil)

		Convey("should render template", func() {
			So(res.MustGetString("test.prop", ""), ShouldEqual, "my-server")
		})

		Convey("should not include inputs in output", func() {
			_, err := res.GetObject("__inputs")
			So(err, ShouldNotBeNil)
		})

	})

}