environs/ec2: test that we can connect to zookeeper after bootstrap.

environs/jujutest/test.go

 package jujutest
 
 import (
         . "launchpad.net/gocheck"
         "launchpad.net/juju/go/environs"
         "launchpad.net/juju/go/state"
         "time"
 )
 
 // InvalidStateInfo holds information about no state - it will always give
@@ skipping 36 matching lines @@
         t.Env = nil
 }
 
 // LiveTests contains tests that are designed to run against a live server
 // (e.g. Amazon EC2).  The Environ is opened once only for all the tests
 // in the suite, stored in Env, and Destroyed after the suite has completed.
 type LiveTests struct {
         Environs *environs.Environs
         Name     string
         Env      environs.Environ
-        // length of time to wait before environ becomes logically consistent.
+        // ConsistencyDelay gives the length of time to wait before
+        // environ becomes logically consistent.
         ConsistencyDelay time.Duration
+
+        // CanOpenState should be true if the testing environment allows
+        // the state to be opened after bootstrapping.
+        CanOpenState bool
+        bootstrapped bool
 }
 
 func (t *LiveTests) SetUpSuite(c *C) {
         e, err := t.Environs.Open(t.Name)
         c.Assert(err, IsNil, Commentf("opening environ %q", t.Name))
         c.Assert(e, NotNil)
         t.Env = e
 }
 
+func (t *LiveTests) BootstrapOnce(c *C) {
+        if t.bootstrapped {
+                return
+        }
+        err := t.Env.Bootstrap()
+        c.Assert(err, IsNil)
+        t.bootstrapped = true
+}
+
+func (t *LiveTests) Destroy(c *C) {
+        err := t.Env.Destroy(nil)
+        c.Assert(err, IsNil)
+        t.bootstrapped = false
+}
+
 func (t *LiveTests) TearDownSuite(c *C) {
         err := t.Env.Destroy(nil)
         c.Check(err, IsNil)
         t.Env = nil
 }
 
 func (t *LiveTests) SetUpTest(*C) {
 }
 
 func (t *LiveTests) TearDownTest(*C) {
 }