def test_costars_import(self): main( current_app.config.get('PROJECT_ROOT') + '/purchasing_test/mock/COSTARS-1.csv', 'COSTARS-1.csv', None, None, None) contracts = get_all_contracts() # assert we got both contracts self.assertEquals(len(contracts), 3) # assert that we got all the line items self.assertEquals(LineItem.query.count(), 12) props = defaultdict(list) companies = get_all_companies() self.assertEquals(len(companies), 3) for company in companies: self.assertEquals(company.contacts.count(), 0) for contract in contracts: self.assertTrue(contract.expiration_date is not None) for property in contract.properties: props[property.key].append(property.value) # assert the county importer works properly self.assertEquals(len(props['Located in']), 2) self.assertEquals(len(props['List of manufacturers']), 2)
def test_costars_import(self): main(current_app.config.get('PROJECT_ROOT') + '/purchasing_test/mock/COSTARS-1.csv', 'COSTARS-1.csv', None, None, None) contracts = ContractBase.query.all() # assert we got both contracts self.assertEquals(len(contracts), 3) # assert that we got all the line items self.assertEquals(LineItem.query.count(), 12) props = defaultdict(list) companies = Company.query.all() self.assertEquals(len(companies), 3) for company in companies: self.assertEquals(company.contacts.count(), 0) for contract in contracts: self.assertTrue(contract.expiration_date is not None) for property in contract.properties: props[property.key].append(property.value) # assert the county importer works properly self.assertEquals(len(props['Located in']), 2) self.assertEquals(len(props['List of manufacturers']), 2)
def test_raises_bad_filename(self): with self.assertRaises(IOError): main('', 'BADFILENAME', None, None, None)