internal/pipeline: add tests for DetectQueueType

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)
+			}
+		})
+	}
+}