From f67e82492534fe45f64066b01c5652e656920e0d Mon Sep 17 00:00:00 2001 From: taylan isikdemir Date: Thu, 22 Feb 2024 13:41:07 -0800 Subject: [PATCH] fix --- common/persistence/nosql/nosqlplugin/cassandra/db_test.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/common/persistence/nosql/nosqlplugin/cassandra/db_test.go b/common/persistence/nosql/nosqlplugin/cassandra/db_test.go index e442f46b17c..f5b28fd0384 100644 --- a/common/persistence/nosql/nosqlplugin/cassandra/db_test.go +++ b/common/persistence/nosql/nosqlplugin/cassandra/db_test.go @@ -25,6 +25,7 @@ import ( "testing" "github.com/golang/mock/gomock" + "github.com/uber/cadence/common/config" "github.com/uber/cadence/common/dynamicconfig" "github.com/uber/cadence/common/log/testlogger" @@ -222,6 +223,7 @@ func TestExecuteBatchWithConsistencyAll(t *testing.T) { }, batchMockPrep: func(batch *gocql.MockBatch) { batch.EXPECT().Consistency(cassandraAllConslevel).Return(batch).Times(1) + batch.EXPECT().Consistency(cassandraDefaultConsLevel).Return(batch).Times(1) }, wantErr: false, }, @@ -237,7 +239,7 @@ func TestExecuteBatchWithConsistencyAll(t *testing.T) { batchMockPrep: func(batch *gocql.MockBatch) { batch.EXPECT().Consistency(cassandraAllConslevel).Return(batch).Times(1) }, - wantErr: false, + wantErr: true, }, }