fix

vdaas · Oct 2, 2020 · b443696 · b443696
1 parent fd108ff
commit b443696
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/internal/db/nosql/cassandra/option_test.go b/internal/db/nosql/cassandra/option_test.go
@@ -916,6 +916,17 @@ func TestWithConsistency(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "set consistency level success with complex string",
+			args: args{
+				consistency: "-One_",
+			},
+			want: want{
+				obj: &T{
+					consistency: gocql.One,
+				},
+			},
+		},
 		{
 			name: "return error when consistency is empty",
 			args: args{
@@ -1001,6 +1012,17 @@ func TestWithSerialConsistency(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "set serial consistency level success with complex string",
+			args: args{
+				consistency: "-serial_",
+			},
+			want: want{
+				obj: &T{
+					serialConsistency: gocql.Serial,
+				},
+			},
+		},
 		{
 			name: "return error when consistency is empty",
 			args: args{