Ejemplo n.º 1
0
    def test_parent_property(self, mock):
        mock.return_value = {'type': ['property', 'parent/property']}
        with self.assertNumQueries(4):
            generate_data_dictionary(self.domain)

        self.assertEqual(CaseType.objects.filter(domain=self.domain).count(), 1)
        self.assertEqual(CaseProperty.objects.filter(case_type__domain=self.domain).count(), 1)
Ejemplo n.º 2
0
    def test_empty_type(self, mock):
        mock.return_value = {'': ['prop']}
        with self.assertNumQueries(2):
            generate_data_dictionary(self.domain)

        self.assertEqual(CaseType.objects.filter(domain=self.domain).count(), 0)
        self.assertEqual(CaseProperty.objects.filter(case_type__domain=self.domain).count(), 0)
Ejemplo n.º 3
0
def generate_data_dictionary(request, domain):
    try:
        util.generate_data_dictionary(domain)
    except util.OldExportsEnabledException:
        return JsonResponse({
            "failed": "Data Dictionary requires access to new exports"
        }, status=400)

    return JsonResponse({"status": "success"})
Ejemplo n.º 4
0
    def test_one_type(self, mock):
        mock.return_value = {'type': ['property']}
        with self.assertNumQueries(4):
            generate_data_dictionary(self.domain)

        self.assertEqual(
            CaseType.objects.filter(domain=self.domain).count(), 1)
        self.assertEqual(
            CaseProperty.objects.filter(case_type__domain=self.domain).count(),
            1)
Ejemplo n.º 5
0
    def test_no_types(self, mock):
        mock.return_value = {}
        with self.assertNumQueries(1):
            generate_data_dictionary(self.domain)

        self.assertEqual(
            CaseType.objects.filter(domain=self.domain).count(), 0)
        self.assertEqual(
            CaseProperty.objects.filter(case_type__domain=self.domain).count(),
            0)
Ejemplo n.º 6
0
 def handle(self, **options):
     failed_domains = []
     domains = options['domains'] or [d['key'] for d in Domain.get_all(include_docs=False)]
     print('Generating data dictionary for {} domains'.format(len(domains)))
     for domain in with_progress_bar(domains):
         try:
             generate_data_dictionary(domain)
         except OldExportsEnabledException:
             failed_domains.append(domain)
     print('--- Failed Domains ---')
     for domain in failed_domains:
         print(domain)
Ejemplo n.º 7
0
 def handle(self, **options):
     failed_domains = []
     domains = options['domains'] or [d['key'] for d in Domain.get_all(include_docs=False)]
     print('Generating data dictionary for {} domains'.format(len(domains)))
     for domain in with_progress_bar(domains):
         try:
             generate_data_dictionary(domain)
         except OldExportsEnabledException:
             failed_domains.append(domain)
     print('--- Failed Domains ---')
     for domain in failed_domains:
         print(domain)
Ejemplo n.º 8
0
 def handle(self, **options):
     print('Generating data dictionary for domains')
     failed_domains = []
     for domain_dict in with_progress_bar(
             Domain.get_all(include_docs=False)):
         domain = domain_dict['key']
         try:
             generate_data_dictionary(domain)
         except OldExportsEnabledException:
             failed_domains.append(domain)
     print('--- Failed Domains ---')
     for domain in failed_domains:
         print(domain)
Ejemplo n.º 9
0
    def test_already_existing_property(self, mock):
        mock.return_value = {'type': ['property']}
        case_type = CaseType(domain=self.domain, name='type')
        case_type.save()
        CaseProperty(case_type=case_type, name='property').save()

        self.assertEqual(CaseType.objects.filter(domain=self.domain).count(), 1)
        self.assertEqual(CaseProperty.objects.filter(case_type__domain=self.domain).count(), 1)

        with self.assertNumQueries(3):
            generate_data_dictionary(self.domain)

        self.assertEqual(CaseType.objects.filter(domain=self.domain).count(), 1)
        self.assertEqual(CaseProperty.objects.filter(case_type__domain=self.domain).count(), 1)
Ejemplo n.º 10
0
    def test_already_existing_property(self, mock):
        mock.return_value = {'type': ['property']}
        case_type = CaseType(domain=self.domain, name='type')
        case_type.save()
        CaseProperty(case_type=case_type, name='property').save()

        self.assertEqual(CaseType.objects.filter(domain=self.domain).count(), 1)
        self.assertEqual(CaseProperty.objects.filter(case_type__domain=self.domain).count(), 1)

        with self.assertNumQueries(3):
            generate_data_dictionary(self.domain)

        self.assertEqual(CaseType.objects.filter(domain=self.domain).count(), 1)
        self.assertEqual(CaseProperty.objects.filter(case_type__domain=self.domain).count(), 1)
Ejemplo n.º 11
0
def generate_data_dictionary(request, domain):
    if data_dictionary_rebuild_rate_limiter.allow_usage(domain):
        data_dictionary_rebuild_rate_limiter.report_usage(domain)
        try:
            util.generate_data_dictionary(domain)
        except util.OldExportsEnabledException:
            return JsonResponse({
                "failed": "Data Dictionary requires access to new exports"
            }, status=400)

        return JsonResponse({"status": "success"})
    else:
        return JsonResponse({
            "failed": "Rate limit exceeded. Please try again later."
        }, status=429)