Merge branch '15028-cwl-v1.1' refs #15028
[arvados.git] / services / keepstore / s3_volume_test.go
index baa9dda9e6619c32b6c00c0a1671b218999ed4f8..6377420ff4b381cba49b07d2813fb4803f03aa62 100644 (file)
@@ -172,7 +172,7 @@ func (s *StubbedS3Suite) testContextCancel(c *check.C, testFunc func(context.Con
        vol.Endpoint = srv.URL
        v = &TestableS3Volume{S3Volume: &vol}
        metrics := newVolumeMetricsVecs(prometheus.NewRegistry())
-       v.Start(metrics.opsCounters, metrics.errCounters, metrics.ioBytes)
+       v.Start(metrics)
 
        ctx, cancel := context.WithCancel(context.Background())
 
@@ -433,7 +433,7 @@ func (s *StubbedS3Suite) newTestableVolume(c *check.C, raceWindow time.Duration,
                serverClock: clock,
        }
        metrics := newVolumeMetricsVecs(prometheus.NewRegistry())
-       v.Start(metrics.opsCounters, metrics.errCounters, metrics.ioBytes)
+       v.Start(metrics)
        err = v.bucket.PutBucket(s3.ACL("private"))
        c.Assert(err, check.IsNil)
        return v
@@ -451,7 +451,7 @@ Volumes:
        c.Check(cfg.Volumes[0].GetStorageClasses(), check.DeepEquals, []string{"class_a", "class_b"})
 }
 
-func (v *TestableS3Volume) Start(opsCounters, errCounters, ioBytes *prometheus.CounterVec) error {
+func (v *TestableS3Volume) Start(vm *volumeMetricsVecs) error {
        tmp, err := ioutil.TempFile("", "keepstore")
        v.c.Assert(err, check.IsNil)
        defer os.Remove(tmp.Name())
@@ -462,7 +462,7 @@ func (v *TestableS3Volume) Start(opsCounters, errCounters, ioBytes *prometheus.C
        v.S3Volume.AccessKeyFile = tmp.Name()
        v.S3Volume.SecretKeyFile = tmp.Name()
 
-       v.c.Assert(v.S3Volume.Start(opsCounters, errCounters, ioBytes), check.IsNil)
+       v.c.Assert(v.S3Volume.Start(vm), check.IsNil)
        return nil
 }
 
@@ -493,3 +493,7 @@ func (v *TestableS3Volume) TouchWithDate(locator string, lastPut time.Time) {
 func (v *TestableS3Volume) Teardown() {
        v.server.Quit()
 }
+
+func (v *TestableS3Volume) ReadWriteOperationLabelValues() (r, w string) {
+       return "get", "put"
+}