From 4b7a4bd1b87d1dbba283e577aa03ce7e390d85d8 Mon Sep 17 00:00:00 2001
From: Nick White <git@njw.name>
Date: Mon, 2 Aug 2021 10:45:53 +0100
Subject: internal/pipeline: Add test (incomplete but working) for UploadImages

---
 local.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'local.go')

diff --git a/local.go b/local.go
index 615f9a6..58830c7 100644
--- a/local.go
+++ b/local.go
@@ -21,6 +21,7 @@ const qidOCR = "queueOCR"
 const qidAnalyse = "queueAnalyse"
 const qidTest = "queueTest"
 const storageId = "storage"
+const testStorageId = "test"
 
 // LocalConn is a simple implementation of the pipeliner interface
 // that doesn't rely on any "cloud" services, instead doing everything
@@ -66,6 +67,11 @@ func (a *LocalConn) Init() error {
 
 // TestInit does nothing for local connections
 func (a *LocalConn) TestInit() error {
+	err := os.Mkdir(filepath.Join(a.TempDir, testStorageId), 0700)
+	if err != nil && !os.IsExist(err) {
+		return fmt.Errorf("Error creating test storage directory: %v", err)
+	}
+
 	return nil
 }
 
@@ -134,6 +140,11 @@ func (a *LocalConn) WIPStorageId() string {
 	return storageId
 }
 
+func (a *LocalConn) TestStorageId() string {
+	return testStorageId
+}
+
+
 func prefixwalker(dirpath string, prefix string, list *[]ObjMeta) filepath.WalkFunc {
 	return func(path string, info os.FileInfo, err error) error {
 		if err != nil {
-- 
cgit v1.2.1-24-ge1ad