From 8e699203bc5f2940ee61db03e429583532459c3b Mon Sep 17 00:00:00 2001
From: "Michael G. Schwern" <schwern@pobox.com>
Date: Wed, 15 Feb 2023 17:15:25 -0800
Subject: [PATCH] fix: Make sure the pubsub tests clean up after themselves
 even if they fail.

---
 momento/pub_sub_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/momento/pub_sub_test.go b/momento/pub_sub_test.go
index 4f586f2d..9a8ba256 100644
--- a/momento/pub_sub_test.go
+++ b/momento/pub_sub_test.go
@@ -16,8 +16,8 @@ var cacheName = os.Getenv("TEST_CACHE_NAME")
 
 func TestMain(m *testing.M) {
 	setup()
+	defer teardown(client)
 	m.Run()
-	teardown(client)
 }
 
 func getClient() SimpleCacheClient {