fix

uber · Feb 22, 2024 · f67e824 · f67e824
1 parent 29db00f
commit f67e824
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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,
 		},
 	}