def test_extract_investigator_multiple_cases(self): investigator = InvestigatorFactory(first_name='Luke', last_name='Skywalker') InvestigatorAllegationFactory.create_batch(3, investigator=investigator) rows = self.extract_data() expect(rows).to.have.length(3) for row in rows: expect(row['involvements']).to.have.length(1) expect(row['involvements'][0]['full_name']).to.eq('Luke Skywalker')
def test_investigator_num_cases(self): investigator = InvestigatorFactory() InvestigatorAllegationFactory.create_batch(3, investigator=investigator) allegation = AllegationFactory() InvestigatorAllegationFactory(investigator=investigator, allegation=allegation) rows = self.extract_data() expect(rows).to.have.length(4) expect(rows[-1]['involvements']).to.have.length(1) expect(rows[-1]['involvements'][0]['num_cases']).to.eq(4)