def test_none_to_str(self):
"""Test if method transforms None to empty string."""
value = None
parser = BillParser()
result = parser.none_to_str(value)
self.assertEqual(result, '')
def bills_stream(self, url):
status, data = self.download_data(url)
if status == 200:
parser = BillParser()
for e in data:
parsed_data = parser.parse(e)
bill = Bill.objects.create(**parsed_data)
yield bill
def test_has_committees(self):
"""Test if the object has committees."""
parser = BillParser()
parsed_data = parser.parse(self.raw_data[3])
bill = Bill.objects.create(**parsed_data)
count = bill.committees.count()
self.assertNotEqual(count, 0)
def test_has_passings(self):
"""Test if the object has passings (chronology)."""
parser = BillParser()
parsed_data = parser.parse(self.raw_data[3])
bill = Bill.objects.create(**parsed_data)
count = bill.chronology.count()
self.assertNotEqual(count, 0)
def test_has_main_executives(self):
"""Test if the object has main executives."""
parser = BillParser()
parsed_data = parser.parse(self.raw_data[3])
bill = Bill.objects.create(**parsed_data)
count = bill.main_executives.count()
self.assertNotEqual(count, 0)
def test_parse_root(self):
"""Test if method transform dict key to a list."""
root = {'date': '2015-01-29'}
list_root = [{'date': '2015-01-29'}]
parser = BillParser()
result = parser.parse_root(root)
self.assertTrue(type(result) is list)
self.assertEqual(result, list_root)
def setUp(self):
super().setUp()
self.raw_data = json.load(
open(
os.path.join(settings.BASE_DIR,
'bills/tests/bill_parse_data.json')))
self.bill1_raw_data = self.raw_data[3]
self.parser = BillParser()
self.maxDiff = None
def test_has_bind_bills(self):
"""Test if the object has related bill ids."""
parser = BillParser()
parsed_data = parser.parse(self.raw_data[0])
Bill.objects.create(**parsed_data)
parsed_data = parser.parse(self.raw_data[1])
Bill.objects.create(**parsed_data)
parsed_data = parser.parse(self.raw_data[2])
bill = Bill.objects.create(**parsed_data)
count = bill.bind_bills.count()
self.assertNotEqual(count, 0)