github.com/Pankov404/juju@v0.0.0-20150703034450-be266991dceb/upgrades/storage_test.go (about)

     1  // Copyright 2015 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package upgrades_test
     5  
     6  import (
     7  	jc "github.com/juju/testing/checkers"
     8  	gc "gopkg.in/check.v1"
     9  
    10  	jujutesting "github.com/juju/juju/juju/testing"
    11  	"github.com/juju/juju/provider/ec2"
    12  	"github.com/juju/juju/state"
    13  	"github.com/juju/juju/storage/poolmanager"
    14  	"github.com/juju/juju/upgrades"
    15  )
    16  
    17  type defaultStoragePoolsSuite struct {
    18  	jujutesting.JujuConnSuite
    19  }
    20  
    21  var _ = gc.Suite(&defaultStoragePoolsSuite{})
    22  
    23  func (s *defaultStoragePoolsSuite) TestDefaultStoragePools(c *gc.C) {
    24  	err := upgrades.AddDefaultStoragePools(s.State)
    25  	settings := state.NewStateSettings(s.State)
    26  	err = poolmanager.AddDefaultStoragePools(settings)
    27  	c.Assert(err, jc.ErrorIsNil)
    28  	pm := poolmanager.New(settings)
    29  	for _, pName := range []string{"ebs-ssd"} {
    30  		p, err := pm.Get(pName)
    31  		c.Assert(err, jc.ErrorIsNil)
    32  		c.Assert(p.Provider(), gc.Equals, ec2.EBS_ProviderType)
    33  	}
    34  }