sdk/go/arvados/config_test.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4
   5 package arvados
   6
   7 import (
   8         "encoding/json"
   9
  10         "github.com/ghodss/yaml"
  11         check "gopkg.in/check.v1"
  12 )
  13
  14 var _ = check.Suite(&ConfigSuite{})
  15
  16 type ConfigSuite struct{}
  17
  18 func (s *ConfigSuite) TestStringSetAsArray(c *check.C) {
  19         var cluster Cluster
  20         yaml.Unmarshal([]byte(`
  21 API:
  22   DisabledAPIs: [jobs.list]`), &cluster)
  23         c.Check(len(cluster.API.DisabledAPIs), check.Equals, 1)
  24         _, ok := cluster.API.DisabledAPIs["jobs.list"]
  25         c.Check(ok, check.Equals, true)
  26 }
  27
  28 func (s *ConfigSuite) TestInstanceTypesAsHash(c *check.C) {
  29         var cluster Cluster
  30         yaml.Unmarshal([]byte("InstanceTypes:\n  foo:\n    ProviderType: bar\n"), &cluster)
  31         c.Check(len(cluster.InstanceTypes), check.Equals, 1)
  32         c.Check(cluster.InstanceTypes["foo"].Name, check.Equals, "foo")
  33         c.Check(cluster.InstanceTypes["foo"].ProviderType, check.Equals, "bar")
  34 }
  35
  36 func (s *ConfigSuite) TestInstanceTypeSize(c *check.C) {
  37         var it InstanceType
  38         err := yaml.Unmarshal([]byte("Name: foo\nIncludedScratch: 4GB\nRAM: 4GiB\n"), &it)
  39         c.Check(err, check.IsNil)
  40         c.Check(int64(it.IncludedScratch), check.Equals, int64(4000000000))
  41         c.Check(int64(it.RAM), check.Equals, int64(4294967296))
  42 }
  43
  44 func (s *ConfigSuite) TestInstanceTypeFixup(c *check.C) {
  45         for _, confdata := range []string{
  46                 // Current format: map of entries
  47                 `{foo4: {IncludedScratch: 4GB}, foo8: {ProviderType: foo_8, AddedScratch: 8GB}}`,
  48         } {
  49                 c.Log(confdata)
  50                 var itm InstanceTypeMap
  51                 err := yaml.Unmarshal([]byte(confdata), &itm)
  52                 c.Check(err, check.IsNil)
  53
  54                 c.Check(itm["foo4"].Name, check.Equals, "foo4")
  55                 c.Check(itm["foo4"].ProviderType, check.Equals, "foo4")
  56                 c.Check(itm["foo4"].Scratch, check.Equals, ByteSize(4000000000))
  57                 c.Check(itm["foo4"].AddedScratch, check.Equals, ByteSize(0))
  58                 c.Check(itm["foo4"].IncludedScratch, check.Equals, ByteSize(4000000000))
  59
  60                 c.Check(itm["foo8"].Name, check.Equals, "foo8")
  61                 c.Check(itm["foo8"].ProviderType, check.Equals, "foo_8")
  62                 c.Check(itm["foo8"].Scratch, check.Equals, ByteSize(8000000000))
  63                 c.Check(itm["foo8"].AddedScratch, check.Equals, ByteSize(8000000000))
  64                 c.Check(itm["foo8"].IncludedScratch, check.Equals, ByteSize(0))
  65         }
  66 }
  67
  68 func (s *ConfigSuite) TestURLTrailingSlash(c *check.C) {
  69         var a, b map[URL]bool
  70         json.Unmarshal([]byte(`{"https://foo.example": true}`), &a)
  71         json.Unmarshal([]byte(`{"https://foo.example/": true}`), &b)
  72         c.Check(a, check.DeepEquals, b)
  73 }