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)