def test_nested():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
assert len(test.items) == 2
test.items[1].name = 'hello'
orig_sub_version = test.items[1].unique_id
assert len(test.items) == 2
mo.store(test)
assert test.items[1].unique_id == orig_sub_version
def test_unique_links():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
sub_version = test.items[1].unique_id
test.items = [test.items[1]]
mo.store(test)
test.items = []
test = mo.retrieve('group', unique_id=test.unique_id)[0]
assert len(test.items) == 1, len(test.items)
assert test.items[0].unique_id == sub_version
def test_circular_reference():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
orig_id = test.unique_id
test.items[0].items.append(test)
mo.store(test)
test.items = []
test = mo.retrieve('group', unique_id=test.unique_id)[0]
sub0 = test.items[0]
assert len(sub0.items) == 2, sub0.items
assert sub0.items[1].unique_id == orig_id
assert test.unique_id == orig_id
def test_preserve_provenance():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
test2 = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
mo.store([test, test2])
assert len(test.items) == 2
test.items = []
test2.items = []
mo.store([test, test2])
assert len(test.items) == 0
previous = mo.retrieve('group', unique_id=test.prev_uid)[0]
assert len(previous.items) == 2, repr(previous)
def test_clone():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
test2 = test.clone()
assert test2.unique_id != test.unique_id
assert test2.prev_uid == test.unique_id
assert test2.items[0].unique_id == test.items[0].unique_id
test3 = test.clone(True)
assert test3.unique_id != test.unique_id
assert test3.prev_uid == test.unique_id
assert test3.items[0].unique_id != test.items[0].unique_id
assert test3.items[0].prev_uid == test.items[0].unique_id
def test_remove_objects():
group = mo.Group(name='foo',
items=[mo.Group(name='baz', description='hello')])
sub_id = group.items[0].unique_id
mo.store(group)
db = mo.retrieve('groups', unique_id=sub_id)[0]
assert db.unique_id == sub_id
mo.remove_objects(group, _override=True)
test = mo.retrieve('groups', name='foo')
assert not test
test_sub = mo.retrieve('groups_items', target_id=sub_id)
assert not test_sub
def test_simple():
test = mo.Group()
uid = test.unique_id
mo.store(test)
assert test.unique_id == uid
assert test.prev_uid != ''
test.name = 'hello'
mo.store(test)
assert test.unique_id == uid
assert test.prev_uid != ''
def test_recover():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
test.name = 'howdy'
top_version = test.unique_id
sub_version = test.items[1].unique_id
mo.store(test)
mo.store(test) # should have no effect
assert len(test.items) == 2
assert test.unique_id == top_version
# make sure we can recover the previous version
test.items = []
assert test.unique_id == top_version
test = mo.retrieve('group', unique_id=top_version)[0]
assert test.unique_id == top_version
assert len(test.items) == 2, len(test.items)
assert test.unique_id == top_version
assert test.items[1].unique_id == sub_version
def test_dill():
test = mo.Group(items=[mo.Group(description='hello')])
blob = dill.dumps(test)
new = dill.loads(blob)
assert new.items[0].description == 'hello'
def test_store_stubs():
test = mo.Group(items=[mo.Group(items=[mo.LcmsRun()]), mo.LcmsRun()])
mo.store(test)
test = mo.retrieve('group', unique_id=test.unique_id)[0]
assert isinstance(test.items[0], mo.Group)
mo.store(test)
def test_list_item_changed():
g = mo.Group()
g.items = []
g._changed = False
g.items.append(mo.Group())
assert g._changed
