#23 fixes context in tests

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) {

store/redis/redis_test.go

@@ -1,4 +1,4 @@
-// go:build unit
+//go:build unit
 
 package redis