def test_group_dictize_with_package_count(self):
# group_list_dictize calls it like this by default
group_ = factories.Group()
factories.Dataset(groups=[{'name': group_['name']}])
group_obj = model.Session.query(model.Group).filter_by().first()
context = {'model': model, 'session': model.Session,
'dataset_counts': model_dictize.get_group_dataset_counts()
}
group = model_dictize.group_dictize(group_obj, context,
packages_field='dataset_count')
assert_equal(group['package_count'], 1)
def test_group_dictize_for_org_with_package_count(self):
# group_list_dictize calls it like this by default
org_ = factories.Organization()
factories.Dataset(owner_org=org_['id'])
org_obj = model.Session.query(model.Group).filter_by().first()
context = {'model': model, 'session': model.Session,
'dataset_counts': model_dictize.get_group_dataset_counts()
}
org = model_dictize.group_dictize(org_obj, context,
packages_field='dataset_count')
assert_equal(org['packages'], 1)
def test_group_dictize_for_org_with_package_count(self):
# group_list_dictize calls it like this by default
org_ = factories.Organization()
factories.Dataset(owner_org=org_['id'])
org_obj = model.Session.query(model.Group).filter_by().first()
context = {'model': model, 'session': model.Session,
'dataset_counts': model_dictize.get_group_dataset_counts()
}
org = model_dictize.group_dictize(org_obj, context,
packages_field='dataset_count')
assert_equal(org['package_count'], 1)
def group_tree_filtered(context, data_dict):
currentuser = context['user']
sysadmin = authz.is_sysadmin(currentuser)
model = context['model']
group_type = data_dict.get('type', 'group')
group_counts = model_dictize.get_group_dataset_counts()['owner_org']
filtered_groups = []
for group in model.Group.get_top_level_groups(type=group_type):
branch = _group_tree_branch(group, type=group_type, group_counts=None if sysadmin else group_counts)
if len(branch['children']) > 0 or branch['id'] in group_counts or sysadmin:
filtered_groups.append(branch)
return filtered_groups
def test_group_dictize_with_package_count(self):
# group_list_dictize calls it like this by default
group_obj = factories.Group.model()
other_group_obj = factories.Group.model()
factories.Dataset(groups=[{"name": group_obj.name}])
factories.Dataset(groups=[{"name": other_group_obj.name}])
context = {
"model": model,
"session": model.Session,
"dataset_counts": model_dictize.get_group_dataset_counts(),
}
group = model_dictize.group_dictize(
group_obj, context, packages_field="dataset_count"
)
assert group["package_count"] == 1
def test_group_dictize_for_org_with_package_count(self):
# group_list_dictize calls it like this by default
org_obj = factories.Organization.model()
other_org_ = factories.Organization()
factories.Dataset(owner_org=org_obj.id)
factories.Dataset(owner_org=other_org_["id"])
context = {
"model": model,
"session": model.Session,
"dataset_counts": model_dictize.get_group_dataset_counts(),
}
org = model_dictize.group_dictize(
org_obj, context, packages_field="dataset_count"
)
assert org["package_count"] == 1
