github.com/axw/juju@v0.0.0-20161005053422-4bd6544d08d4/apiserver/common/storagecommon/mock_test.go (about)

     1  // Copyright 2015 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package storagecommon_test
     5  
     6  import (
     7  	"github.com/juju/errors"
     8  	"github.com/juju/testing"
     9  	"gopkg.in/juju/names.v2"
    10  
    11  	"github.com/juju/juju/apiserver/common/storagecommon"
    12  	"github.com/juju/juju/state"
    13  	"github.com/juju/juju/storage"
    14  	"github.com/juju/juju/storage/poolmanager"
    15  )
    16  
    17  type fakeStorage struct {
    18  	testing.Stub
    19  	storagecommon.StorageInterface
    20  	storageInstance        func(names.StorageTag) (state.StorageInstance, error)
    21  	storageInstanceVolume  func(names.StorageTag) (state.Volume, error)
    22  	volumeAttachment       func(names.MachineTag, names.VolumeTag) (state.VolumeAttachment, error)
    23  	blockDevices           func(names.MachineTag) ([]state.BlockDeviceInfo, error)
    24  	watchVolumeAttachment  func(names.MachineTag, names.VolumeTag) state.NotifyWatcher
    25  	watchBlockDevices      func(names.MachineTag) state.NotifyWatcher
    26  	watchStorageAttachment func(names.StorageTag, names.UnitTag) state.NotifyWatcher
    27  }
    28  
    29  func (s *fakeStorage) StorageInstance(tag names.StorageTag) (state.StorageInstance, error) {
    30  	s.MethodCall(s, "StorageInstance", tag)
    31  	return s.storageInstance(tag)
    32  }
    33  
    34  func (s *fakeStorage) StorageInstanceVolume(tag names.StorageTag) (state.Volume, error) {
    35  	s.MethodCall(s, "StorageInstanceVolume", tag)
    36  	return s.storageInstanceVolume(tag)
    37  }
    38  
    39  func (s *fakeStorage) VolumeAttachment(m names.MachineTag, v names.VolumeTag) (state.VolumeAttachment, error) {
    40  	s.MethodCall(s, "VolumeAttachment", m, v)
    41  	return s.volumeAttachment(m, v)
    42  }
    43  
    44  func (s *fakeStorage) BlockDevices(m names.MachineTag) ([]state.BlockDeviceInfo, error) {
    45  	s.MethodCall(s, "BlockDevices", m)
    46  	return s.blockDevices(m)
    47  }
    48  
    49  func (s *fakeStorage) WatchVolumeAttachment(m names.MachineTag, v names.VolumeTag) state.NotifyWatcher {
    50  	s.MethodCall(s, "WatchVolumeAttachment", m, v)
    51  	return s.watchVolumeAttachment(m, v)
    52  }
    53  
    54  func (s *fakeStorage) WatchBlockDevices(m names.MachineTag) state.NotifyWatcher {
    55  	s.MethodCall(s, "WatchBlockDevices", m)
    56  	return s.watchBlockDevices(m)
    57  }
    58  
    59  func (s *fakeStorage) WatchStorageAttachment(st names.StorageTag, u names.UnitTag) state.NotifyWatcher {
    60  	s.MethodCall(s, "WatchStorageAttachment", st, u)
    61  	return s.watchStorageAttachment(st, u)
    62  }
    63  
    64  type fakeStorageInstance struct {
    65  	state.StorageInstance
    66  	tag   names.StorageTag
    67  	owner names.Tag
    68  	kind  state.StorageKind
    69  }
    70  
    71  func (i *fakeStorageInstance) StorageTag() names.StorageTag {
    72  	return i.tag
    73  }
    74  
    75  func (i *fakeStorageInstance) Tag() names.Tag {
    76  	return i.tag
    77  }
    78  
    79  func (i *fakeStorageInstance) Owner() names.Tag {
    80  	return i.owner
    81  }
    82  
    83  func (i *fakeStorageInstance) Kind() state.StorageKind {
    84  	return i.kind
    85  }
    86  
    87  type fakeStorageAttachment struct {
    88  	state.StorageAttachment
    89  	storageTag names.StorageTag
    90  }
    91  
    92  func (a *fakeStorageAttachment) StorageInstance() names.StorageTag {
    93  	return a.storageTag
    94  }
    95  
    96  type fakeVolume struct {
    97  	state.Volume
    98  	tag    names.VolumeTag
    99  	params *state.VolumeParams
   100  	info   *state.VolumeInfo
   101  }
   102  
   103  func (v *fakeVolume) VolumeTag() names.VolumeTag {
   104  	return v.tag
   105  }
   106  
   107  func (v *fakeVolume) Tag() names.Tag {
   108  	return v.tag
   109  }
   110  
   111  func (v *fakeVolume) Params() (state.VolumeParams, bool) {
   112  	if v.params == nil {
   113  		return state.VolumeParams{}, false
   114  	}
   115  	return *v.params, true
   116  }
   117  
   118  func (v *fakeVolume) Info() (state.VolumeInfo, error) {
   119  	if v.info == nil {
   120  		return state.VolumeInfo{}, errors.NotProvisionedf("volume %v", v.tag.Id())
   121  	}
   122  	return *v.info, nil
   123  }
   124  
   125  type fakeVolumeAttachment struct {
   126  	state.VolumeAttachment
   127  	info *state.VolumeAttachmentInfo
   128  }
   129  
   130  func (v *fakeVolumeAttachment) Info() (state.VolumeAttachmentInfo, error) {
   131  	if v.info == nil {
   132  		return state.VolumeAttachmentInfo{}, errors.NotProvisionedf("volume attachment")
   133  	}
   134  	return *v.info, nil
   135  }
   136  
   137  type fakePoolManager struct {
   138  	poolmanager.PoolManager
   139  }
   140  
   141  func (pm *fakePoolManager) Get(name string) (*storage.Config, error) {
   142  	return nil, errors.NotFoundf("pool")
   143  }
   144  
   145  type nopSyncStarter struct{}
   146  
   147  func (nopSyncStarter) StartSync() {}