fix tests

uber · Jul 13, 2017 · aeb1b8f · aeb1b8f
1 parent efdcfc3
commit aeb1b8f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/cmd/query/app/builder/elastic_search.go b/cmd/query/app/builder/elastic_search.go
@@ -62,5 +62,9 @@ func (e *esBuilder) NewSpanReader() (spanstore.Reader, error) {
 }
 
 func (e *esBuilder) NewDependencyReader() (dependencystore.Reader, error) {
-	return esDependencyStore.NewDependencyStore(), nil
+	client, err := e.getClient()
+	if err != nil {
+		return nil, err
+	}
+	return esDependencyStore.NewDependencyStore(client, e.logger), nil
 }
diff --git a/cmd/query/app/builder/elastic_search_test.go b/cmd/query/app/builder/elastic_search_test.go
@@ -53,6 +53,9 @@ func TestESBuilderFailure(t *testing.T) {
 		spanReader, err := esBuilder.NewSpanReader()
 		assert.Error(t, err)
 		assert.Nil(t, spanReader)
+		dependencyReader, err := esBuilder.NewDependencyReader()
+		assert.Error(t, err)
+		assert.Nil(t, dependencyReader)
 	})
 }