mirror of
https://github.com/HeyPuter/puter.git
synced 2026-01-05 04:30:33 -06:00
c93a53ead2
* ci: init e2e test for browser env stash changes * test: update fsentry definition, add tests stash changes * test: pass puter-js mkdir test * test: add test for puter-js move * tidy code * tidy code * doc: add docs for playwright test * recover memoryfs * test: puter-js readdir/stat * test: puter-js write * test: puter-js read * test: puter-js move_cart * test: fix failed tests on move * tests: rename files * test: puter-js copy_cart * tests: puter-js batch/delete, read config from file * ci: add vitest * ci: update names and timeout * ci: simplify playwright-test * ci: simplify api-test * move "api-tester" from tools to tests * test: update example config * test: remove folder tests/api-tester/ci * test: unify config location * test: remove unused files * ci: fix wrong config * ci: fix wrong path * test: add docs * ci: update timeout, print artifact url
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
import os
|
|
import time
|
|
|
|
import cxc_toolkit
|
|
import requests
|
|
import yaml
|
|
|
|
|
|
PUTER_ROOT = os.getcwd()
|
|
|
|
|
|
def init_backend_config():
|
|
"""
|
|
Initialize a default config in ./volatile/config/config.json.
|
|
"""
|
|
# init config.json
|
|
server_process = cxc_toolkit.exec.run_background("npm start")
|
|
|
|
# wait 10s for the server to start
|
|
time.sleep(10)
|
|
server_process.terminate()
|
|
|
|
|
|
# Possible reasons for failure:
|
|
# - The backend server is not initialized, run "npm start" to initialize it.
|
|
# - Admin password in the kv service is flushed, have to trigger the creation of the admin user.
|
|
# 1. sqlite3 ./volatile/runtime/puter-database.sqlite
|
|
# 2. DELETE FROM user WHERE username = 'admin';
|
|
def get_admin_password() -> str:
|
|
"""
|
|
Get the admin password from the backend server, throw an error if not found.
|
|
"""
|
|
LOG_PATH = "/tmp/backend.log"
|
|
backend_process = cxc_toolkit.exec.run_background("npm start", log_path=LOG_PATH)
|
|
|
|
# NB: run_command + kill_on_output may wait indefinitely, use run_background + hard limit instead
|
|
time.sleep(10)
|
|
|
|
backend_process.terminate()
|
|
|
|
# read the log file
|
|
with open(LOG_PATH, "r") as f:
|
|
lines = f.readlines()
|
|
for line in lines:
|
|
if "password for admin" in line:
|
|
print(f"found password line: ---{line}---")
|
|
admin_password = line.split("password for admin is:")[1].strip()
|
|
print(f"Extracted admin password: {admin_password}")
|
|
return admin_password
|
|
|
|
raise RuntimeError(f"no admin password found, check {LOG_PATH} for details")
|
|
|
|
|
|
def get_token(admin_password: str) -> str:
|
|
"""
|
|
Get the token from the backend server, throw an error if not found.
|
|
"""
|
|
server_url = "http://api.puter.localhost:4100/login"
|
|
login_data = {"username": "admin", "password": admin_password}
|
|
response = requests.post(
|
|
server_url,
|
|
headers={
|
|
"Content-Type": "application/json",
|
|
"Accept": "application/json",
|
|
"Origin": "http://api.puter.localhost:4100",
|
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
|
|
},
|
|
json=login_data,
|
|
timeout=30,
|
|
)
|
|
|
|
response_json = response.json()
|
|
if "token" not in response_json:
|
|
raise RuntimeError("No token found")
|
|
return response_json["token"]
|
|
|
|
|
|
def init_client_config(token: str):
|
|
"""
|
|
Initialize a client config in ./tests/client-config.yaml.
|
|
"""
|
|
example_config_path = f"{PUTER_ROOT}/tests/example-client-config.yaml"
|
|
config_path = f"{PUTER_ROOT}/tests/client-config.yaml"
|
|
|
|
# load
|
|
with open(example_config_path, "r") as f:
|
|
config = yaml.safe_load(f)
|
|
|
|
# update
|
|
config["auth_token"] = token
|
|
|
|
# write
|
|
with open(config_path, "w") as f:
|
|
yaml.dump(config, f, default_flow_style=False, indent=2)
|