From a6f289eefe54024a8fcc5d9b4d5c48941b8ffb50 Mon Sep 17 00:00:00 2001 From: Ruben Vargas Date: Wed, 9 Oct 2019 23:30:10 -0500 Subject: [PATCH] Updte reader tests, now expect valueCtx --- plugin/storage/es/spanstore/reader_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/plugin/storage/es/spanstore/reader_test.go b/plugin/storage/es/spanstore/reader_test.go index f2ec9bdd52ac..7f12d94cd870 100644 --- a/plugin/storage/es/spanstore/reader_test.go +++ b/plugin/storage/es/spanstore/reader_test.go @@ -803,7 +803,7 @@ func mockMultiSearchService(r *spanReaderTest) *mock.Call { multiSearchService.On("Index", mock.AnythingOfType("string"), mock.AnythingOfType("string"), mock.AnythingOfType("string"), mock.AnythingOfType("string"), mock.AnythingOfType("string")).Return(multiSearchService) r.client.On("MultiSearch").Return(multiSearchService) - return multiSearchService.On("Do", mock.AnythingOfType("*context.emptyCtx")) + return multiSearchService.On("Do", mock.AnythingOfType("*context.valueCtx")) } func mockArchiveMultiSearchService(r *spanReaderTest, indexName string) *mock.Call { @@ -811,7 +811,7 @@ func mockArchiveMultiSearchService(r *spanReaderTest, indexName string) *mock.Ca multiSearchService.On("Add", mock.Anything, mock.Anything, mock.Anything).Return(multiSearchService) multiSearchService.On("Index", indexName).Return(multiSearchService) r.client.On("MultiSearch").Return(multiSearchService) - return multiSearchService.On("Do", mock.AnythingOfType("*context.emptyCtx")) + return multiSearchService.On("Do", mock.AnythingOfType("*context.valueCtx")) } func mockSearchService(r *spanReaderTest) *mock.Call {