Add list analyzers by type test

ilyaglow · May 25, 2018 · d7213b5 · d7213b5
1 parent 001f276
commit d7213b5
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/analyzer_test.go b/analyzer_test.go
@@ -25,6 +25,26 @@ func TestListAnalyzers(t *testing.T) {
 	}
 }
 
+func TestListByTypeAnalyzers(t *testing.T) {
+	client, mux, _, closer := setup()
+	defer closer()
+
+	mux.HandleFunc("/"+analyzersByType+analyzerType, func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		w.Write(analyzersJSON)
+	})
+
+	got, _, err := client.Analyzers.ListByType(context.Background(), analyzerType)
+	if err != nil {
+		t.Errorf("Analyzer.ListByType returned error: %v", err)
+	}
+	if want := wantList; !reflect.DeepEqual(got, want) {
+		t.Errorf("Analyzer.ListByType = %+v, want %+v", got, want)
+	}
+}
+
+var analyzerType = "ip"
+
 var analyzersJSON = []byte(`
 [
   {