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,
  },
  }