state: adding of FlagWatcher and integration into Service

state/watcher_test.go

 package state_test
 
 import (
         . "launchpad.net/gocheck"
         "launchpad.net/gozk/zookeeper"
         "launchpad.net/juju-core/state"
         "time"
 )
 
 var serviceWatchConfigData = []map[string]interface{}{
@@ skipping 50 matching lines @@
         select {
         case got, _ := <-configWatcher.Changes():
                 c.Fatalf("got unexpected change: %#v", got)
         case <-time.After(100 * time.Millisecond):
         }
 
         err = configWatcher.Stop()
         c.Assert(err, IsNil)
 }
 
+var serviceExposedTests = []struct {
+        test func(s *state.Service) error
+        want bool
+}{
+        {func(s *state.Service) error { return nil }, false},
+        {func(s *state.Service) error { return s.SetExposed() }, true},
+        {func(s *state.Service) error { return s.ClearExposed() }, false},
+        {func(s *state.Service) error { return s.SetExposed() }, true},
+}
+
+func (s *StateSuite) TestServiceExposedConfig(c *C) {
+        dummy := s.addDummyCharm(c)
+        wordpress, err := s.st.AddService("wordpress", dummy)
+        c.Assert(err, IsNil)
+        c.Assert(wordpress.Name(), Equals, "wordpress")
+
+        exposed, err := wordpress.IsExposed()
+        c.Assert(err, IsNil)
+        c.Assert(exposed, Equals, false)
+        exposedWatcher := wordpress.WatchExposed()
+
+        for i, test := range serviceExposedTests {
+                c.Logf("test %d", i)
+                err := test.test(wordpress)
+                c.Assert(err, IsNil)
+                select {
+                case got, ok := <-exposedWatcher.Changes():
+                        c.Assert(ok, Equals, true)
+                        c.Assert(got, Equals, test.want)
+                case <-time.After(200 * time.Millisecond):
+                        c.Fatalf("didn't get change: %#v", test.want)
+                }
+        }
+
+        select {
+        case got, _ := <-exposedWatcher.Changes():
+                c.Fatalf("got unexpected change: %#v", got)
+        case <-time.After(100 * time.Millisecond):
+        }
+
+        err = exposedWatcher.Stop()
+        c.Assert(err, IsNil)
+}
+
 func (s *StateSuite) TestServiceWatchConfigIllegalData(c *C) {
         dummy := s.addDummyCharm(c)
         wordpress, err := s.st.AddService("wordpress", dummy)
         c.Assert(err, IsNil)
         c.Assert(wordpress.Name(), Equals, "wordpress")
         configWatcher := wordpress.WatchConfig()
 
         // Receive empty change after service adding.
         select {
         case got, ok := <-configWatcher.Changes():
@@ skipping 372 matching lines @@
         {[]string{"B"}, []string{"A", "B", "C"}, nil},
         {[]string{"A", "D", "C"}, []string{}, []string{"A", "D", "C"}},
         {[]string{}, []string{"A", "D", "C"}, nil},
 }
 
 func (*StateSuite) TestDiff(c *C) {
         for _, test := range diffTests {
                 c.Assert(test.want, DeepEquals, state.Diff(test.A, test.B))
         }
 }