def test_same_day(self): data1 = sample_reimbursement_data.copy() data1['document_id'] = 42 * 2 data1['issue_date'] = '1970-12-31' Reimbursement.objects.create(**data1) data2 = sample_reimbursement_data.copy() data2['document_id'] = 42 * 3 data2['issue_date'] = '1970-12-31' Reimbursement.objects.create(**data2) data3 = sample_reimbursement_data.copy() data3['document_id'] = 42 * 4 data3['issue_date'] = '1970-12-31' Reimbursement.objects.create(**data3) unique_id = dict( year=1970, applicant_id=13, document_id=84 ) qs = Reimbursement.objects.same_day(**unique_id) self.assertEqual(2, qs.count())
def setUp(self): data = [ sample_reimbursement_data.copy(), sample_reimbursement_data.copy(), sample_reimbursement_data.copy(), sample_reimbursement_data.copy() ] data[1]['cnpj_cpf'] = '22222222222' data[1]['document_id'] = 42 * 2 data[1]['issue_date'] = date(1969, 12, 31) data[1]['probability'] = None data[1]['subquota_id'] = 22 data[2]['applicant_id'] = 13 * 3 data[2]['cnpj_cpf'] = '22222222222' data[2]['document_id'] = 42 * 3 data[2]['probability'] = 0.1 data[2]['subquota_id'] = 22 data[3]['applicant_id'] = 13 * 4 data[3]['cnpj_cpf'] = '22222222222' data[3]['document_id'] = 42 * 4 data[3]['probability'] = 0.9 data[3]['subquota_id'] = 22 data[3]['year'] = 1983 data[3]['issue_date'] = '1970-02-01' for d in data: Reimbursement.objects.create(**d) url_name = 'api:reimbursement-list' self.all = resolve_url(url_name) self.by_year = resolve_url(url_name, year=1970) self.by_applicant = resolve_url(url_name, year=1970, applicant_id=13)