Arvados-DCO-1.1-Signed-off-by: Radhika Chippada <radhika@curoverse.com>
[arvados.git] / sdk / go / keepclient / discover_test.go
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: Apache-2.0
4
5 package keepclient
6
7 import (
8         "crypto/md5"
9         "fmt"
10         "net/http"
11         "os"
12
13         "gopkg.in/check.v1"
14
15         "git.curoverse.com/arvados.git/sdk/go/arvadosclient"
16         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
17 )
18
19 func (s *ServerRequiredSuite) TestOverrideDiscovery(c *check.C) {
20         defer os.Setenv("ARVADOS_KEEP_SERVICES", "")
21
22         hash := fmt.Sprintf("%x+3", md5.Sum([]byte("TestOverrideDiscovery")))
23         st := StubGetHandler{
24                 c,
25                 hash,
26                 arvadostest.ActiveToken,
27                 http.StatusOK,
28                 []byte("TestOverrideDiscovery")}
29         ks := RunSomeFakeKeepServers(st, 2)
30
31         os.Setenv("ARVADOS_KEEP_SERVICES", "")
32         arv1, err := arvadosclient.MakeArvadosClient()
33         c.Assert(err, check.IsNil)
34         arv1.ApiToken = arvadostest.ActiveToken
35
36         os.Setenv("ARVADOS_KEEP_SERVICES", ks[0].url+"  "+ks[1].url+" ")
37         arv2, err := arvadosclient.MakeArvadosClient()
38         c.Assert(err, check.IsNil)
39         arv2.ApiToken = arvadostest.ActiveToken
40
41         // ARVADOS_KEEP_SERVICES was empty when we created arv1, but
42         // it pointed to our stub servers when we created
43         // arv2. Regardless of what it's set to now, a keepclient for
44         // arv2 should use our stub servers, but one created for arv1
45         // should not.
46
47         kc1, err := MakeKeepClient(arv1)
48         c.Assert(err, check.IsNil)
49         kc2, err := MakeKeepClient(arv2)
50         c.Assert(err, check.IsNil)
51
52         _, _, _, err = kc1.Get(hash)
53         c.Check(err, check.NotNil)
54         _, _, _, err = kc2.Get(hash)
55         c.Check(err, check.IsNil)
56 }