fix flaky internal_integration interupt specs

onsi · Oct 14, 2022 · 2105ba3 · 2105ba3
1 parent 2cc2219
commit 2105ba3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internal/internal_integration/interrupt_and_timeout_test.go b/internal/internal_integration/interrupt_and_timeout_test.go
@@ -426,8 +426,9 @@ var _ = Describe("Interrupts and Timeouts", func() {
 					BeforeEach(rt.T("bef-outer"))
 					Context("nested", func() {
 						BeforeEach(rt.T("bef-inner"))
-						It("A", rt.T("A", func() {
+						It("A", rt.TSC("A", func(c SpecContext) {
 							interruptHandler.Interrupt(interrupt_handler.InterruptCauseSignal)
+							<-c.Done()
 						}))
 						It("B", rt.T("B"))
 						AfterEach(rt.TSC("aft-inner", func(c SpecContext) {
@@ -472,8 +473,9 @@ var _ = Describe("Interrupts and Timeouts", func() {
 					BeforeEach(rt.T("bef-outer"))
 					Context("nested", func() {
 						BeforeEach(rt.T("bef-inner"))
-						It("A", rt.T("A", func() {
+						It("A", rt.TSC("A", func(c SpecContext) {
 							interruptHandler.Interrupt(interrupt_handler.InterruptCauseSignal)
+							<-c.Done()
 						}))
 						It("B", rt.T("B"))
 						AfterEach(rt.TSC("aft-inner", func(c SpecContext) {