mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-04 16:13:23 +00:00
Alex
aa0e4ea280
`docsgpt up` copies the standalone Compose file shipped with this package version into the stack directory (~/.docsgpt/server by default), writes its .env and starts the stack on the images of the same version. A first run asks who should reach DocsGPT (this computer, the network with a token, or a domain with HTTPS) and which model provider to use; flags answer the same questions for scripts. Re-running keeps secrets and settings and moves the image tag, and the database password is only generated for a new database. Also: down, status, logs, token, open, env, upgrade (uv tool installs upgrade themselves and run `up` again) and uninstall (keeps settings and data unless --purge). The commands import no Flask, Celery or settings. The wheel carries deployment/docker-compose-standalone.yaml as docsgpt/deploy/docker-compose.yaml; the sdist includes the source file.
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
"""Reading and updating the stack's .env without losing what the user wrote."""
|
|
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from docsgpt.deploy import envfile
|
|
|
|
|
|
class TestRead:
|
|
def test_comments_blanks_quotes_and_export(self, tmp_path):
|
|
path = tmp_path / ".env"
|
|
path.write_text(
|
|
"# settings\n"
|
|
"\n"
|
|
"LLM_PROVIDER=openai\n"
|
|
"export API_KEY='sk-123'\n"
|
|
'LLM_NAME="gpt 5"\n'
|
|
"EMPTY=\n"
|
|
"not a line\n"
|
|
)
|
|
assert envfile.read(path) == {
|
|
"LLM_PROVIDER": "openai",
|
|
"API_KEY": "sk-123",
|
|
"LLM_NAME": "gpt 5",
|
|
"EMPTY": "",
|
|
}
|
|
|
|
def test_a_missing_file_is_empty(self, tmp_path):
|
|
assert envfile.read(tmp_path / ".env") == {}
|
|
|
|
def test_the_last_duplicate_wins(self, tmp_path):
|
|
path = tmp_path / ".env"
|
|
path.write_text("A=1\nA=2\n")
|
|
assert envfile.read(path) == {"A": "2"}
|
|
|
|
|
|
class TestUpdate:
|
|
def test_changes_values_in_place_and_keeps_everything_else(self, tmp_path):
|
|
path = tmp_path / ".env"
|
|
path.write_text("# my settings\nLLM_PROVIDER=openai\n\nCUSTOM=keep me\nDOCSGPT_IMAGE_TAG=0.19.0\n")
|
|
envfile.update(path, {"DOCSGPT_IMAGE_TAG": "0.21.0"})
|
|
assert path.read_text() == "# my settings\nLLM_PROVIDER=openai\n\nCUSTOM=keep me\nDOCSGPT_IMAGE_TAG=0.21.0\n"
|
|
|
|
def test_appends_new_keys_and_removes_none(self, tmp_path):
|
|
path = tmp_path / ".env"
|
|
path.write_text("A=1\nB=2")
|
|
envfile.update(path, {"B": None, "C": "3"})
|
|
assert path.read_text() == "A=1\nC=3\n"
|
|
|
|
def test_duplicates_collapse_to_the_first_line(self, tmp_path):
|
|
path = tmp_path / ".env"
|
|
path.write_text("A=1\nX=y\nA=2\n")
|
|
envfile.update(path, {"A": "3"})
|
|
assert path.read_text() == "A=3\nX=y\n"
|
|
|
|
@pytest.mark.parametrize("value", ["plain", "with space", "hash # inside", "it's", 'say "hi"', "back\\slash", ""])
|
|
def test_values_round_trip(self, tmp_path, value):
|
|
path = tmp_path / ".env"
|
|
envfile.update(path, {"VALUE": value})
|
|
assert envfile.read(path)["VALUE"] == value
|
|
|
|
@pytest.mark.skipif(sys.platform == "win32", reason="POSIX permissions")
|
|
def test_a_new_file_is_private(self, tmp_path):
|
|
path = tmp_path / "stack" / ".env"
|
|
envfile.update(path, {"JWT_SECRET_KEY": "s"})
|
|
assert oct(os.stat(path).st_mode & 0o777) == oct(0o600)
|
|
|
|
def test_rejects_a_newline_in_a_value(self, tmp_path):
|
|
with pytest.raises(ValueError, match="newline"):
|
|
envfile.update(tmp_path / ".env", {"A": "1\n2"})
|