def test_prune_key_in_vizcols(self):
def cond(assertion):
matches = mmo.find(self.model, ["test", "dept_RID_key"])
assertion(len(matches) == 1)
self._pre(cond)
mmo.prune(self.model, ["test", "dept_RID_key"])
self._post(cond)
def test_prune_col_in_search_box(self):
def cond(assertion):
assertion(
len(mmo.find(self.model, ["test", "person", "last_name"])) ==
1)
self._pre(cond)
mmo.prune(self.model, ["test", "person", "last_name"])
self._post(cond)
def test_prune_col_in_sourcedefs_sources(self):
def cond(assertion):
matches = mmo.find(self.model, ["test", "person", "RID"])
assertion(
any([
m.tag == tag.source_definitions
and m.mapping == 'dept_size' for m in matches
]))
self._pre(cond)
mmo.prune(self.model, ["test", "person", "RID"])
self._post(cond)
def test_prune_col_in_vizcols_pseudocol(self):
def cond(assertion):
matches = mmo.find(self.model, ["test", "dept", "name"])
assertion(
any([
m.anchor.name == 'person' and m.tag == tag.visible_columns
and isinstance(m.mapping, dict) for m in matches
]))
self._pre(cond)
mmo.prune(self.model, ["test", "dept", "name"])
self._post(cond)
def test_prune_col_in_sourcedefs_columns(self):
def cond(assertion):
matches = mmo.find(self.model, ["test", "person", "dept"])
assertion(
any([
m.anchor.name == 'person'
and m.tag == tag.source_definitions and m.mapping == 'dept'
for m in matches
]))
self._pre(cond)
mmo.prune(self.model, ["test", "person", "dept"])
self._post(cond)
def test_prune_fkey_in_sourcedefs_recurse(self):
def cond(assertion):
assertion(
any([
isinstance(vizcol, dict)
and vizcol.get("sourcekey") == "dept_size"
for vizcol in self.model.schemas['test'].tables['person'].
annotations[tag.visible_columns]['detailed']
]))
self._pre(cond)
mmo.prune(self.model, ["test", "person_dept_fkey"])
self._post(cond)
def test_prune_fkey_in_vizfkeys(self):
fkname = ["test", "person_dept_fkey"]
def cond(assertion):
matches = mmo.find(self.model, fkname)
assertion(
any([
m.tag == tag.visible_foreign_keys and m.mapping == fkname
for m in matches
]))
self._pre(cond)
mmo.prune(self.model, fkname)
self._post(cond)
def test_prune_fkey_in_sourcedefs_fkeys(self):
fkname = ["test", "person_dept_fkey"]
def cond(assertion):
matches = mmo.find(self.model, fkname)
assertion(
any([
m.tag == tag.source_definitions and m.mapping == fkname
for m in matches
]))
self._pre(cond)
mmo.prune(self.model, fkname)
self._post(cond)
