turns out stage names are useful information, so let's keep it

Juniper · Oct 14, 2023 · d36959a · d36959a
1 parent 29acb3e
commit d36959a
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 20 deletions.
diff --git a/apstra/api_iba_probes.go b/apstra/api_iba_probes.go
@@ -13,24 +13,25 @@ const (
 )
 
 type IbaProbe struct {
-	Id                ObjectId    `json:"id"`
-	Label             string      `json:"label"`
-	TaskError         string      `json:"task_error"`
-	AnomalyCount      int         `json:"anomaly_count"`
-	Tags              []string    `json:"tags"`
-	LastError         interface{} `json:"last_error"`
-	UpdatedAt         string      `json:"updated_at"`
-	UpdatedBy         string      `json:"updated_by"`
-	Disabled          bool        `json:"disabled"`
-	ConfigCompletedAt string      `json:"config_completed_at"`
-	State             string      `json:"state"`
-	Version           int         `json:"version"`
-	HostNode          string      `json:"host_node"`
-	TaskState         string      `json:"task_state"`
-	ConfigStartedAt   string      `json:"config_started_at"`
-	IbaUnit           string      `json:"iba_unit"`
-	PredefinedProbe   string      `json:"predefined_probe"`
-	Description       string      `json:"description"`
+	Id                ObjectId                 `json:"id"`
+	Label             string                   `json:"label"`
+	TaskError         string                   `json:"task_error"`
+	Stages            []map[string]interface{} `json:"stages"`
+	AnomalyCount      int                      `json:"anomaly_count"`
+	Tags              []string                 `json:"tags"`
+	LastError         interface{}              `json:"last_error"`
+	UpdatedAt         string                   `json:"updated_at"`
+	UpdatedBy         string                   `json:"updated_by"`
+	Disabled          bool                     `json:"disabled"`
+	ConfigCompletedAt string                   `json:"config_completed_at"`
+	State             string                   `json:"state"`
+	Version           int                      `json:"version"`
+	HostNode          string                   `json:"host_node"`
+	TaskState         string                   `json:"task_state"`
+	ConfigStartedAt   string                   `json:"config_started_at"`
+	IbaUnit           string                   `json:"iba_unit"`
+	PredefinedProbe   string                   `json:"predefined_probe"`
+	Description       string                   `json:"description"`
 }
 
 func (o *Client) getAllIbaProbes(ctx context.Context, bpId ObjectId) ([]IbaProbe, error) {

diff --git a/apstra/api_iba_probes_test.go b/apstra/api_iba_probes_test.go
@@ -70,13 +70,14 @@ func TestIbaProbes(t *testing.T) {
 			t.Logf("Description %s", p.Description)
 			t.Log(p)
 			t.Logf("Delete probe")
+			for _, i := range p.Stages {
+				t.Logf("Stage name %s", i["name"])
+			}
 			err = bpClient.DeleteIbaProbe(ctx, probeId)
 			if err != nil {
 				t.Fatal(err)
 			}
-
 			t.Logf("Delete Probe again, this should fail")
-
 			err = bpClient.DeleteIbaProbe(ctx, probeId)
 			if err == nil {
 				t.Fatal("Probe Deletion should have failed")