def test_prepare__if_such_document_is_not_in_schema__should_raise_error(
self, load_fixture, test_db):
schema = load_fixture('schema1').get_schema()
del schema['Schema1Doc1']
action = DropDocument('Schema1Doc1')
with pytest.raises(SchemaError):
action.prepare(test_db, schema, MigrationPolicy.strict)
def test_backward__should_do_nothing(self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
dump = dump_db()
action = DropDocument('Schema1Doc1')
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_backward()
assert dump == dump_db()
def test_forward__should_drop_collection(self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
dump = dump_db()
action = DropDocument('Schema1Doc1')
action.prepare(test_db, schema, MigrationPolicy.strict)
expect = deepcopy(dump)
del expect['schema1_doc1']
action.run_forward()
assert expect == dump_db()
def test_backward__on_unexistance_collection_specified__should_do_nothing(
self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
schema['Schema1Doc1'].parameters['collection'] = 'unknown_collection'
dump = dump_db()
action = DropDocument('Schema1Doc1')
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_backward()
assert dump == dump_db()
def test_build_object__on_embdedded_document_type__should_return_none(self):
left_schema = Schema({
'Document1': Schema.Document({
'field1': {'param1': 'schemavalue1', 'param2': 'schemavalue2'},
}, parameters={'collection': 'document1'}),
'~EmbeddedDocument': Schema.Document({
'field1': {'param_new': 'schemavalue_new'},
}, parameters={'collection': 'document_new', 'test_parameter': 'test_value'})
})
right_schema = Schema({
'Document1': Schema.Document({
'field1': {'param1': 'schemavalue1', 'param2': 'schemavalue2'},
}, parameters={'collection': 'document1'}),
})
res = DropDocument.build_object('~EmbeddedDocument', left_schema, right_schema)
assert res is None
def test_build_object__if_document_is_not_dropping_in_schema__should_return_none(
self, document_type
):
left_schema = Schema({
'Document1': Schema.Document({
'field1': {'param1': 'schemavalue1', 'param2': 'schemavalue2'},
}, parameters={'collection': 'document1'}),
'Document2': Schema.Document({
'field21': {'param21': 'schemavalue21', 'param22': 'schemavalue22'},
}, parameters={'collection': 'document21'}),
})
right_schema = Schema({
'Document1': Schema.Document({
'field1': {'param1': 'schemavalue1', 'param2': 'schemavalue2'},
}, parameters={'collection': 'document1'}),
})
res = DropDocument.build_object(document_type, left_schema, right_schema)
assert res is None
def test_build_object__if_document_is_droppoing__should_return_object(self):
left_schema = Schema({
'Document1': Schema.Document({
'field1': {'param1': 'schemavalue1', 'param2': 'schemavalue2'},
}, parameters={'collection': 'document1'}),
'Document2': Schema.Document({
'field21': {'param21': 'schemavalue21', 'param22': 'schemavalue22'},
}, parameters={'collection': 'document21'}),
})
right_schema = Schema({
'Document1': Schema.Document({
'field1': {'param1': 'schemavalue1', 'param2': 'schemavalue2'},
}, parameters={'collection': 'document1'}),
})
res = DropDocument.build_object('Document2', left_schema, right_schema)
assert isinstance(res, DropDocument)
assert res.document_type == 'Document2'
assert res.parameters == {}