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'])