def test_bill_cosponsors(self):
cosponsors = nyt.bills.cosponsors('hr1', 111)
hr1 = nyt.bills.get('hr1', 111)
process_bill(hr1['bill_uri'])
bill = Bill.objects.get(congress=111, number='hr1')
self.assertEqual(
set(c.id for c in bill.cosponsors.all()),
set(c['cosponsor_id'] for c in cosponsors['cosponsors'])
)
def test_bill_subjects(self):
subjects = nyt.bills.subjects('hr1', 111)
hr1 = nyt.bills.get('hr1', 111)
process_bill(hr1['bill_uri'])
bill = Bill.objects.get(congress=111, number='hr1')
self.assertEqual(
set(s.name for s in bill.subjects.all()),
set(s['name'] for s in subjects['subjects'])
)
def test_bill_actions(self):
bill_info = nyt.bills.get('hr1', 111) # the stimulus
process_bill(bill_info['bill_uri'])
bill = Bill.objects.get(congress=111, number='hr1')
self.assertEqual(
len(bill_info['actions']),
bill.actions.count()
)
for a1, a2 in zip(bill_info['actions'], bill.actions.all()):
self.assertEqual(a1['description'], a2.description)
self.assertEqual(
parse_date(a1['datetime']),
a2.datetime
)
def test_bill_details(self):
bill_uri = random.choice(list(self.recent_bills))
bill_info = nyt.fetch(bill_uri)
process_bill(bill_uri)
bill = Bill.objects.get(nyt_uri=bill_uri)
self.assertEqual(
bill_info['sponsor_uri'],
bill.sponsor.nyt_uri
)
self.assertEqual(
bill_info['title'],
bill.title
)
self.assertEqual(
bill.nyt_uri,
bill_uri,
)
