move cache to datasource_cache

2020-06-04 12:10:34 +03:00
parent 89dfb1e228
commit 95c8e9cfa6
6 changed files with 61 additions and 30 deletions
--- a/pkg/datasource/datasource_cache_test.go
+++ b/pkg/datasource/datasource_cache_test.go
@@ -0,0 +1,100 @@
+package datasource
+
+import (
+	"testing"
+
+	"github.com/grafana/grafana-plugin-sdk-go/backend"
+	"gotest.tools/assert"
+)
+
+func TestHashDatasourceInfo(t *testing.T) {
+	tests := []struct {
+		name         string
+		dsInfoBefore *backend.DataSourceInstanceSettings
+		dsInfoAfter  *backend.DataSourceInstanceSettings
+		equal        bool
+	}{
+		{
+			name: "Same datasource settings",
+			dsInfoBefore: &backend.DataSourceInstanceSettings{
+				ID:       1,
+				Name:     "Zabbix",
+				URL:      "https://localhost:3306/zabbix/api_jsonrpc.php",
+				JSONData: []byte("{}"),
+				DecryptedSecureJSONData: map[string]string{
+					"username": "grafanaZabbixUser",
+					"password": "$uper$ecr3t!!!",
+				},
+			},
+			dsInfoAfter: &backend.DataSourceInstanceSettings{
+				ID:       1,
+				Name:     "Zabbix",
+				URL:      "https://localhost:3306/zabbix/api_jsonrpc.php",
+				JSONData: []byte("{}"),
+				DecryptedSecureJSONData: map[string]string{
+					"username": "grafanaZabbixUser",
+					"password": "$uper$ecr3t!!!",
+				},
+			},
+			equal: true,
+		},
+		{
+			name: "Password changed",
+			dsInfoBefore: &backend.DataSourceInstanceSettings{
+				ID:       1,
+				Name:     "Zabbix",
+				URL:      "https://localhost:3306/zabbix/api_jsonrpc.php",
+				JSONData: []byte("{}"),
+				DecryptedSecureJSONData: map[string]string{
+					"username": "grafanaZabbixUser",
+					"password": "$uper$ecr3t!!!",
+				},
+			},
+			dsInfoAfter: &backend.DataSourceInstanceSettings{
+				ID:       1,
+				Name:     "Zabbix",
+				URL:      "https://localhost:3306/zabbix/api_jsonrpc.php",
+				JSONData: []byte("{}"),
+				DecryptedSecureJSONData: map[string]string{
+					"username": "grafanaZabbixUser",
+					"password": "new$uper$ecr3t!!!",
+				},
+			},
+			equal: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			before := HashDatasourceInfo(tt.dsInfoBefore)
+			after := HashDatasourceInfo(tt.dsInfoAfter)
+			got := before == after
+			assert.Equal(t, tt.equal, got)
+		})
+	}
+}
+
+func BenchmarkHashDatasourceInfo(b *testing.B) {
+	benches := []struct {
+		name   string
+		dsInfo *backend.DataSourceInstanceSettings
+	}{
+		{
+			"Normal Datasource Info",
+			&backend.DataSourceInstanceSettings{
+				ID:       4,
+				Name:     "MyZabbixDatasource",
+				URL:      "https://localhost:3306/zabbix/api_jsonrpc.php",
+				JSONData: []byte(`{ "addThresholds": true, "disableReadOnlyUsersAck": true }`),
+				DecryptedSecureJSONData: map[string]string{
+					"username": "grafanaZabbixUser",
+					"password": "$uper$ecr3t!!!",
+				},
+			},
+		},
+	}
+	for _, bt := range benches {
+		b.Run(bt.name, func(b *testing.B) {
+			HashDatasourceInfo(bt.dsInfo)
+		})
+	}
+}