Merge branch '13219-jobs-time-limit'

[arvados.git] / services / keep-balance / main_test.go

diff --git a/services/keep-balance/main_test.go b/services/keep-balance/main_test.go
index 4a56098b29d1dbb7d8352d02c8ee0b492a6aea76..a2804344bf440fdfb89aae104ba6a7fb14b6a250 100644 (file)
--- a/services/keep-balance/main_test.go
+++ b/services/keep-balance/main_test.go
@@ -1,9 +1,13 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
 package main
 
 import (
-       "encoding/json"
        "time"
 
+       "github.com/ghodss/yaml"
        check "gopkg.in/check.v1"
 )
 
@@ -13,7 +17,7 @@ type mainSuite struct{}
 
 func (s *mainSuite) TestExampleJSON(c *check.C) {
        var config Config
-       c.Check(json.Unmarshal(exampleConfigFile, &config), check.IsNil)
+       c.Check(yaml.Unmarshal(exampleConfigFile, &config), check.IsNil)
        c.Check(config.KeepServiceTypes, check.DeepEquals, []string{"disk"})
        c.Check(config.Client.AuthToken, check.Equals, "xyzzy")
        c.Check(time.Duration(config.RunPeriod), check.Equals, 600*time.Second)
@@ -21,8 +25,7 @@ func (s *mainSuite) TestExampleJSON(c *check.C) {
 
 func (s *mainSuite) TestConfigJSONWithKeepServiceList(c *check.C) {
        var config Config
-       c.Check(json.Unmarshal([]byte(`
-               {
+       c.Check(yaml.Unmarshal([]byte(`{
                    "Client": {
                        "APIHost": "zzzzz.arvadosapi.com:443",
                        "AuthToken": "xyzzy",