diff --git a/tracing.go b/tracing.go index 6b7c5cf..0467310 100644 --- a/tracing.go +++ b/tracing.go @@ -65,6 +65,13 @@ type Status struct { FingerprintExpected string `json:"fingerprintExpected,omitempty"` FingerprintComputed string `json:"fingerprintComputed,omitempty"` Pods []NamespacedName `json:"pods,omitempty"` + NodeName string `json:"nodeName,omitempty"` +} + +func MakeStatus(nodeName string) Status { + return Status{ + NodeName: nodeName, + } } func (st *Status) Start(numPods int) { @@ -90,6 +97,7 @@ func (st *Status) Check(expected string) { func (st Status) Repr() string { var sb strings.Builder + sb.WriteString(fmt.Sprintf("> processing node %q\n", st.NodeName)) sb.WriteString(fmt.Sprintf("> processing %d pods\n", len(st.Pods))) for _, pod := range st.Pods { sb.WriteString("+ " + pod.Namespace + "/" + pod.Name + "\n") @@ -110,6 +118,7 @@ func (st Status) Clone() Status { FingerprintExpected: st.FingerprintExpected, FingerprintComputed: st.FingerprintComputed, Pods: pods, + NodeName: st.NodeName, } return ret } diff --git a/tracing_test.go b/tracing_test.go index aec9285..d4ccc72 100644 --- a/tracing_test.go +++ b/tracing_test.go @@ -95,7 +95,8 @@ func TestTraceStatusJSON(t *testing.T) { } } -var expectedStatusRepr = `> processing 5 pods +var expectedStatusRepr = `> processing node "test-node" +> processing 5 pods + ns1/n1 + ns1/n2 + ns2/n1 @@ -129,7 +130,7 @@ func TestTraceStatusRepr(t *testing.T) { }, } - st := Status{} + st := MakeStatus("test-node") fp := NewTracingFingerprint(len(pods), &st) for _, pod := range pods { fp.Add(pod.Namespace, pod.Name)