Add tests for influx consistency level parsing

cluster/points_writer.go

@@ -234,8 +234,8 @@ func (w *PointsWriter) writeToShard(shard *meta.ShardInfo, database, retentionPo
 
 	// response channel for each shard writer go routine
 	type AsyncWriteResult struct {
-		Owner  meta.ShardOwner
-		Err    error
+		Owner meta.ShardOwner
+		Err   error
 	}
 	ch := make(chan *AsyncWriteResult, len(shard.Owners))
 

cmd/influx/main.go

@@ -53,7 +53,7 @@ type CommandLine struct {
 	Execute          string
 	ShowVersion      bool
 	Import           bool
-	PPS              int    // Controls how many points per second the import will allow via throttling
+	PPS              int // Controls how many points per second the import will allow via throttling
 	Path             string
 	Compressed       bool
 }

cmd/influx/main_test.go

@@ -91,6 +91,31 @@ func TestParseCommand_Use(t *testing.T) {
 	}
 }
 
+func TestParseCommand_Consistency(t *testing.T) {
+	t.Parallel()
+	c := main.CommandLine{}
+	tests := []struct {
+		cmd string
+	}{
+		{cmd: "consistency one"},
+		{cmd: " consistency one"},
+		{cmd: "consistency one "},
+		{cmd: "consistency one;"},
+		{cmd: "consistency one; "},
+		{cmd: "Consistency one"},
+	}
+
+	for _, test := range tests {
+		if !c.ParseCommand(test.cmd) {
+			t.Fatalf(`Command "consistency" failed for %q.`, test.cmd)
+		}
+
+		if c.WriteConsistency != "one" {
+			t.Fatalf(`Command "consistency" changed consistency to %q. Expected one`, c.WriteConsistency)
+		}
+	}
+}
+
 func TestParseCommand_Insert(t *testing.T) {
 	t.Parallel()
 	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {