make tests for OrgIdPublic work

grafana · Apr 10, 2018 · e72b7e4 · e72b7e4
1 parent b25d02c
commit e72b7e4
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/idx/cassandra/cassandra_test.go b/idx/cassandra/cassandra_test.go
@@ -122,6 +122,9 @@ func getMetricData(orgId, depth, count, interval int, prefix string) []*schema.M
 }
 
 func TestGetAddKey(t *testing.T) {
+	idx.OrgIdPublic = 100
+	defer func() { idx.OrgIdPublic = 0 }()
+
 	ix := New()
 	initForTests(ix)
 
@@ -256,6 +259,8 @@ func TestAddToWriteQueue(t *testing.T) {
 }
 
 func TestFind(t *testing.T) {
+	idx.OrgIdPublic = 100
+	defer func() { idx.OrgIdPublic = 0 }()
 	ix := New()
 	initForTests(ix)
 	for _, s := range getMetricData(idx.OrgIdPublic, 2, 5, 10, "metric.demo") {

diff --git a/idx/memory/memory_test.go b/idx/memory/memory_test.go
@@ -83,6 +83,9 @@ func TestGetAddKey(t *testing.T) {
 }
 
 func testGetAddKey(t *testing.T) {
+	idx.OrgIdPublic = 100
+	defer func() { idx.OrgIdPublic = 0 }()
+
 	ix := New()
 	ix.Init()
 
@@ -126,6 +129,9 @@ func TestFind(t *testing.T) {
 }
 
 func testFind(t *testing.T) {
+	idx.OrgIdPublic = 100
+	defer func() { idx.OrgIdPublic = 0 }()
+
 	ix := New()
 	ix.Init()
 	for _, s := range getMetricData(idx.OrgIdPublic, 2, 5, 10, "metric.demo", false) {
@@ -253,6 +259,9 @@ func TestDelete(t *testing.T) {
 }
 
 func testDelete(t *testing.T) {
+	idx.OrgIdPublic = 100
+	defer func() { idx.OrgIdPublic = 0 }()
+
 	ix := New()
 	ix.Init()
 
@@ -268,6 +277,9 @@ func testDelete(t *testing.T) {
 }
 
 func TestDeleteTagged(t *testing.T) {
+	idx.OrgIdPublic = 100
+	defer func() { idx.OrgIdPublic = 0 }()
+
 	ix := New()
 	ix.Init()