mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-01-21 07:55:00 +00:00
152 lines
4.9 KiB
Python
152 lines
4.9 KiB
Python
#
|
|
# This file is part of FreedomBox.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import logging
|
|
import random
|
|
import tempfile
|
|
|
|
import requests
|
|
|
|
from support import config
|
|
|
|
from .service import wait_for_page_update
|
|
|
|
sys_modules = [
|
|
'avahi', 'backups', 'cockpit', 'config', 'datetime', 'diagnostics',
|
|
'dynamicdns', 'firewall', 'letsencrypt', 'monkeysphere', 'names',
|
|
'networks', 'pagekite', 'power', 'snapshot', 'upgrades', 'users'
|
|
]
|
|
|
|
default_url = config['DEFAULT']['url']
|
|
|
|
|
|
def login(browser, url, username, password):
|
|
browser.visit(url)
|
|
|
|
# XXX browser.visit goes to the web page with no cookies,
|
|
# hence there should be some kind of session storage for this to work
|
|
|
|
login_button = browser.find_link_by_href('/plinth/accounts/login/')
|
|
if login_button:
|
|
login_button.first.click()
|
|
if login_button:
|
|
browser.fill('username', username)
|
|
browser.fill('password', password)
|
|
submit(browser)
|
|
else:
|
|
browser.visit(default_url + '/plinth/firstboot/welcome')
|
|
submit(browser) # click the "Start Setup" button
|
|
create_admin_account(browser, username, password)
|
|
|
|
|
|
def is_login_prompt(browser):
|
|
return all(
|
|
[browser.find_by_id('id_username'),
|
|
browser.find_by_id('id_password')])
|
|
|
|
|
|
def nav_to_module(browser, module):
|
|
with wait_for_page_update(browser):
|
|
browser.find_link_by_href('/plinth/').first.click()
|
|
sys_or_apps = 'sys' if module in sys_modules else 'apps'
|
|
with wait_for_page_update(browser):
|
|
browser.find_link_by_href(
|
|
'/plinth/{}/'.format(sys_or_apps)).first.click()
|
|
with wait_for_page_update(browser):
|
|
browser.find_link_by_href('/plinth/{0}/{1}/'.format(
|
|
sys_or_apps, module)).first.click()
|
|
|
|
|
|
def create_user(browser, name, password):
|
|
nav_to_module(browser, 'users')
|
|
with wait_for_page_update(browser):
|
|
browser.find_link_by_href('/plinth/sys/users/create/').first.click()
|
|
browser.find_by_id('id_username').fill(name)
|
|
browser.find_by_id('id_password1').fill(password)
|
|
browser.find_by_id('id_password2').fill(password)
|
|
submit(browser)
|
|
|
|
|
|
def rename_user(browser, old_name, new_name):
|
|
nav_to_module(browser, 'users')
|
|
with wait_for_page_update(browser):
|
|
browser.find_link_by_href('/plinth/sys/users/' + old_name +
|
|
'/edit/').first.click()
|
|
browser.find_by_id('id_username').fill(new_name)
|
|
submit(browser)
|
|
|
|
|
|
def delete_user(browser, name):
|
|
nav_to_module(browser, 'users')
|
|
delete_link = browser.find_link_by_href('/plinth/sys/users/' + name +
|
|
'/delete/')
|
|
if delete_link:
|
|
with wait_for_page_update(browser):
|
|
delete_link.first.click()
|
|
submit(browser)
|
|
|
|
|
|
def is_user(browser, name):
|
|
nav_to_module(browser, 'users')
|
|
return browser.is_text_present(name)
|
|
|
|
|
|
def create_admin_account(browser, username, password):
|
|
browser.find_by_id('id_username').fill(username)
|
|
browser.find_by_id('id_password1').fill(password)
|
|
browser.find_by_id('id_password2').fill(password)
|
|
submit(browser)
|
|
|
|
|
|
def submit(browser, element=None, form_class=None):
|
|
with wait_for_page_update(browser):
|
|
if element:
|
|
element.click()
|
|
elif form_class:
|
|
browser.find_by_css(
|
|
'.{} input[type=submit]'.format(form_class)).click()
|
|
else:
|
|
browser.find_by_css('input[type=submit]').click()
|
|
|
|
|
|
def create_sample_local_file():
|
|
"""Create a sample file for upload using browser."""
|
|
contents = bytearray(random.getrandbits(8) for _ in range(64))
|
|
with tempfile.NamedTemporaryFile(delete=False) as temp_file:
|
|
temp_file.write(contents)
|
|
|
|
return temp_file.name, contents
|
|
|
|
|
|
def download_file(url):
|
|
"""Download a file to disk given a URL."""
|
|
with tempfile.NamedTemporaryFile(delete=False) as temp_file:
|
|
logging.captureWarnings(True)
|
|
request = requests.get(url, verify=False)
|
|
logging.captureWarnings(False)
|
|
temp_file.write(request.content)
|
|
|
|
return temp_file.name
|
|
|
|
|
|
def compare_files(file1, file2):
|
|
"""Assert that the contents of two files are the same."""
|
|
file1_contents = open(file1, 'rb').read()
|
|
file2_contents = open(file2, 'rb').read()
|
|
|
|
assert file1_contents == file2_contents
|