attacher_test.go 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289
  1. /*
  2. Copyright 2016 The Kubernetes Authors.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package awsebs
  14. import (
  15. "errors"
  16. "testing"
  17. "k8s.io/klog"
  18. "k8s.io/api/core/v1"
  19. "k8s.io/apimachinery/pkg/api/resource"
  20. "k8s.io/apimachinery/pkg/types"
  21. "k8s.io/kubernetes/pkg/volume"
  22. volumetest "k8s.io/kubernetes/pkg/volume/testing"
  23. "k8s.io/legacy-cloud-providers/aws"
  24. )
  25. func TestGetVolumeName_Volume(t *testing.T) {
  26. plugin := newPlugin()
  27. name := aws.KubernetesVolumeID("my-aws-volume")
  28. spec := createVolSpec(name, false)
  29. volumeName, err := plugin.GetVolumeName(spec)
  30. if err != nil {
  31. t.Errorf("GetVolumeName error: %v", err)
  32. }
  33. if volumeName != string(name) {
  34. t.Errorf("GetVolumeName error: expected %s, got %s", name, volumeName)
  35. }
  36. }
  37. func TestGetVolumeName_PersistentVolume(t *testing.T) {
  38. plugin := newPlugin()
  39. name := aws.KubernetesVolumeID("my-aws-pv")
  40. spec := createPVSpec(name, true)
  41. volumeName, err := plugin.GetVolumeName(spec)
  42. if err != nil {
  43. t.Errorf("GetVolumeName error: %v", err)
  44. }
  45. if volumeName != string(name) {
  46. t.Errorf("GetVolumeName error: expected %s, got %s", name, volumeName)
  47. }
  48. }
  49. // One testcase for TestAttachDetach table test below
  50. type testcase struct {
  51. name aws.KubernetesVolumeID
  52. // For fake AWS:
  53. attach attachCall
  54. detach detachCall
  55. t *testing.T
  56. // Actual test to run
  57. test func(test *testcase) (string, error)
  58. // Expected return of the test
  59. expectedDevice string
  60. expectedError error
  61. }
  62. func TestAttachDetach(t *testing.T) {
  63. diskName := aws.KubernetesVolumeID("disk")
  64. nodeName := types.NodeName("instance")
  65. spec := createVolSpec(diskName, false)
  66. attachError := errors.New("Fake attach error")
  67. detachError := errors.New("Fake detach error")
  68. tests := []testcase{
  69. // Successful Attach call
  70. {
  71. name: "Attach_Positive",
  72. attach: attachCall{diskName, nodeName, "/dev/sda", nil},
  73. test: func(testcase *testcase) (string, error) {
  74. attacher := newAttacher(testcase)
  75. return attacher.Attach(spec, nodeName)
  76. },
  77. expectedDevice: "/dev/sda",
  78. },
  79. // Attach call fails
  80. {
  81. name: "Attach_Negative",
  82. attach: attachCall{diskName, nodeName, "", attachError},
  83. test: func(testcase *testcase) (string, error) {
  84. attacher := newAttacher(testcase)
  85. return attacher.Attach(spec, nodeName)
  86. },
  87. expectedError: attachError,
  88. },
  89. // Detach succeeds
  90. {
  91. name: "Detach_Positive",
  92. detach: detachCall{diskName, nodeName, "/dev/sda", nil},
  93. test: func(testcase *testcase) (string, error) {
  94. detacher := newDetacher(testcase)
  95. mountPath := "/mnt/" + string(diskName)
  96. return "", detacher.Detach(mountPath, nodeName)
  97. },
  98. },
  99. // Detach fails
  100. {
  101. name: "Detach_Negative",
  102. detach: detachCall{diskName, nodeName, "", detachError},
  103. test: func(testcase *testcase) (string, error) {
  104. detacher := newDetacher(testcase)
  105. mountPath := "/mnt/" + string(diskName)
  106. return "", detacher.Detach(mountPath, nodeName)
  107. },
  108. expectedError: detachError,
  109. },
  110. }
  111. for _, testcase := range tests {
  112. testcase.t = t
  113. device, err := testcase.test(&testcase)
  114. if err != testcase.expectedError {
  115. t.Errorf("%s failed: expected err=%q, got %q", testcase.name, testcase.expectedError.Error(), err.Error())
  116. }
  117. if device != testcase.expectedDevice {
  118. t.Errorf("%s failed: expected device=%q, got %q", testcase.name, testcase.expectedDevice, device)
  119. }
  120. t.Logf("Test %q succeeded", testcase.name)
  121. }
  122. }
  123. // newPlugin creates a new gcePersistentDiskPlugin with fake cloud, NewAttacher
  124. // and NewDetacher won't work.
  125. func newPlugin() *awsElasticBlockStorePlugin {
  126. host := volumetest.NewFakeVolumeHost("/tmp", nil, nil)
  127. plugins := ProbeVolumePlugins()
  128. plugin := plugins[0]
  129. plugin.Init(host)
  130. return plugin.(*awsElasticBlockStorePlugin)
  131. }
  132. func newAttacher(testcase *testcase) *awsElasticBlockStoreAttacher {
  133. return &awsElasticBlockStoreAttacher{
  134. host: nil,
  135. awsVolumes: testcase,
  136. }
  137. }
  138. func newDetacher(testcase *testcase) *awsElasticBlockStoreDetacher {
  139. return &awsElasticBlockStoreDetacher{
  140. awsVolumes: testcase,
  141. }
  142. }
  143. func createVolSpec(name aws.KubernetesVolumeID, readOnly bool) *volume.Spec {
  144. return &volume.Spec{
  145. Volume: &v1.Volume{
  146. VolumeSource: v1.VolumeSource{
  147. AWSElasticBlockStore: &v1.AWSElasticBlockStoreVolumeSource{
  148. VolumeID: string(name),
  149. ReadOnly: readOnly,
  150. },
  151. },
  152. },
  153. }
  154. }
  155. func createPVSpec(name aws.KubernetesVolumeID, readOnly bool) *volume.Spec {
  156. return &volume.Spec{
  157. PersistentVolume: &v1.PersistentVolume{
  158. Spec: v1.PersistentVolumeSpec{
  159. PersistentVolumeSource: v1.PersistentVolumeSource{
  160. AWSElasticBlockStore: &v1.AWSElasticBlockStoreVolumeSource{
  161. VolumeID: string(name),
  162. ReadOnly: readOnly,
  163. },
  164. },
  165. },
  166. },
  167. }
  168. }
  169. // Fake AWS implementation
  170. type attachCall struct {
  171. diskName aws.KubernetesVolumeID
  172. nodeName types.NodeName
  173. retDeviceName string
  174. ret error
  175. }
  176. type detachCall struct {
  177. diskName aws.KubernetesVolumeID
  178. nodeName types.NodeName
  179. retDeviceName string
  180. ret error
  181. }
  182. func (testcase *testcase) AttachDisk(diskName aws.KubernetesVolumeID, nodeName types.NodeName) (string, error) {
  183. expected := &testcase.attach
  184. if expected.diskName == "" && expected.nodeName == "" {
  185. // testcase.attach looks uninitialized, test did not expect to call
  186. // AttachDisk
  187. testcase.t.Errorf("Unexpected AttachDisk call!")
  188. return "", errors.New("unexpected AttachDisk call")
  189. }
  190. if expected.diskName != diskName {
  191. testcase.t.Errorf("Unexpected AttachDisk call: expected diskName %s, got %s", expected.diskName, diskName)
  192. return "", errors.New("Unexpected AttachDisk call: wrong diskName")
  193. }
  194. if expected.nodeName != nodeName {
  195. testcase.t.Errorf("Unexpected AttachDisk call: expected nodeName %s, got %s", expected.nodeName, nodeName)
  196. return "", errors.New("Unexpected AttachDisk call: wrong nodeName")
  197. }
  198. klog.V(4).Infof("AttachDisk call: %s, %s, returning %q, %v", diskName, nodeName, expected.retDeviceName, expected.ret)
  199. return expected.retDeviceName, expected.ret
  200. }
  201. func (testcase *testcase) DetachDisk(diskName aws.KubernetesVolumeID, nodeName types.NodeName) (string, error) {
  202. expected := &testcase.detach
  203. if expected.diskName == "" && expected.nodeName == "" {
  204. // testcase.detach looks uninitialized, test did not expect to call
  205. // DetachDisk
  206. testcase.t.Errorf("Unexpected DetachDisk call!")
  207. return "", errors.New("unexpected DetachDisk call")
  208. }
  209. if expected.diskName != diskName {
  210. testcase.t.Errorf("Unexpected DetachDisk call: expected diskName %s, got %s", expected.diskName, diskName)
  211. return "", errors.New("Unexpected DetachDisk call: wrong diskName")
  212. }
  213. if expected.nodeName != nodeName {
  214. testcase.t.Errorf("Unexpected DetachDisk call: expected nodeName %s, got %s", expected.nodeName, nodeName)
  215. return "", errors.New("Unexpected DetachDisk call: wrong nodeName")
  216. }
  217. klog.V(4).Infof("DetachDisk call: %s, %s, returning %q, %v", diskName, nodeName, expected.retDeviceName, expected.ret)
  218. return expected.retDeviceName, expected.ret
  219. }
  220. func (testcase *testcase) DiskIsAttached(diskName aws.KubernetesVolumeID, nodeName types.NodeName) (bool, error) {
  221. // DetachDisk no longer relies on DiskIsAttached api call
  222. return false, nil
  223. }
  224. func (testcase *testcase) DisksAreAttached(nodeDisks map[types.NodeName][]aws.KubernetesVolumeID) (map[types.NodeName]map[aws.KubernetesVolumeID]bool, error) {
  225. return nil, errors.New("Not implemented")
  226. }
  227. func (testcase *testcase) CreateDisk(volumeOptions *aws.VolumeOptions) (volumeName aws.KubernetesVolumeID, err error) {
  228. return "", errors.New("Not implemented")
  229. }
  230. func (testcase *testcase) DeleteDisk(volumeName aws.KubernetesVolumeID) (bool, error) {
  231. return false, errors.New("Not implemented")
  232. }
  233. func (testcase *testcase) GetVolumeLabels(volumeName aws.KubernetesVolumeID) (map[string]string, error) {
  234. return map[string]string{}, errors.New("Not implemented")
  235. }
  236. func (testcase *testcase) GetDiskPath(volumeName aws.KubernetesVolumeID) (string, error) {
  237. return "", errors.New("Not implemented")
  238. }
  239. func (testcase *testcase) ResizeDisk(
  240. volumeName aws.KubernetesVolumeID,
  241. oldSize resource.Quantity,
  242. newSize resource.Quantity) (resource.Quantity, error) {
  243. return oldSize, errors.New("Not implemented")
  244. }