def get_enabled_partners_data_keys():
s = get_admin_session()
url = get_url(pages.ManagePartnersPage.url_path)
r = s.get(url)
soup = BeautifulSoup(r.text)
partners = soup.findAll(lambda tag: 'data-key' in tag.attrs and 'Enabled' in tag.text)
partners_ids = [p['data-key'] for p in partners]
return partners_ids
def get_published_parent_pages_ids():
s = get_admin_session()
url = get_url(pages.ManagePagesPage.url_path)
r = s.get(url)
soup = BeautifulSoup(r.text)
pages_published = soup.findAll(lambda tag: 'data-key' in tag.attrs and 'Published' in tag.text)
pages_parent = list(filter(lambda x: 'glyphicon-paperclip' not in str(x), pages_published))
pages_parent_ids = [p['data-key'] for p in pages_parent]
return pages_parent_ids
def delete_user(user_id):
s = get_admin_session()
params = {'id': user_id}
r = s.get(get_url(pages.UpdateUserProfileDetailsPage.url_path), params=params)
if 'The requested page does not exist' in r.text:
assert_equal(r.status_code, 404)
return
if 'Access denied' in r.text:
assert_equal(r.status_code, 403)
return
payload = {'_csrf': get_csrf_token(r)}
url = ''.join([APP_URL, '/user/admin/delete?id={}'.format(user_id)])
r = s.post(url, data=payload)
assert_equal(r.status_code, 200)
assert_in('User has been deleted', r.text)
def create_inventory_group(inventory_group_data):
payload = {
'InventoryGroup[name]': inventory_group_data['name'],
'InventoryGroup[slug]': inventory_group_data['slug'],
'InventoryGroup[partner_id]': inventory_group_data['partner_id'],
'InventoryGroup[content]': inventory_group_data['content'],
}
s = get_admin_session()
url = get_url(pages.ManageInventorygroupsPage.url_path + '?sort=-id')
s.get(url)
url = get_url(URL_PREFIXES['create_source'].format(name=SOURCE_NAMES_MAP['inventory_group']))
r = s.get(url)
data_key = _get_group_data_key(r)
assert_equal(r.status_code, 200)
payload['_csrf'] = get_csrf_token(r, on_form=True)
payload['InventoryGroup[id]'] = data_key
r = s.post(url, data=payload)
assert_equal(r.status_code, 200)
url = get_url(pages.ManageInventorygroupsPage.url_path + '?sort=-id')
r = s.get(url)
assert_in(inventory_group_data['name'], r.text)
_id = _get_group_id(r, data_key)
return {'id': _id, 'data_key': data_key}
def create_blocked_user(user_data):
s = get_admin_session()
user_info = _create_user_record(s, user_data)
_update_user_record(s, user_info.get('id'), user_data)
_block_user(s, user_info.get('id'))
return user_info
