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_contracts(self):
insert_a_contract()
insert_a_contract()
insert_a_contract()
self.assertEquals(len(get_all_contracts()), 3)
