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_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_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"])