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

---
 aws.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'aws.go')

diff --git a/aws.go b/aws.go
index 21a215c..8e9410f 100644
--- a/aws.go
+++ b/aws.go
@@ -52,6 +52,7 @@ type AwsConn struct {
 	uploader                                  *s3manager.Uploader
 	wipequrl, prequrl, ocrpgqurl, analysequrl string
 	testqurl                                  string
+	teststorageid                             string
 	wipstorageid                              string
 }
 
@@ -139,6 +140,7 @@ func (a *AwsConn) TestInit() error {
 		return errors.New(fmt.Sprintf("Error getting test queue URL: %s\n", err))
 	}
 	a.testqurl = *result.QueueUrl
+	a.teststorageid = storageTest
 	return nil
 }
 
@@ -346,6 +348,10 @@ func (a *AwsConn) AnalyseQueueId() string {
 	return a.analysequrl
 }
 
+func (a *AwsConn) TestStorageId() string {
+	return a.teststorageid
+}
+
 func (a *AwsConn) WIPStorageId() string {
 	return a.wipstorageid
 }
@@ -612,7 +618,7 @@ func (a *AwsConn) Log(v ...interface{}) {
 // mkpipeline sets up necessary buckets and queues for the pipeline
 // TODO: also set up the necessary security group and iam stuff
 func (a *AwsConn) MkPipeline() error {
-	buckets := []string{storageWip}
+	buckets := []string{storageTest, storageWip}
 	queues := []string{queuePreProc, queueWipeOnly, queueAnalyse, queueOcrPage, queueTest}
 
 	for _, bucket := range buckets {
-- 
cgit v1.2.1-24-ge1ad