From b301cf628ae37405518c9f7eeb3bf4a0f181b302 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philip=20Dub=C3=A9?= <gunkmute@gmail.com>
Date: Tue, 27 Aug 2019 17:32:00 +0000
Subject: [PATCH] Test worker_cleanup_job_schema_cache actually drops schemas

---
 .../expected/task_tracker_cleanup_job.out     | 25 +++++++++++++++++++
 .../regress/sql/task_tracker_cleanup_job.sql  |  6 +++++
 2 files changed, 31 insertions(+)

diff --git a/src/test/regress/expected/task_tracker_cleanup_job.out b/src/test/regress/expected/task_tracker_cleanup_job.out
index fd6a5d77e..664084a77 100644
--- a/src/test/regress/expected/task_tracker_cleanup_job.out
+++ b/src/test/regress/expected/task_tracker_cleanup_job.out
@@ -5,6 +5,31 @@ SET citus.next_shard_id TO 1060000;
 \set JobId 401010
 \set CompletedTaskId 801107
 \set RunningTaskId 801108
+-- Test worker_cleanup_job_schema_cache
+SELECT * FROM task_tracker_assign_task(2, 2, '');
+ task_tracker_assign_task 
+--------------------------
+ 
+(1 row)
+
+SELECT count(*) FROM pg_catalog.pg_namespace WHERE nspname = 'pg_merge_job_0002';
+ count 
+-------
+     1
+(1 row)
+
+SELECT worker_cleanup_job_schema_cache();
+ worker_cleanup_job_schema_cache 
+---------------------------------
+ 
+(1 row)
+
+SELECT count(*) FROM pg_catalog.pg_namespace WHERE nspname = 'pg_merge_job_0002';
+ count 
+-------
+     0
+(1 row)
+
 -- We assign two tasks to the task tracker. The first task should complete and
 -- the second task should continue to keep running.
 SELECT task_tracker_assign_task(:JobId, :CompletedTaskId,
diff --git a/src/test/regress/sql/task_tracker_cleanup_job.sql b/src/test/regress/sql/task_tracker_cleanup_job.sql
index 153fa7078..9a8b9af82 100644
--- a/src/test/regress/sql/task_tracker_cleanup_job.sql
+++ b/src/test/regress/sql/task_tracker_cleanup_job.sql
@@ -10,6 +10,12 @@ SET citus.next_shard_id TO 1060000;
 \set CompletedTaskId 801107
 \set RunningTaskId 801108
 
+-- Test worker_cleanup_job_schema_cache
+SELECT * FROM task_tracker_assign_task(2, 2, '');
+SELECT count(*) FROM pg_catalog.pg_namespace WHERE nspname = 'pg_merge_job_0002';
+SELECT worker_cleanup_job_schema_cache();
+SELECT count(*) FROM pg_catalog.pg_namespace WHERE nspname = 'pg_merge_job_0002';
+
 -- We assign two tasks to the task tracker. The first task should complete and
 -- the second task should continue to keep running.