def test_ok(self): """Test fetching people from onebody.""" import_onebody_csv() assert models.RecipientGroup.objects.count() == 1 assert models.Recipient.objects.count() == 7 assert models.RecipientGroup.objects.get( name="[onebody]").recipient_set.count() == 7
def pull_onebody_csv(): from onebody.importer import import_onebody_csv import_onebody_csv()
def test_csv_fails(self): """Test fetching people from onebody.""" with pytest.raises(OnebodyException): import_onebody_csv() assert models.RecipientGroup.objects.count() == 0 assert models.Recipient.objects.count() == 0