def create_project_quotas(project=None,
parsed_project_quotas=None,
session=None):
project_quota = models.ProjectQuotas(
project_id=project.id, parsed_project_quotas=parsed_project_quotas)
project_quota_repo = repositories.get_project_quotas_repository()
project_quota_repo.create_from(project_quota, session=session)
return project_quota
def test_create_new_project_quotas_with_all_default_quotas(self):
project = models.Project()
project.id = '12345'
project.external_id = '67890'
project_quotas = models.ProjectQuotas(project.id, None)
self.assertEqual('12345', project_quotas.project_id)
self.assertIsNone(project_quotas.secrets)
self.assertIsNone(project_quotas.orders)
self.assertIsNone(project_quotas.containers)
self.assertIsNone(project_quotas.consumers)
self.assertIsNone(project_quotas.cas)
def test_create_new_project_quotas_with_some_default_quotas(self):
project = models.Project()
project.id = '12345'
project.external_id = '67890'
parsed_project_quotas = {
'secrets': 101,
'containers': 103,
'consumers': 105
}
project_quotas = models.ProjectQuotas(project.id,
parsed_project_quotas)
self.assertEqual('12345', project_quotas.project_id)
self.assertEqual(101, project_quotas.secrets)
self.assertIsNone(project_quotas.orders)
self.assertEqual(103, project_quotas.containers)
self.assertEqual(105, project_quotas.consumers)
self.assertIsNone(project_quotas.cas)
def test_project_quotas_check_to_dict_fields(self):
project = models.Project()
project.id = '12345'
project.external_id = '67890'
parsed_project_quotas = {
'secrets': 101,
'orders': 102,
'containers': 103,
'consumers': 105,
'cas': 106
}
project_quotas = models.ProjectQuotas(project.id,
parsed_project_quotas)
self.assertEqual(project.id,
project_quotas.to_dict_fields()['project_id'])
self.assertEqual(101, project_quotas.to_dict_fields()['secrets'])
self.assertEqual(102, project_quotas.to_dict_fields()['orders'])
self.assertEqual(103, project_quotas.to_dict_fields()['containers'])
self.assertEqual(105, project_quotas.to_dict_fields()['consumers'])
self.assertEqual(106, project_quotas.to_dict_fields()['cas'])
