func TestDataSourceProxy(t *testing.T) { Convey("When getting graphite datasource proxy", t, func() { clearCache() ds := m.DataSource{Url: "htttp://graphite:8080", Type: m.DS_GRAPHITE} targetUrl, err := url.Parse(ds.Url) proxy := NewReverseProxy(&ds, "/render", targetUrl) proxy.Transport, err = DataProxyTransport(&ds) So(err, ShouldBeNil) transport, ok := proxy.Transport.(*http.Transport) So(ok, ShouldBeTrue) So(transport.TLSClientConfig.InsecureSkipVerify, ShouldBeTrue) requestUrl, _ := url.Parse("http://grafana.com/sub") req := http.Request{URL: requestUrl} proxy.Director(&req) Convey("Can translate request url and path", func() { So(req.URL.Host, ShouldEqual, "graphite:8080") So(req.URL.Path, ShouldEqual, "/render") }) }) Convey("When getting influxdb datasource proxy", t, func() { clearCache() ds := m.DataSource{ Type: m.DS_INFLUXDB_08, Url: "http://influxdb:8083", Database: "site", User: "******", Password: "******", } targetUrl, _ := url.Parse(ds.Url) proxy := NewReverseProxy(&ds, "", targetUrl) requestUrl, _ := url.Parse("http://grafana.com/sub") req := http.Request{URL: requestUrl} proxy.Director(&req) Convey("Should add db to url", func() { So(req.URL.Path, ShouldEqual, "/db/site/") }) Convey("Should add username and password", func() { queryVals := req.URL.Query() So(queryVals["u"][0], ShouldEqual, "user") So(queryVals["p"][0], ShouldEqual, "password") }) }) Convey("When caching a datasource proxy", t, func() { clearCache() ds := m.DataSource{ Id: 1, Url: "http://k8s:8001", Type: "Kubernetes", } t1, err := DataProxyTransport(&ds) So(err, ShouldBeNil) t2, err := DataProxyTransport(&ds) So(err, ShouldBeNil) Convey("Should be using the cached proxy", func() { So(t2, ShouldEqual, t1) }) }) Convey("When getting kubernetes datasource proxy", t, func() { clearCache() setting.SecretKey = "password" json := simplejson.New() json.Set("tlsAuth", true) json.Set("tlsAuthWithCACert", true) t := time.Now() ds := m.DataSource{ Url: "http://k8s:8001", Type: "Kubernetes", Updated: t.Add(-2 * time.Minute), } transport, err := DataProxyTransport(&ds) So(err, ShouldBeNil) Convey("Should have no cert", func() { So(transport.TLSClientConfig.InsecureSkipVerify, ShouldEqual, true) }) ds.JsonData = json ds.SecureJsonData = map[string][]byte{ "tlsCACert": util.Encrypt([]byte(caCert), "password"), "tlsClientCert": util.Encrypt([]byte(clientCert), "password"), "tlsClientKey": util.Encrypt([]byte(clientKey), "password"), } ds.Updated = t.Add(-1 * time.Minute) transport, err = DataProxyTransport(&ds) So(err, ShouldBeNil) Convey("Should add cert", func() { So(transport.TLSClientConfig.InsecureSkipVerify, ShouldEqual, false) So(len(transport.TLSClientConfig.Certificates), ShouldEqual, 1) }) ds.JsonData = nil ds.SecureJsonData = map[string][]byte{} ds.Updated = t transport, err = DataProxyTransport(&ds) So(err, ShouldBeNil) Convey("Should remove cert", func() { So(transport.TLSClientConfig.InsecureSkipVerify, ShouldEqual, true) So(len(transport.TLSClientConfig.Certificates), ShouldEqual, 0) }) }) }