github.com/cloud-green/juju@v0.0.0-20151002100041-a00291338d3d/cmd/juju/storage/list_test.go

// Copyright 2015 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.

package storage_test

import (
	"github.com/juju/cmd"
	jc "github.com/juju/testing/checkers"
	gc "gopkg.in/check.v1"

	"github.com/juju/juju/apiserver/params"
	"github.com/juju/juju/cmd/envcmd"
	"github.com/juju/juju/cmd/juju/storage"
	_ "github.com/juju/juju/provider/dummy"
	"github.com/juju/juju/testing"
)

type ListSuite struct {
	SubStorageSuite
	mockAPI *mockListAPI
}

var _ = gc.Suite(&ListSuite{})

func (s *ListSuite) SetUpTest(c *gc.C) {
	s.SubStorageSuite.SetUpTest(c)

	s.mockAPI = &mockListAPI{}
	s.PatchValue(storage.GetStorageListAPI, func(c *storage.ListCommand) (storage.StorageListAPI, error) {
		return s.mockAPI, nil
	})

}

func runList(c *gc.C, args []string) (*cmd.Context, error) {
	return testing.RunCommand(c, envcmd.Wrap(&storage.ListCommand{}), args...)
}

func (s *ListSuite) TestList(c *gc.C) {
	s.assertValidList(
		c,
		nil,
		// Default format is tabular
		`
\[Storage\]    
UNIT         ID          LOCATION STATUS   MESSAGE 
postgresql/0 db-dir/1100 hither   attached         
transcode/0  db-dir/1000          pending          
transcode/0  shared-fs/0 there    attached         
transcode/1  shared-fs/0 here     attached         

`[1:],
		"",
	)
}

func (s *ListSuite) TestListYAML(c *gc.C) {
	s.assertValidList(
		c,
		[]string{"--format", "yaml"},
		`
storage:
  db-dir/1000:
    kind: block
    status:
      current: pending
      since: .*
    persistent: false
    attachments:
      units:
        transcode/0: {}
  db-dir/1100:
    kind: block
    status:
      current: attached
      since: .*
    persistent: true
    attachments:
      units:
        postgresql/0:
          location: hither
  shared-fs/0:
    kind: filesystem
    status:
      current: attached
      since: .*
    persistent: true
    attachments:
      units:
        transcode/0:
          location: there
        transcode/1:
          location: here
`[1:],
		"",
	)
}

func (s *ListSuite) TestListOwnerStorageIdSort(c *gc.C) {
	s.mockAPI.includeErrors = true
	s.assertValidList(
		c,
		nil,
		// Default format is tabular
		`
\[Storage\]    
UNIT         ID          LOCATION STATUS   MESSAGE 
postgresql/0 db-dir/1100 hither   attached         
transcode/0  db-dir/1000          pending          
transcode/0  shared-fs/0 there    attached         
transcode/1  shared-fs/0 here     attached         

`[1:],
		"error for storage-db-dir-1010\n",
	)
}

func (s *ListSuite) assertValidList(c *gc.C, args []string, expectedValid, expectedErr string) {
	context, err := runList(c, args)
	c.Assert(err, jc.ErrorIsNil)

	obtainedErr := testing.Stderr(context)
	c.Assert(obtainedErr, gc.Matches, expectedErr)

	obtainedValid := testing.Stdout(context)
	c.Assert(obtainedValid, gc.Matches, expectedValid)
}

type mockListAPI struct {
	includeErrors bool
}

func (s mockListAPI) Close() error {
	return nil
}

func (s mockListAPI) List() ([]params.StorageDetailsResult, error) {
	// postgresql/0 has "db-dir/1100"
	// transcode/1 has "db-dir/1000"
	// transcode/0 and transcode/1 share "shared-fs/0"
	//
	// there is also a storage instance "db-dir/1010" which
	// returns an error when listed.
	results := []params.StorageDetailsResult{{
		Legacy: params.LegacyStorageDetails{
			StorageTag: "storage-db-dir-1000",
			OwnerTag:   "unit-transcode-0",
			UnitTag:    "unit-transcode-0",
			Kind:       params.StorageKindBlock,
			Status:     "pending",
		},
	}, {
		Result: &params.StorageDetails{
			StorageTag: "storage-db-dir-1100",
			OwnerTag:   "unit-postgresql-0",
			Kind:       params.StorageKindBlock,
			Status: params.EntityStatus{
				Status: params.StatusAttached,
				Since:  &epoch,
			},
			Persistent: true,
			Attachments: map[string]params.StorageAttachmentDetails{
				"unit-postgresql-0": params.StorageAttachmentDetails{
					Location: "hither",
				},
			},
		},
	}, {
		Result: &params.StorageDetails{
			StorageTag: "storage-shared-fs-0",
			OwnerTag:   "service-transcode",
			Kind:       params.StorageKindFilesystem,
			Status: params.EntityStatus{
				Status: params.StatusAttached,
				Since:  &epoch,
			},
			Persistent: true,
			Attachments: map[string]params.StorageAttachmentDetails{
				"unit-transcode-0": params.StorageAttachmentDetails{
					Location: "there",
				},
				"unit-transcode-1": params.StorageAttachmentDetails{
					Location: "here",
				},
			},
		},
	}}
	if s.includeErrors {
		results = append(results, params.StorageDetailsResult{
			Error: &params.Error{
				Message: "error for storage-db-dir-1010",
			},
		})
	}
	return results, nil
}