summaryrefslogtreecommitdiff
path: root/internal/pipeline/put_test.go
diff options
context:
space:
mode:
authorNick White <git@njw.name>2021-07-27 14:22:03 +0100
committerNick White <git@njw.name>2021-07-27 14:22:03 +0100
commit8780dda26a819b4b93caba75a893fb082849406e (patch)
tree98715595ec26a3714c726a6e98dd2e54b5a01219 /internal/pipeline/put_test.go
parent13b8605ba5f8554aedf658aacee5bc436887cbfa (diff)
internal/pipeline: add tests for DetectQueueType
Diffstat (limited to 'internal/pipeline/put_test.go')
-rw-r--r--internal/pipeline/put_test.go24
1 files changed, 24 insertions, 0 deletions
diff --git a/internal/pipeline/put_test.go b/internal/pipeline/put_test.go
index 9cc8b02..af18ab4 100644
--- a/internal/pipeline/put_test.go
+++ b/internal/pipeline/put_test.go
@@ -7,6 +7,7 @@ package pipeline
import (
"errors"
"os"
+ "rescribe.xyz/bookpipeline"
"testing"
)
@@ -49,3 +50,26 @@ func Test_CheckImages(t *testing.T) {
})
}
}
+
+func Test_DetectQueueType(t *testing.T) {
+ conn := &bookpipeline.LocalConn{}
+
+ cases := []struct {
+ dir string
+ qid string
+ }{
+ {"testdata/mostlypng", conn.WipeQueueId()},
+ {"testdata/mostlyjpg", conn.PreQueueId()},
+ {"testdata/equalpngandjpg", conn.PreQueueId()},
+ {"testdata/nonexistent", conn.PreQueueId()},
+ }
+
+ for _, c := range cases {
+ t.Run(c.dir, func(t *testing.T) {
+ qid := DetectQueueType(c.dir, conn)
+ if qid != c.qid {
+ t.Fatalf("Error, expected qid %v, got qid %v", qid, c.qid)
+ }
+ })
+ }
+}