def test_migrate_from0to1(self):
set_db_schema_version(self.db, 0)
with open(os.path.join(os.path.dirname(__file__), 'data/tender-contract-complete.json'), 'r') as df:
data = json.loads(df.read())
t = Tender(data)
t.store(self.db)
tender = self.db.get(t.id)
self.assertEqual(tender['awards'][0]['value'], data['awards'][0]['value'])
self.assertEqual(tender['awards'][0]['suppliers'], data['awards'][0]['suppliers'])
contract_data = deepcopy(tender['contracts'][0])
del contract_data['value']
del contract_data['suppliers']
contract_data['tender_id'] = tender['_id']
contract_data['tender_token'] = 'xxx'
contract_data['procuringEntity'] = tender['procuringEntity']
contract = Contract(contract_data)
contract.dateModified = get_now()
contract.store(self.db)
contract_data = self.db.get(contract.id)
self.assertNotIn("value", contract_data)
self.assertNotIn("suppliers", contract_data)
migrate_data(self.app.app.registry, 2)
migrated_item = self.db.get(contract.id)
self.assertIn("value", migrated_item)
self.assertEqual(migrated_item['value'], tender['awards'][0]['value'])
self.assertIn("suppliers", migrated_item)
self.assertEqual(migrated_item['suppliers'], tender['awards'][0]['suppliers'])
def test_migrate_from1to2(self):
set_db_schema_version(self.db, 1)
u = Contract(test_contract_data)
u.contractID = "UA-X"
u.store(self.db)
data = self.db.get(u.id)
data["documents"] = [{
"id":
"ebcb5dd7f7384b0fbfbed2dc4252fa6e",
"title":
"name.txt",
"url":
"/tenders/{}/documents/ebcb5dd7f7384b0fbfbed2dc4252fa6e?download=10367238a2964ee18513f209d9b6d1d3"
.format(u.id),
"datePublished":
"2016-06-01T00:00:00+03:00",
"dateModified":
"2016-06-01T00:00:00+03:00",
"format":
"text/plain",
}]
_id, _rev = self.db.save(data)
self.app.app.registry.docservice_url = 'http://localhost'
migrate_data(self.app.app.registry, 2)
migrated_item = self.db.get(u.id)
self.assertIn('http://localhost/get/10367238a2964ee18513f209d9b6d1d3?',
migrated_item['documents'][0]['url'])
self.assertIn(
'Prefix={}%2Febcb5dd7f7384b0fbfbed2dc4252fa6e'.format(u.id),
migrated_item['documents'][0]['url'])
self.assertIn('KeyID=', migrated_item['documents'][0]['url'])
self.assertIn('Signature=', migrated_item['documents'][0]['url'])
def test_migrate_from1to2(self):
set_db_schema_version(self.db, 1)
with open(os.path.join(os.path.dirname(__file__), 'data/tender-contract-complete.json'), 'r') as df:
data = json.loads(df.read())
t = Tender(data)
t.store(self.db)
tender = self.db.get(t.id)
self.assertEqual(tender['awards'][0]['value'], data['awards'][0]['value'])
self.assertEqual(tender['awards'][0]['suppliers'], data['awards'][0]['suppliers'])
contract_data = deepcopy(tender['contracts'][0])
del contract_data['value']
del contract_data['suppliers']
contract_data['tender_id'] = tender['_id']
contract_data['tender_token'] = 'xxx'
contract_data['procuringEntity'] = tender['procuringEntity']
contract = Contract(contract_data)
contract.store(self.db)
contract_data = self.db.get(contract.id)
self.assertNotIn("value", contract_data)
self.assertNotIn("suppliers", contract_data)
migrate_data(self.app.app.registry, 2)
migrated_item = self.db.get(contract.id)
self.assertIn("value", migrated_item)
self.assertEqual(migrated_item['value'], tender['awards'][0]['value'])
self.assertIn("suppliers", migrated_item)
self.assertEqual(migrated_item['suppliers'], tender['awards'][0]['suppliers'])
def test_migrate_from0to1(self):
set_db_schema_version(self.db, 0)
with open(
os.path.join(os.path.dirname(__file__),
"data/tender-contract-complete.json"), "r") as df:
data = json.loads(df.read())
t = Tender(data)
t.store(self.db)
tender = self.db.get(t.id)
self.assertEqual(tender["awards"][0]["value"],
data["awards"][0]["value"])
self.assertEqual(tender["awards"][0]["suppliers"],
data["awards"][0]["suppliers"])
contract_data = deepcopy(tender["contracts"][0])
del contract_data["value"]
del contract_data["suppliers"]
contract_data["tender_id"] = tender["_id"]
contract_data["tender_token"] = "xxx"
contract_data["procuringEntity"] = tender["procuringEntity"]
contract = Contract(contract_data)
contract.dateModified = get_now()
contract.store(self.db)
contract_data = self.db.get(contract.id)
self.assertNotIn("value", contract_data)
self.assertNotIn("suppliers", contract_data)
migrate_data(self.app.app.registry, 1)
migrated_item = self.db.get(contract.id)
tender["awards"][0]["value"]["amount"] = str(
tender["awards"][0]["value"]["amount"])
self.assertIn("value", migrated_item)
tender["awards"][0]["value"]["amount"] = str(
tender["awards"][0]["value"]["amount"])
self.assertEqual(migrated_item["value"], tender["awards"][0]["value"])
self.assertIn("suppliers", migrated_item)
self.assertEqual(migrated_item["suppliers"],
tender["awards"][0]["suppliers"])
def test_migrate_from1to2(self):
set_db_schema_version(self.db, 1)
u = Contract(test_contract_data)
u.contractID = "UA-X"
u.store(self.db)
data = self.db.get(u.id)
data["documents"] = [
{
"id": "ebcb5dd7f7384b0fbfbed2dc4252fa6e",
"title": "name.txt",
"url": "/tenders/{}/documents/ebcb5dd7f7384b0fbfbed2dc4252fa6e?download=10367238a2964ee18513f209d9b6d1d3".format(u.id),
"datePublished": "2016-06-01T00:00:00+03:00",
"dateModified": "2016-06-01T00:00:00+03:00",
"format": "text/plain",
}
]
_id, _rev = self.db.save(data)
self.app.app.registry.docservice_url = 'http://localhost'
migrate_data(self.app.app.registry, 2)
migrated_item = self.db.get(u.id)
self.assertIn('http://localhost/get/10367238a2964ee18513f209d9b6d1d3?', migrated_item['documents'][0]['url'])
self.assertIn('Prefix={}%2Febcb5dd7f7384b0fbfbed2dc4252fa6e'.format(u.id), migrated_item['documents'][0]['url'])
self.assertIn('KeyID=', migrated_item['documents'][0]['url'])
self.assertIn('Signature=', migrated_item['documents'][0]['url'])
def test_migrate(self):
self.assertEqual(get_db_schema_version(self.db), SCHEMA_VERSION)
migrate_data(self.app.app.registry, 1)
self.assertEqual(get_db_schema_version(self.db), SCHEMA_VERSION)
def setUp(self):
super(MigrateTest, self).setUp()
migrate_data(self.app.app.registry)
def test_migrate(self):
self.assertEqual(get_db_schema_version(self.db), SCHEMA_VERSION)
migrate_data(self.app.app.registry, 1)
self.assertEqual(get_db_schema_version(self.db), SCHEMA_VERSION)
def setUp(self):
super(MigrateTest, self).setUp()
migrate_data(self.app.app.registry)
