← Back to branch summary

~themue/juju-core/go-state-lifecycle-relation

« back to all changes in this revision

Viewing changes to cmd/juju/addunit_test.go

  • Committer: Frank Mueller
  • Date: 2012-08-21 08:48:07 UTC
  • mfrom: (354.1.63 juju-core)
  • Revision ID: themue@gmail.com-20120821084807-l97cttfvqwa3omge
state: merged trunk to stay up-to-date

Show diffs side-by-side

added added

removed removed

Lines of Context:
cmd/juju/addunit_test.go
 
1
package main
 
2
 
 
3
import (
 
4
        "bytes"
 
5
        . "launchpad.net/gocheck"
 
6
        "launchpad.net/juju-core/charm"
 
7
        "launchpad.net/juju-core/cmd"
 
8
        "launchpad.net/juju-core/testing"
 
9
)
 
10
 
 
11
type AddUnitSuite struct {
 
12
        DeploySuite
 
13
}
 
14
 
 
15
var _ = Suite(&AddUnitSuite{})
 
16
 
 
17
func (s *AddUnitSuite) SetUpTest(c *C) {
 
18
        s.DeploySuite.SetUpTest(c)
 
19
}
 
20
 
 
21
func (s *AddUnitSuite) TearDownTest(c *C) {
 
22
        s.DeploySuite.TearDownTest(c)
 
23
}
 
24
 
 
25
func runAddUnit(c *C, args ...string) error {
 
26
        com := &AddUnitCommand{}
 
27
        err := com.Init(newFlagSet(), args)
 
28
        c.Assert(err, IsNil)
 
29
        return com.Run(&cmd.Context{c.MkDir(), &bytes.Buffer{}, &bytes.Buffer{}})
 
30
}
 
31
 
 
32
func (s *AddUnitSuite) TestAddUnit(c *C) {
 
33
        testing.Charms.BundlePath(s.seriesPath, "dummy")
 
34
        err := runDeploy(c, "local:dummy", "some-service-name")
 
35
        c.Assert(err, IsNil)
 
36
        curl := charm.MustParseURL("local:precise/dummy-1")
 
37
        s.assertService(c, "some-service-name", curl, 1, 0)
 
38
 
 
39
        err = runAddUnit(c, "some-service-name")
 
40
        c.Assert(err, IsNil)
 
41
        s.assertService(c, "some-service-name", curl, 2, 0)
 
42
 
 
43
        err = runAddUnit(c, "--num-units", "2", "some-service-name")
 
44
        c.Assert(err, IsNil)
 
45
        s.assertService(c, "some-service-name", curl, 4, 0)
 
46
}