From 27c6be0d1b5bb8a70c3b1c4018bd93bdb9f12295 Mon Sep 17 00:00:00 2001
From: Alex Senger <a.senger@gridx.de>
Date: Fri, 20 Sep 2024 11:37:52 +0200
Subject: [PATCH] #23 fixes context in tests

---
 store/postgres/postgres_test.go | 2 +-
 store/redis/redis_test.go       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/store/postgres/postgres_test.go b/store/postgres/postgres_test.go
index 0cc3353..acb09b4 100644
--- a/store/postgres/postgres_test.go
+++ b/store/postgres/postgres_test.go
@@ -152,7 +152,7 @@ func TestCheckpoint_Get_NoRows(t *testing.T) {
 	if err := mock.ExpectationsWereMet(); err != nil {
 		t.Errorf("there were unfulfilled expectations: %s", err)
 	}
-	_ = ck.Shutdown()
+	_ = ck.Shutdown(ctx)
 }
 
 func TestCheckpoint_Get_QueryError(t *testing.T) {
diff --git a/store/redis/redis_test.go b/store/redis/redis_test.go
index da333c3..5a7aa0c 100644
--- a/store/redis/redis_test.go
+++ b/store/redis/redis_test.go
@@ -1,4 +1,4 @@
-// go:build unit
+//go:build unit
 
 package redis