pkg/sanity: close connection last

Ginkgo invokes AfterEach in the order in which they are defined,
therefore we should define the AfterEach where the connection is closed
as late as possible. That way the tests can have their own AfterEach
where they can still use the connection.

pkg/sanity/tests.go

@@ -42,16 +42,15 @@ func DescribeSanity(text string, body func(*SanityContext)) bool {
 func registerTestsInGinkgo(sc *SanityContext) {
 	for _, test := range tests {
 		Describe(test.text, func() {
-
 			BeforeEach(func() {
 				sc.setup()
 			})
 
+			test.body(sc)
+
 			AfterEach(func() {
 				sc.teardown()
 			})
-
-			test.body(sc)
 		})
 	}
 }