#1322 add tests for volumen set notes

Author: ATGE

tests/CLI/modules/block_tests.py

@@ -64,6 +64,7 @@ def test_volume_detail(self):
         self.assertEqual({
             'Username': 'username',
             'LUN Id': '2',
+            'Notes': "{'status': 'available'}",
             'Endurance Tier': 'READHEAVY_TIER',
             'IOPs': 1000,
             'Snapshot Capacity (GB)': '10',
@@ -132,6 +133,7 @@ def test_volume_list(self):
                 'iops': None,
                 'ip_addr': '10.1.2.3',
                 'lunId': None,
+                'notes': "{'status': 'availabl",
                 'rep_partner_count': None,
                 'storage_type': 'ENDURANCE',
                 'username': 'username',
@@ -726,3 +728,21 @@ def test_dep_dupe_convert(self):
         result = self.run_command(['block', 'volume-convert', '102'])
 
         self.assert_no_fail(result)
+
+    @mock.patch('SoftLayer.BlockStorageManager.volume_set_note')
+    def test_volume_set_note(self, set_note):
+        set_note.return_value = True
+
+        result = self.run_command(['block', 'volume-set-note', '102', '--note=testing'])
+
+        self.assert_no_fail(result)
+        self.assertIn("successfully!", result.output)
+
+    @mock.patch('SoftLayer.BlockStorageManager.volume_set_note')
+    def test_volume_not_set_note(self, set_note):
+        set_note.return_value = False
+
+        result = self.run_command(['block', 'volume-set-note', '102', '--note=testing'])
+
+        self.assert_no_fail(result)
+        self.assertIn("Note could not be set!", result.output)

tests/CLI/modules/file_tests.py

@@ -50,6 +50,7 @@ def test_volume_list(self):
                 'username': 'user',
                 'active_transactions': None,
                 'mount_addr': '127.0.0.1:/TEST',
+                'notes': None,
                 'rep_partner_count': None
             }],
             json.loads(result.output))
@@ -137,6 +138,7 @@ def test_volume_detail(self):
             'Data Center': 'dal05',
             'Type': 'ENDURANCE',
             'ID': 100,
+            'Notes': "{'status': 'available'}",
             '# of Active Transactions': '1',
             'Ongoing Transaction': 'This is a buffer time in which the customer may cancel the server',
             'Replicant Count': '1',
@@ -705,3 +707,21 @@ def test_dep_dupe_convert(self):
         result = self.run_command(['file', 'volume-convert', '102'])
 
         self.assert_no_fail(result)
+
+    @mock.patch('SoftLayer.FileStorageManager.volume_set_note')
+    def test_volume_set_note(self, set_note):
+        set_note.return_value = True
+
+        result = self.run_command(['file', 'volume-set-note', '102', '--note=testing'])
+
+        self.assert_no_fail(result)
+        self.assertIn("successfully!", result.output)
+
+    @mock.patch('SoftLayer.FileStorageManager.volume_set_note')
+    def test_volume_not_set_note(self, set_note):
+        set_note.return_value = False
+
+        result = self.run_command(['file', 'volume-set-note', '102', '--note=testing'])
+
+        self.assert_no_fail(result)
+        self.assertIn("Note could not be set!", result.output)

tests/managers/block_tests.py

@@ -111,7 +111,8 @@ def test_get_block_volume_details(self):
                         'replicationPartners[id,username,' \
                         'serviceResourceBackendIpAddress,' \
                         'serviceResource[datacenter[name]],' \
-                        'replicationSchedule[type[keyname]]]'
+                        'replicationSchedule[type[keyname]]],' \
+                        'notes'
 
         self.assert_called_with(
             'SoftLayer_Network_Storage',

tests/managers/file_tests.py

@@ -145,7 +145,8 @@ def test_get_file_volume_details(self):
                         'replicationPartners[id,username,'\
                         'serviceResourceBackendIpAddress,'\
                         'serviceResource[datacenter[name]],'\
-                        'replicationSchedule[type[keyname]]]'
+                        'replicationSchedule[type[keyname]]],' \
+                        'notes'
 
         self.assert_called_with(
             'SoftLayer_Network_Storage',