Index: state/unit_test.go |
=== modified file 'state/unit_test.go' |
--- state/unit_test.go 2012-09-24 21:31:51 +0000 |
+++ state/unit_test.go 2012-09-25 08:28:02 +0000 |
@@ -355,6 +355,13 @@ |
} |
func (s *UnitSuite) TestWatchUnit(c *C) { |
+ altunit, err := s.State.Unit(s.unit.Name()) |
+ c.Assert(err, IsNil) |
+ err = altunit.SetPublicAddress("newer-address") |
+ c.Assert(err, IsNil) |
+ _, err = s.unit.PublicAddress() |
+ c.Assert(err, ErrorMatches, `public address of unit ".*" not found`) |
+ |
w := s.unit.Watch() |
defer func() { |
c.Assert(w.Stop(), IsNil) |
@@ -364,6 +371,9 @@ |
case u, ok := <-w.Changes(): |
c.Assert(ok, Equals, true) |
c.Assert(u.Name(), Equals, s.unit.Name()) |
+ addr, err := u.PublicAddress() |
+ c.Assert(err, IsNil) |
+ c.Assert(addr, Equals, "newer-address") |
case <-time.After(500 * time.Millisecond): |
c.Fatalf("did not get change: %v", s.unit) |
} |