Fail when run against Python juju environment

environs/jujutest/livetests.go

 // Copyright 2011, 2012, 2013 Canonical Ltd.
 // Licensed under the AGPLv3, see LICENCE file for details.
 
 package jujutest
 
 import (
         "bytes"
         "fmt"
         "io"
         "io/ioutil"
         . "launchpad.net/gocheck"
         "launchpad.net/juju-core/charm"
         "launchpad.net/juju-core/constraints"
         "launchpad.net/juju-core/environs"
         "launchpad.net/juju-core/environs/config"
         "launchpad.net/juju-core/environs/tools"
         "launchpad.net/juju-core/errors"
         "launchpad.net/juju-core/instance"
         "launchpad.net/juju-core/juju"
         "launchpad.net/juju-core/juju/testing"
         "launchpad.net/juju-core/state"
         "launchpad.net/juju-core/state/api"
         coretesting "launchpad.net/juju-core/testing"
         "launchpad.net/juju-core/utils"
         "launchpad.net/juju-core/version"
+        "strings"
         "time"
 )
 
 // 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 {
         coretesting.LoggingSuite
 
         // TestConfig contains the configuration attributes for opening an environment.
@@ skipping 428 matching lines @@
         storage := environ.Storage()
         reader, err := storage.Get("bootstrap-verify")
         c.Assert(err, IsNil)
         defer reader.Close()
         contents, err := ioutil.ReadAll(reader)
         c.Assert(err, IsNil)
         c.Check(string(contents), Equals,
                 "juju-core storage writing verified: ok\n")
 }
 
+func restoreBootstrapVerificationFile(c *C, storage environs.Storage) {
+        content := "juju-core storage writing verified: ok\n"
+        contentReader := strings.NewReader(content)
+        err := storage.Put("bootstrap-verify", contentReader,
+                int64(len(content)))
+        c.Assert(err, IsNil)
+}
+
+func (t *LiveTests) TestCheckEnvironmentOnConnect(c *C) {
+        // When new connection is established to a bootstraped environment,
+        // it is checked that we are running against a juju-core environment.
+        t.BootstrapOnce(c)
+
+        conn, err := juju.NewConn(t.Env)
+        c.Assert(err, IsNil)
+        conn.Close()
+}
+
+func (t *LiveTests) TestCheckEnvironmentOnConnectNoVerificationFile(c *C) {
+        // When new connection is established to a bootstraped environment,
+        // it is checked that we are running against a juju-core environment.
+        //
+        // Absence of a verification file means it is a juju-core environment
+        // with an older version, which is fine.
+        t.BootstrapOnce(c)
+        environ := t.Env
+        storage := environ.Storage()
+        err := storage.Remove("bootstrap-verify")
+        c.Assert(err, IsNil)
+        defer restoreBootstrapVerificationFile(c, storage)
+
+        conn, err := juju.NewConn(t.Env)
+        c.Assert(err, IsNil)
+        conn.Close()
+}
+
+func (t *LiveTests) TestCheckEnvironmentOnConnectBadVerificationFile(c *C) {
+        // When new connection is established to a bootstraped environment,
+        // it is checked that we are running against a juju-core environment.
+        //
+        // If the verification file has unexpected content, it is not
+        // a juju-core environment (likely to a Python juju environment).
+        t.BootstrapOnce(c)
+        environ := t.Env
+        storage := environ.Storage()
+
+        // Finally, replace the content with an arbitrary string.
+        badVerificationContent := "bootstrap storage verification"
+        reader := strings.NewReader(badVerificationContent)
+        err := storage.Put(
+                "bootstrap-verify",
+                reader,
+                int64(len(badVerificationContent)))
+        c.Assert(err, IsNil)
+        defer restoreBootstrapVerificationFile(c, storage)
+
+        // Running NewConn() should fail.
+        _, err = juju.NewConn(t.Env)
+        c.Assert(err, Equals, environs.InvalidEnvironmentError)
+}
+
 type tooler interface {
         Life() state.Life
         AgentTools() (*state.Tools, error)
         Refresh() error
         String() string
 }
 
 type watcher interface {
         Stop() error
         Err() error
@@ skipping 328 matching lines @@
                 return err
         }
         var buf bytes.Buffer
         _, err = io.Copy(&buf, rc)
         rc.Close()
         if err != nil {
                 return err
         }
         return target.Put(targetPath, &buf, int64(buf.Len()))
 }