def test_forward__if_required_and_default_is_set_and_field_in_db__should_not_touch_field(
self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
default = 'test!'
ids = set()
for doc in test_db['schema1_doc1'].find({}, limit=2):
test_db['schema1_doc1'].update_one(
{'_id': doc['_id']}, {'$set': {
'test_field': 'old_value'
}})
ids.add(doc['_id'])
action = CreateField('Schema1Doc1',
'test_field',
choices=None,
db_field='test_field',
default=default,
max_length=None,
min_length=None,
null=False,
primary_key=False,
regex=None,
required=True,
sparse=False,
type_key='StringField',
unique=False,
unique_with=None)
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_forward()
assert all(d['test_field'] == 'old_value'
for d in test_db['schema1_doc1'].find() if d['_id'] in ids)
def test_forward__if_required_and_default_is_set__should_create_field_and_set_a_value(
self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
dump = dump_db()
default = 'test!'
expect = deepcopy(dump)
parser = jsonpath_rw.parse('schema1_doc1[*]')
for rec in parser.find(expect):
rec.value['test_field'] = default
action = CreateField('Schema1Doc1',
'test_field',
choices=None,
db_field='test_field',
default=default,
max_length=None,
min_length=None,
null=False,
primary_key=False,
regex=None,
required=True,
sparse=False,
type_key='StringField',
unique=False,
unique_with=None)
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_forward()
assert expect == dump_db()
def test_backward__should_drop_field(self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
del schema['~Schema1EmbDoc1']['embdoc1_str']
dump = dump_db()
expect = deepcopy(dump)
parsers = load_fixture('schema1').get_embedded_jsonpath_parsers(
'~Schema1EmbDoc1')
for rec in itertools.chain.from_iterable(
p.find(expect) for p in parsers):
if 'embdoc1_str' in rec.value:
del rec.value['embdoc1_str']
action = CreateField('~Schema1EmbDoc1',
'embdoc1_str',
choices=None,
db_field='embdoc1_str',
default=None,
max_length=None,
min_length=None,
null=False,
primary_key=False,
regex=None,
required=False,
sparse=False,
type_key='StringField',
unique=False,
unique_with=None)
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_backward()
assert expect == dump_db()
def test_forward__if_default_is_not_set__should_do_nothing(
self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
dump = dump_db()
action = CreateField('~Schema1EmbDoc1',
'test_field',
choices=None,
db_field='test_field',
default=None,
max_length=None,
min_length=None,
null=False,
primary_key=False,
regex=None,
required=False,
sparse=False,
type_key='StringField',
unique=False,
unique_with=None)
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_forward()
assert dump == dump_db()
def test_backward__should_drop_field(self, load_fixture, test_db, dump_db):
schema = load_fixture('schema1').get_schema()
del schema['Schema1Doc1']['doc1_str']
expect = dump_db()
parser = jsonpath_rw.parse('schema1_doc1[*]')
for rec in parser.find(expect):
if 'doc1_str' in rec.value:
del rec.value['doc1_str']
action = CreateField('Schema1Doc1',
'doc1_str',
choices=None,
db_field='doc1_str',
default=None,
max_length=None,
min_length=None,
null=False,
primary_key=False,
regex=None,
required=True,
sparse=False,
type_key='StringField',
unique=False,
unique_with=None)
action.prepare(test_db, schema, MigrationPolicy.strict)
action.run_backward()
assert expect == dump_db()
def test_prepare__if_such_field_in_document_is_in_schema__should_raise_error(
self, load_fixture, test_db):
schema = load_fixture('schema1').get_schema()
action = CreateField('Schema1Doc1',
'doc1_str',
choices=None,
db_field='doc1_str',
default=None,
max_length=None,
min_length=None,
null=False,
primary_key=False,
regex=None,
required=True,
sparse=False,
type_key='StringField',
unique=False,
unique_with=None)
with pytest.raises(SchemaError):
action.prepare(test_db, schema, MigrationPolicy.strict)