From f8afd00f2826a4140d3be3b22e27aa1613e0a2ce Mon Sep 17 00:00:00 2001 From: Mahad Zaryab Date: Mon, 2 Dec 2024 12:46:00 -0500 Subject: [PATCH] Fix Test Signed-off-by: Mahad Zaryab --- plugin/storage/cassandra/factory_test.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/plugin/storage/cassandra/factory_test.go b/plugin/storage/cassandra/factory_test.go index 71add9cdb93..20f7ca54d5f 100644 --- a/plugin/storage/cassandra/factory_test.go +++ b/plugin/storage/cassandra/factory_test.go @@ -117,7 +117,8 @@ func TestCreateSpanReaderError(t *testing.T) { mock.Anything).Return(query) query.On("Exec").Return(errors.New("table does not exist")) f := NewFactory() - f.sessionBuilderFn = new(mockSessionBuilder).add(session, nil).build + f.archiveConfig = &config.Configuration{} + f.sessionBuilderFn = new(mockSessionBuilder).add(session, nil).add(session, nil).build require.NoError(t, f.Initialize(metrics.NullFactory, zap.NewNop())) r, err := f.CreateSpanReader() require.Error(t, err)