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_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)
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 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 test_migrate_data_return_none(self):
self.app.app.registry.settings['plugins'] = 'fake_plugin'
self.assertIsNone(migrate_data(self.app.app.registry))