diff options
Diffstat (limited to 'tests/test_check_volume_status.py')
| -rw-r--r-- | tests/test_check_volume_status.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tests/test_check_volume_status.py b/tests/test_check_volume_status.py index ad43587..f916174 100644 --- a/tests/test_check_volume_status.py +++ b/tests/test_check_volume_status.py @@ -68,6 +68,11 @@ class TestCheckVolumeStatus(TestCaseBase): .getVolumeQuotaStatus(args)) assert exitStatusCode == utils.PluginStatusCode.OK self.assertEqual("QUOTA: OK", exitStatusMsg) + mock_volumeQuotaStatus.return_value = _getQuotaStatusSoftLimit() + exitStatusCode, exitStatusMsg = (check_volume_status + .getVolumeQuotaStatus(args)) + assert exitStatusCode == utils.PluginStatusCode.WARNING + self.assertEqual("QUOTA:soft limit exceeded on dir3", exitStatusMsg) @mock.patch('glusternagios.glustercli.volumeGeoRepStatus') def test_checkVolumeGeoRepStatus(self, mock_GeoRepStatus): @@ -115,6 +120,12 @@ def _getQuotaStatusHardLimit(): 'soft_ex_dirs': ['dir3']} +def _getQuotaStatusSoftLimit(): + return {'status': 'SOFT_LIMIT_EXCEEDED', + 'hard_ex_dirs': [], + 'soft_ex_dirs': ['dir3']} + + def _getQuotaStatusOk(): return {'status': 'OK', 'hard_ex_dirs': [], |