Fix Test

Signed-off-by: Mahad Zaryab <[email protected]>
jaegertracing · Jan 22, 2025 · 66436be · 66436be
1 parent 47ff4c1
commit 66436be
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/plugin/storage/grpc/shared/grpc_handler_test.go b/plugin/storage/grpc/shared/grpc_handler_test.go
@@ -59,16 +59,11 @@ func withGRPCServer(fn func(r *grpcServerTest)) {
 	depReader := new(dependencyStoreMocks.Reader)
 	streamWriter := new(spanStoreMocks.Writer)
 
-	impl := &mockStoragePlugin{
+	mockPlugin := &mockStoragePlugin{
 		spanReader:   spanReader,
 		spanWriter:   spanWriter,
 		depsReader:   depReader,
 		streamWriter: streamWriter,
-	mockPlugin := &mockStoragePlugin{
-		spanReader:    spanReader,
-		spanWriter:    spanWriter,
-		depsReader:    depReader,
-		streamWriter:  streamWriter,
 	}
 
 	impl := &GRPCHandlerStorageImpl{
@@ -82,10 +77,10 @@ func withGRPCServer(fn func(r *grpcServerTest)) {
 			return mockPlugin.depsReader
 		},
 		ArchiveSpanReader: func() spanstore.Reader {
-			return mockPlugin.archiveReader
+			return mockPlugin.spanReader
 		},
 		ArchiveSpanWriter: func() spanstore.Writer {
-			return mockPlugin.archiveWriter
+			return mockPlugin.spanWriter
 		},
 		StreamingSpanWriter: func() spanstore.Writer {
 			return mockPlugin.streamWriter