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 = 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_old_contracts_import(self): main(current_app.config.get('PROJECT_ROOT') + '/purchasing_test/mock/old_contracts.csv') # assert we get all contracts and companies contracts = get_all_contracts() self.assertEquals(len(contracts), 3) companies = get_all_companies() self.assertEquals(len(companies), 3) controller_nums = [49020, 49011, 49189] for contract in contracts: self.assertTrue(contract.financial_id in controller_nums) controller_nums.remove(contract.financial_id) self.assertEquals(len(controller_nums), 0)
def test_old_contracts_import(self): main( current_app.config.get('PROJECT_ROOT') + '/purchasing_test/mock/old_contracts.csv') # assert we get all contracts and companies contracts = get_all_contracts() self.assertEquals(len(contracts), 3) companies = get_all_companies() self.assertEquals(len(companies), 3) controller_nums = [49020, 49011, 49189] for contract in contracts: self.assertTrue(contract.financial_id in controller_nums) controller_nums.remove(contract.financial_id) self.assertEquals(len(controller_nums), 0)
def test_get_all_companies(self): insert_a_company() insert_a_company(name='new company 2') insert_a_company(name='new company 3') self.assertEquals(len(get_all_companies()), 3)