sharedlibraries: tests: Fix bugs in the test (DataDog#31744)

kkpan11 · Dec 4, 2024 · a79d704 · a79d704
1 parent a9eac76
commit a79d704
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pkg/network/usm/sharedlibraries/ebpf_test.go b/pkg/network/usm/sharedlibraries/ebpf_test.go
@@ -43,16 +43,20 @@ func (s *EbpfProgramSuite) TestCanInstantiateMultipleTimes() {
 
 	prog := GetEBPFProgram(cfg)
 	require.NotNil(t, prog)
-	t.Cleanup(prog.Stop)
+	t.Cleanup(func() {
+		if prog != nil {
+			prog.Stop()
+		}
+	})
 
 	require.NoError(t, prog.InitWithLibsets(LibsetCrypto))
 	prog.Stop()
+	prog = nil
 
 	prog2 := GetEBPFProgram(cfg)
 	require.NotNil(t, prog2)
-
-	require.NoError(t, prog.InitWithLibsets(LibsetCrypto))
 	t.Cleanup(prog2.Stop)
+	require.NoError(t, prog2.InitWithLibsets(LibsetCrypto))
 }
 
 func (s *EbpfProgramSuite) TestProgramReceivesEventsWithSingleLibset() {