diff options
Diffstat (limited to 'snaps/config/tests/project_tests.py')
-rw-r--r-- | snaps/config/tests/project_tests.py | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/snaps/config/tests/project_tests.py b/snaps/config/tests/project_tests.py index 0470d83..99de4a9 100644 --- a/snaps/config/tests/project_tests.py +++ b/snaps/config/tests/project_tests.py @@ -37,6 +37,7 @@ class ProjectConfigUnitTests(unittest.TestCase): self.assertIsNone(settings.description) self.assertTrue(settings.enabled) self.assertEqual(list(), settings.users) + self.assertIsNone(settings.quotas) def test_config_with_name_only(self): settings = ProjectConfig(**{'name': 'foo'}) @@ -45,25 +46,52 @@ class ProjectConfigUnitTests(unittest.TestCase): self.assertIsNone(settings.description) self.assertTrue(settings.enabled) self.assertEqual(list(), settings.users) + self.assertIsNone(settings.quotas) def test_all(self): users = ['test1', 'test2'] + quotas = { + 'cores': 4, 'instances': 5, 'injected_files': 6, + 'injected_file_content_bytes': 60000, 'ram': 70000, 'fixed_ips': 7, + 'key_pairs': 8} settings = ProjectConfig( name='foo', domain='bar', description='foobar', enabled=False, - users=users) + users=users, quotas=quotas) self.assertEqual('foo', settings.name) self.assertEqual('bar', settings.domain_name) self.assertEqual('foobar', settings.description) self.assertFalse(settings.enabled) self.assertEqual(users, settings.users) + self.assertIsNotNone(settings.quotas) + self.assertEquals(4, settings.quotas.get('cores')) + self.assertEquals(5, settings.quotas.get('instances')) + self.assertEquals(6, settings.quotas.get('injected_files')) + self.assertEquals( + 60000, settings.quotas.get('injected_file_content_bytes')) + self.assertEquals(70000, settings.quotas.get('ram')) + self.assertEquals(7, settings.quotas.get('fixed_ips')) + self.assertEquals(8, settings.quotas.get('key_pairs')) def test_config_all(self): users = ['test1', 'test2'] settings = ProjectConfig( **{'name': 'foo', 'domain': 'bar', 'description': 'foobar', - 'enabled': False, 'users': users}) + 'enabled': False, 'users': users, + 'quotas': { + 'cores': 4, 'instances': 5, 'injected_files': 6, + 'injected_file_content_bytes': 60000, 'ram': 70000, + 'fixed_ips': 7, 'key_pairs': 8}}) self.assertEqual('foo', settings.name) self.assertEqual('bar', settings.domain_name) self.assertEqual('foobar', settings.description) self.assertFalse(settings.enabled) self.assertEqual(users, settings.users) + self.assertIsNotNone(settings.quotas) + self.assertEquals(4, settings.quotas.get('cores')) + self.assertEquals(5, settings.quotas.get('instances')) + self.assertEquals(6, settings.quotas.get('injected_files')) + self.assertEquals( + 60000, settings.quotas.get('injected_file_content_bytes')) + self.assertEquals(70000, settings.quotas.get('ram')) + self.assertEquals(7, settings.quotas.get('fixed_ips')) + self.assertEquals(8, settings.quotas.get('key_pairs')) |