Add user infor to switch

cmd/juju/switch_test.go

 // Copyright 2013 Canonical Ltd.
 // Licensed under the AGPLv3, see LICENCE file for details.
 
 package main
 
 import (
         "os"
 
         gc "launchpad.net/gocheck"
 
         "launchpad.net/juju-core/cmd/envcmd"
+        "launchpad.net/juju-core/environs/configstore"
         _ "launchpad.net/juju-core/juju"
         "launchpad.net/juju-core/testing"
 )
 
 type SwitchSimpleSuite struct {
         testing.FakeJujuHomeSuite
 }
 
 var _ = gc.Suite(&SwitchSimpleSuite{})
 
+var testCreds = configstore.APICredentials{
+        User:     "joe",
+        Password: "baloney",
+}
+
+var apiEndpoint = configstore.APIEndpoint{
+        Addresses: []string{"example.com", "kremvax.ru"},
+        CACert:    "cert",
+}
+
+func patchEnvWithUser(c *gc.C, envName string) {
+        testing.WriteEnvironments(c, testing.MultipleEnvConfig)
+
+        // Patch APICredentials so that <envName>.jenv file is avaliable for
+        // switch to read the envirionment user from.
+        store, err := configstore.Default()
+        c.Assert(err, gc.IsNil)
+        info, err := store.CreateInfo(envName)
+        info.SetAPIEndpoint(apiEndpoint)
+        info.SetAPICredentials(testCreds)
+        info.Write()
+}
+
 func (*SwitchSimpleSuite) TestNoEnvironment(c *gc.C) {
         envPath := testing.HomePath(".juju", "environments.yaml")
         err := os.Remove(envPath)
         c.Assert(err, gc.IsNil)
         _, err = testing.RunCommand(c, &SwitchCommand{})
         c.Assert(err, gc.ErrorMatches, "couldn't read the environment")
 }
 
 func (*SwitchSimpleSuite) TestNoDefault(c *gc.C) {
         testing.WriteEnvironments(c, testing.MultipleEnvConfigNoDefault)
@@ skipping 72 matching lines @@
         c.Assert(err, gc.IsNil)
         c.Assert(testing.Stdout(context), gc.Equals, expectedEnvironments)
 }
 
 func (*SwitchSimpleSuite) TestListEnvironmentsAndChange(c *gc.C) {
         testing.WriteEnvironments(c, testing.MultipleEnvConfig)
         _, err := testing.RunCommand(c, &SwitchCommand{}, "--list", "erewhemos-2")
         c.Assert(err, gc.ErrorMatches, "cannot switch and list at the same time")
 }
 
+func (*SwitchSimpleSuite) TestListEnvironmentInfoDefault(c *gc.C) {
+        patchEnvWithUser(c, "erewhemos")
+        context, err := testing.RunCommand(c, &SwitchCommand{}, "--format", "yaml")
+        c.Assert(err, gc.IsNil)
+        c.Assert(testing.Stdout(context), gc.Equals,

[fwereade, 2014/06/02 08:56:03]

It's rarely a good idea to test yaml directly -- better (if more tedious) to
unmarshal whatever got marshalled and check that. (there are multiple valid
forms of the yaml outout, and who knows what might cause a change in what
actually gets written)

+                "user-name: joe\n"+
+                        "environ-name: erewhemos\n"+
+                        "state-servers:\n"+
+                        "- example.com\n"+
+                        "- kremvax.ru\n")
+}
+
+func (*SwitchSimpleSuite) TestListEnvironmentInfoYaml(c *gc.C) {
+        patchEnvWithUser(c, "erewhemos")
+        context, err := testing.RunCommand(c, &SwitchCommand{}, "--format", "yaml")
+        c.Assert(err, gc.IsNil)
+        c.Assert(testing.Stdout(context), gc.Equals,

[fwereade, 2014/06/02 08:56:03]

ditto

+                "user-name: joe\n"+
+                        "environ-name: erewhemos\n"+
+                        "state-servers:\n"+
+                        "- example.com\n"+
+                        "- kremvax.ru\n")
+}
+
+func (*SwitchSimpleSuite) TestListEnvironmentInfoJson(c *gc.C) {
+        patchEnvWithUser(c, "erewhemos")
+        context, err := testing.RunCommand(c, &SwitchCommand{}, "--format", "json")
+        c.Assert(err, gc.IsNil)
+        c.Assert(testing.Stdout(context), gc.Equals, "{\"user-name\":\"joe\",\"environ-name\":\"erewhemos\",\"state-servers\":[\"example.com\",\"kremvax.ru\"]}\n")

[fwereade, 2014/06/02 08:56:03]

ditto

+}
+
+func (*SwitchSimpleSuite) TestListEnvironmentInfoShowOldEnv(c *gc.C) {
+        patchEnvWithUser(c, "erewhemos-2")
+        context, err := testing.RunCommand(c, &SwitchCommand{}, "erewhemos-2", "--format", "json")
+        c.Assert(err, gc.IsNil)
+        c.Assert(testing.Stdout(context), gc.Equals, "{\"user-name\":\"joe\",\"environ-name\":\"erewhemos-2\",\"previous-environ-name\":\"erewhemos\",\"state-servers\":[\"example.com\",\"kremvax.ru\"]}\n")

[fwereade, 2014/06/02 08:56:03]

ditto

+}
+
 func (*SwitchSimpleSuite) TestTooManyParams(c *gc.C) {
         testing.WriteEnvironments(c, testing.MultipleEnvConfig)
         _, err := testing.RunCommand(c, &SwitchCommand{}, "foo", "bar")
         c.Assert(err, gc.ErrorMatches, `unrecognized args: ."bar".`)
 }