6235: Discovery() returns the requested value directly instead of a single-entry...

[arvados.git] / sdk / go / arvadosclient / arvadosclient_test.go

diff --git a/sdk/go/arvadosclient/arvadosclient_test.go b/sdk/go/arvadosclient/arvadosclient_test.go
index 17911c6e21c8e64ec4aaea718ed49e3dc440c0d2..6a9e13bb27a9d39e8af98e4cc5a22fbd01b29f97 100644 (file)
--- a/sdk/go/arvadosclient/arvadosclient_test.go
+++ b/sdk/go/arvadosclient/arvadosclient_test.go
@@ -105,14 +105,14 @@ func (s *ServerRequiredSuite) TestErrorResponse(c *C) {
 
 func (s *ServerRequiredSuite) TestAPIDiscovery_Get_defaultCollectionReplication(c *C) {
        arv, err := MakeArvadosClient()
-       valueMap, err := arv.Discovery("defaultCollectionReplication")
+       value, err := arv.Discovery("defaultCollectionReplication")
        c.Assert(err, IsNil)
-       c.Assert(valueMap["defaultCollectionReplication"], NotNil)
+       c.Assert(value, NotNil)
 }
 
 func (s *ServerRequiredSuite) TestAPIDiscovery_Get_noSuchParameter(c *C) {
        arv, err := MakeArvadosClient()
-       valueMap, err := arv.Discovery("noSuchParameter")
-       c.Assert(err, IsNil)
-       c.Assert(valueMap["noSuchParameter"], IsNil)
+       value, err := arv.Discovery("noSuchParameter")
+       c.Assert(err, NotNil)
+       c.Assert(value, IsNil)
 }