diff --git a/karborclient/tests/unit/v1/test_quota_classes.py b/karborclient/tests/unit/v1/test_quota_classes.py index 436a79e..a86247d 100644 --- a/karborclient/tests/unit/v1/test_quota_classes.py +++ b/karborclient/tests/unit/v1/test_quota_classes.py @@ -47,3 +47,12 @@ class QuotaClassesTest(base.TestCaseShell): 'GET', '/quota_classes/default', headers={}) + + @mock.patch('karborclient.common.http.HTTPClient.json_request') + def test_show_quota_class_with_headers(self, mock_request): + mock_request.return_value = mock_request_return + cs.quota_classes.get('default', session_id='fake_session_id') + mock_request.assert_called_with( + 'GET', + '/quota_classes/default', + headers={'X-Configuration-Session': 'fake_session_id'}) diff --git a/karborclient/tests/unit/v1/test_quotas.py b/karborclient/tests/unit/v1/test_quotas.py index b695b24..b1c3102 100644 --- a/karborclient/tests/unit/v1/test_quotas.py +++ b/karborclient/tests/unit/v1/test_quotas.py @@ -48,6 +48,15 @@ class QuotasTest(base.TestCaseShell): '/quotas/{project_id}'.format(project_id=fakes.PROJECT_ID), headers={}) + @mock.patch('karborclient.common.http.HTTPClient.json_request') + def test_show_quota_with_headers(self, mock_request): + mock_request.return_value = mock_request_return + cs.quotas.get(fakes.PROJECT_ID, False, session_id='fake_session_id') + mock_request.assert_called_with( + 'GET', + '/quotas/{project_id}'.format(project_id=fakes.PROJECT_ID), + headers={'X-Configuration-Session': 'fake_session_id'}) + @mock.patch('karborclient.common.http.HTTPClient.json_request') def test_show_quota_with_detail(self, mock_request): mock_request.return_value = mock_request_return @@ -67,3 +76,13 @@ class QuotasTest(base.TestCaseShell): '/quotas/{project_id}/defaults'.format( project_id=fakes.PROJECT_ID), headers={}) + + @mock.patch('karborclient.common.http.HTTPClient.json_request') + def test_show_quota_default_with_headers(self, mock_request): + mock_request.return_value = mock_request_return + cs.quotas.defaults(fakes.PROJECT_ID, session_id='fake_session_id') + mock_request.assert_called_with( + 'GET', + '/quotas/{project_id}/defaults'.format( + project_id=fakes.PROJECT_ID), + headers={'X-Configuration-Session': 'fake_session_id'})