def test_to_schema_patch__should_return_dictdiffer_diff(self, left_schema):
right_schema = Schema({
'Document1':
Schema.Document(
{
'field1': {
'param11': 'schemavalue11',
'param12': 'schemavalue12'
},
'field2': {
'param21': 'schemavalue21',
'param22': 'schemavalue22'
},
},
parameters={'collection': 'document1'},
indexes={
'index1': {
'fields': [('field1', pymongo.ASCENDING)]
}
}),
'~EmbeddedDocument2':
Schema.Document({
'field1': {
'param3': 'schemavalue3'
},
'field2': {
'param4': 'schemavalue4'
},
})
})
action = DropIndex('Document1', 'index2')
expect = [('change', 'Document1', (left_schema['Document1'],
right_schema['Document1']))]
res = action.to_schema_patch(left_schema)
assert res == expect
def test_forward__if_index_already_dropped__should_ignore_this(
self, test_db, left_schema):
fields = [('field1', pymongo.ASCENDING),
('field2', pymongo.DESCENDING)]
action = DropIndex('Document1', 'index2')
action.prepare(test_db, left_schema, MigrationPolicy.strict)
action.run_forward()
indexes = [
x for x in test_db['document1'].list_indexes()
if x['key'] == SON(fields)
]
assert len(indexes) == 0
def test_forward__if_index_name_is_not_in_params__should_drop_index(
self, test_db, left_schema):
fields = [('field1', pymongo.ASCENDING)]
test_db['document1'].create_index(fields)
action = DropIndex('Document1', 'index1')
action.prepare(test_db, left_schema, MigrationPolicy.strict)
action.run_forward()
indexes = [
x for x in test_db['document1'].list_indexes()
if x['key'] == SON(fields)
]
assert len(indexes) == 0
def test_build_object__if_index_drops__should_return_object(
self, left_schema):
right_schema = Schema({
'Document1':
Schema.Document(
{
'field1': {
'param11': 'schemavalue11',
'param12': 'schemavalue12'
},
'field2': {
'param21': 'schemavalue21',
'param22': 'schemavalue22'
},
},
parameters={'collection': 'document1'},
indexes={}),
'~EmbeddedDocument2':
Schema.Document({
'field1': {
'param3': 'schemavalue3'
},
'field2': {
'param4': 'schemavalue4'
},
})
})
res = DropIndex.build_object('Document1', 'index2', left_schema,
right_schema)
assert isinstance(res, DropIndex)
assert res.document_type == 'Document1'
assert res.index_name == 'index2'
assert res.parameters == {
'fields': [('field1', pymongo.ASCENDING),
('field2', pymongo.DESCENDING)],
'name':
'index2',
'sparse':
True,
}
def test_build_object__if_index_does_not_drop_in_schema__should_return_none(
self, left_schema, index_name):
right_schema = Schema({
'Document1':
Schema.Document(
{
'field1': {
'param11': 'schemavalue11',
'param12': 'schemavalue12'
},
'field3': {
'param31': 'schemavalue31',
'param32': 'schemavalue32'
},
},
parameters={'collection': 'document1'},
indexes={
'index3': {
'fields': [('field2', pymongo.DESCENDING)],
'name': 'index3',
'sparse': True
},
'index1': {
'fields': [('field1', pymongo.DESCENDING)]
}
}),
'~EmbeddedDocument2':
Schema.Document({
'field1': {
'param3': 'schemavalue3'
},
'field2': {
'param4': 'schemavalue4'
},
})
})
res = DropIndex.build_object('Document1', index_name, left_schema,
right_schema)
assert res is None
def test_build_object__if_document_not_in_both_schemas__should_return_none(
self, left_schema, document_type):
right_schema = Schema({
'Document_new':
Schema.Document(
{
'field1': {
'param11': 'schemavalue11',
'param12': 'schemavalue12'
},
'field2': {
'param21': 'schemavalue21',
'param22': 'schemavalue22'
},
'field3': {
'param31': 'schemavalue31',
'param32': 'schemavalue32'
},
},
parameters={'collection': 'document1'},
indexes={}),
'~EmbeddedDocument2':
Schema.Document({
'field1': {
'param3': 'schemavalue3'
},
'field2': {
'param4': 'schemavalue4'
},
})
})
res = DropIndex.build_object(document_type, 'index2', left_schema,
right_schema)
assert res is None
def test_forward_backward__if_index_name_is_in_params__should_create_index(
self, test_db, left_schema):
fields = [('field1', pymongo.ASCENDING),
('field2', pymongo.DESCENDING)]
test_db['document1'].create_index(fields, name='index2', sparse=True)
action = DropIndex('Document1', 'index2')
action.prepare(test_db, left_schema, MigrationPolicy.strict)
action.run_forward()
action.cleanup()
action.prepare(test_db, left_schema, MigrationPolicy.strict)
action.run_backward()
indexes = [
x for x in test_db['document1'].list_indexes()
if x['key'] == SON(fields)
]
assert len(indexes) == 1
assert indexes[0]['name'] == 'index2'
assert indexes[0]['sparse'] is True
def test_prepare__if_index_is_not_in_schema__should_raise_error(
self, test_db, left_schema):
action = DropIndex('Document1', 'unknown_index')
with pytest.raises(SchemaError):
action.prepare(test_db, left_schema, MigrationPolicy.strict)