X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/297c4aaf43858eff5022a1e72eb8e09660bde4b0..HEAD:/services/crunch-dispatch-slurm/squeue_test.go

diff --git a/services/crunch-dispatch-slurm/squeue_test.go b/services/crunch-dispatch-slurm/squeue_test.go
index ef036dabd7..d41e1982b4 100644
--- a/services/crunch-dispatch-slurm/squeue_test.go
+++ b/services/crunch-dispatch-slurm/squeue_test.go
@@ -2,11 +2,12 @@
 //
 // SPDX-License-Identifier: AGPL-3.0
 
-package main
+package dispatchslurm
 
 import (
 	"time"
 
+	"github.com/sirupsen/logrus"
 	. "gopkg.in/check.v1"
 )
 
@@ -24,6 +25,7 @@ func (s *SqueueSuite) TestReleasePending(c *C) {
 		queue: uuids[0] + " 10000 4294000000 PENDING Resources\n" + uuids[1] + " 10000 4294000111 PENDING Resources\n" + uuids[2] + " 10000 0 PENDING BadConstraints\n",
 	}
 	sqc := &SqueueChecker{
+		Logger: logrus.StandardLogger(),
 		Slurm:  slurm,
 		Period: time.Hour,
 	}
@@ -88,6 +90,7 @@ func (s *SqueueSuite) TestReniceAll(c *C) {
 			queue: test.squeue,
 		}
 		sqc := &SqueueChecker{
+			Logger:         logrus.StandardLogger(),
 			Slurm:          slurm,
 			PrioritySpread: test.spread,
 			Period:         time.Hour,
@@ -112,6 +115,7 @@ func (s *SqueueSuite) TestReniceInvalidNiceValue(c *C) {
 		rejectNice10K: true,
 	}
 	sqc := &SqueueChecker{
+		Logger:         logrus.StandardLogger(),
 		Slurm:          slurm,
 		PrioritySpread: 1,
 		Period:         time.Hour,
@@ -155,6 +159,7 @@ func (s *SqueueSuite) TestSetPriorityBeforeQueued(c *C) {
 
 	slurm := &slurmFake{}
 	sqc := &SqueueChecker{
+		Logger: logrus.StandardLogger(),
 		Slurm:  slurm,
 		Period: time.Hour,
 	}