diff --git a/src/rockstor/storageadmin/tests/test_disk_smart.py b/src/rockstor/storageadmin/tests/test_disk_smart.py index 445ed39db..2fd7eff2c 100644 --- a/src/rockstor/storageadmin/tests/test_disk_smart.py +++ b/src/rockstor/storageadmin/tests/test_disk_smart.py @@ -30,7 +30,9 @@ class DiskSmartTests(APITestMixin, APITestCase): @classmethod def setUpClass(cls): super(DiskSmartTests, cls).setUpClass() - + + cls.patch_info = patch('storageadmin.views.disk_smart_info') + cls.mock_info = cls.patch_info.start() @classmethod def tearDownClass(cls): @@ -39,7 +41,12 @@ def tearDownClass(cls): def test_get(self): # get base URL - response = self.client.get('%s/info' % self.BASE_URL) + response = self.client.get('%s' % self.BASE_URL) + self.assertEqual(response.status_code, + status.HTTP_200_OK, msg=response.data) + + # get with disk name + response = self.client.get('%s/sdd' % self.BASE_URL) self.assertEqual(response.status_code, status.HTTP_200_OK, msg=response.data) diff --git a/src/rockstor/storageadmin/tests/test_disks.py b/src/rockstor/storageadmin/tests/test_disks.py index 7e34a353f..a4b586819 100644 --- a/src/rockstor/storageadmin/tests/test_disks.py +++ b/src/rockstor/storageadmin/tests/test_disks.py @@ -37,6 +37,25 @@ def setUpClass(cls): cls.mock_wipe_disk = cls.patch_wipe_disk.start() cls.mock_wipe_disk.return_value = 'out','err', 0 + cls.patch_scan_disks = patch('storageadmin.views.disk.scan_disks') + cls.mock_scan_disks = cls.patch_scan_disks.start() + + cls.patch_blink_disk = patch('storageadmin.views.disk.blink_disk') + cls.mock_blink_disk = cls.patch_blink_disk.start() + + cls.patch_pool_usage = patch('storageadmin.views.disk.pool_usage') + cls.mock_pool_usage = cls.patch_pool_usage.start() + cls.mock_pool_usage.return_value = (14680064, 10, 4194305) + + cls.patch_mount_root = patch('storageadmin.views.disk.mount_root') + cls.mock_mount_root = cls.patch_mount_root.start() + + + cls.patch_pool_raid = patch('storageadmin.views.disk.pool_raid') + cls.mock_pool_raid = cls.patch_pool_raid.start() + + cls.patch_enable_quota = patch('storageadmin.views.disk.enable_quota') + cls.mock_enable_quota = cls.patch_enable_quota.start() @classmethod