Merge pull request #42 from kanmu/add_que_jobs_count_check

Add que_jobs count check
kanmu · Oct 11, 2024 · 30eb2cd · 30eb2cd
2 parents 2d7a8c4 + d147f82
commit 30eb2cd
Showing 1 changed file with 30 additions and 15 deletions.
diff --git a/workerpool_test.go b/workerpool_test.go
@@ -56,11 +56,7 @@ func TestWorkerPool(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	_, err = checkDB.Exec("TRUNCATE TABLE que_jobs")
-	if err != nil {
-		t.Fatal(err)
-	}
-	_, err = checkDB.Exec("TRUNCATE TABLE job_test")
+	_, err = checkDB.Exec("TRUNCATE TABLE que_jobs, job_test")
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -101,6 +97,15 @@ func TestWorkerPool(t *testing.T) {
 			strings.Repeat("job4,queue1 ", 10) {
 		t.Errorf("unexpected result: %v", rows)
 	}
+
+	var queJobsCount int
+	err = checkDB.QueryRow("SELECT COUNT(*) FROM que_jobs").Scan(&queJobsCount)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if queJobsCount != 0 {
+		t.Errorf("unexpected que_jobs count: %d", queJobsCount)
+	}
 }
 
 func TestWorkerPoolMultiQueue(t *testing.T) {
@@ -190,11 +195,7 @@ func TestWorkerPoolMultiQueue(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	_, err = checkDB.Exec("TRUNCATE TABLE que_jobs")
-	if err != nil {
-		t.Fatal(err)
-	}
-	_, err = checkDB.Exec("TRUNCATE TABLE job_test")
+	_, err = checkDB.Exec("TRUNCATE TABLE que_jobs, job_test")
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -245,6 +246,15 @@ func TestWorkerPoolMultiQueue(t *testing.T) {
 			strings.Repeat("job4,queue2 ", 3) {
 		t.Errorf("unexpected result: %v", rows)
 	}
+
+	var queJobsCount int
+	err = checkDB.QueryRow("SELECT COUNT(*) FROM que_jobs").Scan(&queJobsCount)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if queJobsCount != 0 {
+		t.Errorf("unexpected que_jobs count: %d", queJobsCount)
+	}
 }
 
 func TestWorkerPoolMultiDB(t *testing.T) {
@@ -341,11 +351,7 @@ func TestWorkerPoolMultiDB(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	_, err = checkDB.Exec("TRUNCATE TABLE que_jobs")
-	if err != nil {
-		t.Fatal(err)
-	}
-	_, err = checkDB.Exec("TRUNCATE TABLE job_test")
+	_, err = checkDB.Exec("TRUNCATE TABLE que_jobs, job_test")
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -403,4 +409,13 @@ func TestWorkerPoolMultiDB(t *testing.T) {
 			strings.Repeat("job4,queue2 ", 5) {
 		t.Errorf("unexpected result: %v", rows)
 	}
+
+	var queJobsCount int
+	err = checkDB.QueryRow("SELECT COUNT(*) FROM que_jobs").Scan(&queJobsCount)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if queJobsCount != 0 {
+		t.Errorf("unexpected que_jobs count: %d", queJobsCount)
+	}
 }