test_quota.py

"""
Unit tests for the Drycc scheduler module.

Run the tests with './manage.py test scheduler'
"""
from scheduler.tests import TestCase
from scheduler import KubeHTTPException


class QuotaTest(TestCase):

    def test_create_quota(self):
        namespace_name = self.create_namespace()
        spec = {
            'hard': {
                'cpu': '3',
                'pods': '10',
                'secrets': '5'
            }
        }
        self.scheduler.quota.create(namespace_name, 'test1', spec=spec)

        response = self.scheduler.quota.get(namespace_name, 'test1')
        data = response.json()
        self.assertEqual(data.get('spec', {}), spec)
        self.assertEqual(data['metadata']['namespace'], namespace_name)

    def test_create_with_nonexistent_namespace(self):
        with self.assertRaises(
            KubeHTTPException,
            msg='failed to create quota test1 for namespace ghost-namespace: 404 Not Found'
        ):
            self.scheduler.quota.create('ghost-namespace', 'test1', data={})