Skip to content

Commit

Permalink
Add unique pid check
Browse files Browse the repository at this point in the history
  • Loading branch information
winebarrel committed Oct 13, 2024
1 parent c34d2a2 commit c530fdc
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions workerpool_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ package qg_test
import (
"database/sql"
"fmt"
"slices"
"sort"
"strings"
"sync"
Expand Down Expand Up @@ -523,23 +524,31 @@ func TestWorkerPoolSingleQueueMultiDB(t *testing.T) {
}

rows := []string{}
pids := []string{}

for rs.Next() {
var name, _value string
var name, value string
var count int
err = rs.Scan(&name, &_value, &count)
err = rs.Scan(&name, &value, &count)
if err != nil {
t.Fatal(err)
}
rows = append(rows, fmt.Sprintf("%s,%d", name, count))
pids = append(pids, value)
}

sort.Strings(rows)
sort.Strings(pids)
pids = slices.Compact(pids)

if strings.Join(rows, " ") != "job1,1 job1,1 job2,1 job2,1 job3,1 job3,1 job4,1 job4,1" {
t.Errorf("unexpected result: %v", rows)
}

if len(pids) != 8 {
t.Errorf("unexpected unique pid count: %v", len(pids))
}

var queJobsCount int
err = checkDB.QueryRow("SELECT COUNT(*) FROM que_jobs").Scan(&queJobsCount)
if err != nil {
Expand Down

0 comments on commit c530fdc

Please sign in to comment.