Adding defer () function to flush all entries before exit

openconfig · Jan 27, 2025 · 40caea7 · 40caea7
1 parent 5f6a91e
commit 40caea7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/feature/gribi/otg_tests/gribigo_compliance_test/gribigo_compliance_test.go b/feature/gribi/otg_tests/gribigo_compliance_test/gribigo_compliance_test.go
@@ -224,6 +224,14 @@ func testAdditionalCompliance(tcArgs *testArgs, t *testing.T) {
 			fn:   addNHGReferencingToDownPort,
 		},
 	}
+
+	defer func() {
+		// Flush all entries after test.
+		if err := gribi.FlushAll(tcArgs.client); err != nil {
+			t.Error(err)
+		}
+	}()
+
 	for _, tt := range tests {
 		t.Run(tt.desc, func(t *testing.T) {
 			if err := gribi.FlushAll(tcArgs.client); err != nil {