Compare commits
14
Commits
57fa3e5b69
...
1.4.1
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d811faef45 | ||
|
|
85fb4b4e79 | ||
|
|
20ec627515 | ||
|
|
f8e6888d50 | ||
|
|
f9f8506963 | ||
|
|
a63be55cb7 | ||
|
|
0bd6a62eca | ||
|
|
b865da38f3 | ||
|
|
67b4984664 | ||
|
|
5eafd7c7cc | ||
|
|
a72b486b3b | ||
|
|
d051d9deb7 | ||
|
|
267d9e52e0 | ||
|
|
4753d80891 |
+162
-4
@@ -3,6 +3,8 @@ name: CI
|
|||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
branches: [main]
|
branches: [main]
|
||||||
|
pull_request:
|
||||||
|
branches: [main]
|
||||||
schedule:
|
schedule:
|
||||||
- cron: "0 6 * * 1"
|
- cron: "0 6 * * 1"
|
||||||
|
|
||||||
@@ -11,14 +13,170 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v7
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
persist-credentials: false
|
||||||
|
|
||||||
- name: Build Docker image
|
- name: Build Docker image
|
||||||
run: docker build -t adguard-cidre:ci .
|
run: |
|
||||||
|
docker build -t adguard-cidre:ci . 2>&1 | tee build.log
|
||||||
|
if grep -q "Building wheel for" build.log; then
|
||||||
|
echo "::warning::A dependency was built from source — check Python/Alpine compatibility"
|
||||||
|
fi
|
||||||
|
|
||||||
- name: Scan with Trivy
|
- name: Smoke test (syntax check)
|
||||||
|
run: |
|
||||||
|
docker run --rm --entrypoint python adguard-cidre:ci -c "
|
||||||
|
import ast
|
||||||
|
with open('blocklist_scheduler.py') as f:
|
||||||
|
source = f.read()
|
||||||
|
try:
|
||||||
|
ast.parse(source)
|
||||||
|
print('OK: syntax is valid')
|
||||||
|
except SyntaxError as e:
|
||||||
|
print(f'::error::Syntax error: {e}')
|
||||||
|
exit(1)
|
||||||
|
"
|
||||||
|
|
||||||
|
- name: Run unit tests
|
||||||
|
run: |
|
||||||
|
docker build --target test -t adguard-cidre:test .
|
||||||
|
docker run --rm adguard-cidre:test pytest -v
|
||||||
|
|
||||||
|
- name: Check deprecation warnings
|
||||||
|
run: |
|
||||||
|
docker run --rm --entrypoint python adguard-cidre:ci -W error::DeprecationWarning -c "import blocklist_scheduler" 2>&1 | tee deprecation.log || true
|
||||||
|
if grep -qi "deprecat" deprecation.log; then
|
||||||
|
echo "::warning::Deprecation warning detected, check logs"
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: Scan with Trivy (critical - blocking)
|
||||||
run: |
|
run: |
|
||||||
docker run --rm \
|
docker run --rm \
|
||||||
-e DOCKER_HOST=tcp://dockerhost:2375 \
|
-e DOCKER_HOST=tcp://dockerhost:2375 \
|
||||||
--add-host=dockerhost:host-gateway \
|
--add-host=dockerhost:host-gateway \
|
||||||
aquasec/trivy:0.74.0 image --exit-code 0 --severity HIGH,CRITICAL adguard-cidre:ci
|
aquasec/trivy:0.74.0 image --exit-code 1 --severity CRITICAL adguard-cidre:ci
|
||||||
|
|
||||||
|
- name: Scan with Trivy (high - informative)
|
||||||
|
run: |
|
||||||
|
docker run --rm \
|
||||||
|
-e DOCKER_HOST=tcp://dockerhost:2375 \
|
||||||
|
--add-host=dockerhost:host-gateway \
|
||||||
|
aquasec/trivy:0.74.0 image --exit-code 0 --severity HIGH adguard-cidre:ci
|
||||||
|
|
||||||
|
- name: Publish tagged image
|
||||||
|
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
|
||||||
|
run: |
|
||||||
|
BEFORE="${{ github.event.before }}"
|
||||||
|
if [ -n "$BEFORE" ] && [ "$BEFORE" != "0000000000000000000000000000000000000000" ] && git cat-file -e "$BEFORE" 2>/dev/null; then
|
||||||
|
CHANGED=$(git diff --name-only "$BEFORE" "${{ github.sha }}")
|
||||||
|
else
|
||||||
|
CHANGED=$(git diff --name-only HEAD~1 HEAD)
|
||||||
|
fi
|
||||||
|
echo "Changed files:"
|
||||||
|
echo "$CHANGED"
|
||||||
|
|
||||||
|
if ! echo "$CHANGED" | grep -qE '^(Dockerfile|blocklist_scheduler\.py|VERSION)$'; then
|
||||||
|
echo "No container-relevant file changed, skipping publish."
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if echo "$CHANGED" | grep -qE '^VERSION$'; then
|
||||||
|
echo "VERSION was manually edited in this push, using it as-is."
|
||||||
|
else
|
||||||
|
echo "VERSION untouched but container files changed, auto-bumping the build number (Z)."
|
||||||
|
OLD_VERSION=$(tr -d '[:space:]' < VERSION)
|
||||||
|
IFS='.' read -r MAJOR MINOR PATCH <<< "$OLD_VERSION"
|
||||||
|
NEW_VERSION="${MAJOR}.${MINOR}.$((PATCH + 1))"
|
||||||
|
echo "$NEW_VERSION" > VERSION
|
||||||
|
|
||||||
|
git config user.name "adguard-cidre-ci"
|
||||||
|
git config user.email "[email protected]"
|
||||||
|
git add VERSION
|
||||||
|
git commit -m "Bump build version to $NEW_VERSION [skip ci]"
|
||||||
|
|
||||||
|
# Belt and suspenders: actions/checkout can leave its own ephemeral
|
||||||
|
# credential injected as an extraheader, which would silently override
|
||||||
|
# the URL-embedded token below. persist-credentials:false on checkout
|
||||||
|
# should already prevent this, but strip it here too just in case.
|
||||||
|
git config --unset-all http.https://git.djeex.fr/.extraheader || true
|
||||||
|
|
||||||
|
git push "https://Djeex:${{ secrets.CI_PUSH_TOKEN }}@git.djeex.fr/Djeex/adguard-cidre.git" HEAD:main
|
||||||
|
fi
|
||||||
|
|
||||||
|
VERSION=$(tr -d '[:space:]' < VERSION)
|
||||||
|
IFS='.' read -r MAJOR MINOR PATCH <<< "$VERSION"
|
||||||
|
MINOR_TAG="${MAJOR}.${MINOR}"
|
||||||
|
|
||||||
|
IMAGE=git.djeex.fr/djeex/adguard-cidre
|
||||||
|
echo "${{ secrets.REGISTRY_TOKEN }}" | docker login git.djeex.fr -u Djeex --password-stdin
|
||||||
|
|
||||||
|
# Retag the already-built, already-scanned image — never rebuild for publish,
|
||||||
|
# so what ships is byte-for-byte what Trivy just scanned.
|
||||||
|
docker tag adguard-cidre:ci "$IMAGE:latest"
|
||||||
|
docker tag adguard-cidre:ci "$IMAGE:$MINOR_TAG"
|
||||||
|
docker tag adguard-cidre:ci "$IMAGE:$VERSION"
|
||||||
|
docker push "$IMAGE:latest"
|
||||||
|
docker push "$IMAGE:$MINOR_TAG"
|
||||||
|
docker push "$IMAGE:$VERSION"
|
||||||
|
|
||||||
|
TRIGGER_MSG=$(git log -1 --format=%s "${{ github.sha }}")
|
||||||
|
PR_NUM=$(echo "$TRIGGER_MSG" | grep -oE '#[0-9]+' | head -1 | tr -d '#' || true)
|
||||||
|
|
||||||
|
CATEGORY="🔧 Maintenance"
|
||||||
|
CHANGE_TITLE="$TRIGGER_MSG"
|
||||||
|
|
||||||
|
if [ -n "$PR_NUM" ]; then
|
||||||
|
PR_JSON=$(curl -s -H "Authorization: token ${{ secrets.CI_PUSH_TOKEN }}" \
|
||||||
|
"https://git.djeex.fr/api/v1/repos/Djeex/adguard-cidre/pulls/$PR_NUM")
|
||||||
|
PR_TITLE=$(echo "$PR_JSON" | jq -r '.title // empty' 2>/dev/null || true)
|
||||||
|
LABELS=$(echo "$PR_JSON" | jq -r '.labels[]?.name' 2>/dev/null || true)
|
||||||
|
|
||||||
|
if [ -n "$PR_TITLE" ]; then
|
||||||
|
CHANGE_TITLE="$PR_TITLE"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if echo "$LABELS" | grep -qx 'bug'; then
|
||||||
|
CATEGORY="⚠️ Hotfix"
|
||||||
|
elif echo "$LABELS" | grep -qx 'major'; then
|
||||||
|
CATEGORY="💥 Breaking change"
|
||||||
|
elif echo "$LABELS" | grep -qx 'minor'; then
|
||||||
|
CATEGORY="✨ Update"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
CHANGED_LIST=$(echo "$CHANGED" | sed 's/^/- /')
|
||||||
|
|
||||||
|
REPO_URL="https://git.djeex.fr/Djeex/adguard-cidre"
|
||||||
|
SHORT_SHA=$(echo "${{ github.sha }}" | cut -c1-7)
|
||||||
|
SOURCE_LINE="[${SHORT_SHA}](${REPO_URL}/commit/${{ github.sha }})"
|
||||||
|
if [ -n "$PR_NUM" ]; then
|
||||||
|
SOURCE_LINE="[#${PR_NUM}](${REPO_URL}/pulls/${PR_NUM}) · ${SOURCE_LINE}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
BODY=$(cat <<EOF
|
||||||
|
## Changelog
|
||||||
|
|
||||||
|
### ${CATEGORY}
|
||||||
|
${CHANGE_TITLE}
|
||||||
|
|
||||||
|
**Source:** ${SOURCE_LINE}
|
||||||
|
**Image:** \`${IMAGE}:${VERSION}\`
|
||||||
|
|
||||||
|
**Changed files:**
|
||||||
|
${CHANGED_LIST}
|
||||||
|
EOF
|
||||||
|
)
|
||||||
|
|
||||||
|
JSON_PAYLOAD=$(jq -n \
|
||||||
|
--arg tag "$VERSION" \
|
||||||
|
--arg name "$VERSION" \
|
||||||
|
--arg body "$BODY" \
|
||||||
|
'{tag_name: $tag, name: $name, target_commitish: "main", body: $body}')
|
||||||
|
|
||||||
|
curl -s -o /dev/null -w "Release API response: %{http_code}\n" -X POST \
|
||||||
|
-H "Authorization: token ${{ secrets.CI_PUSH_TOKEN }}" \
|
||||||
|
-H "Content-Type: application/json" \
|
||||||
|
-d "$JSON_PAYLOAD" \
|
||||||
|
"https://git.djeex.fr/api/v1/repos/Djeex/adguard-cidre/releases"
|
||||||
@@ -1,2 +1,4 @@
|
|||||||
/adguard/*.log
|
/adguard/*.log
|
||||||
/tmp/
|
/tmp/
|
||||||
|
__pycache__/
|
||||||
|
.pytest_cache/
|
||||||
|
|||||||
+11
-3
@@ -1,14 +1,22 @@
|
|||||||
FROM python:3.13-alpine
|
FROM python:3.14.7-alpine AS base
|
||||||
|
|
||||||
ENV TZ=Europe/Paris
|
ENV TZ=Europe/Paris
|
||||||
|
|
||||||
RUN apk add --no-cache tzdata curl \
|
RUN apk add --no-cache tzdata curl \
|
||||||
&& cp /usr/share/zoneinfo/$TZ /etc/localtime \
|
&& cp /usr/share/zoneinfo/$TZ /etc/localtime \
|
||||||
&& echo $TZ > /etc/timezone \
|
&& echo $TZ > /etc/timezone
|
||||||
&& pip install --no-cache-dir requests pyyaml schedule
|
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY requirements.txt .
|
||||||
|
RUN pip install --no-cache-dir -r requirements.txt
|
||||||
|
|
||||||
COPY blocklist_scheduler.py .
|
COPY blocklist_scheduler.py .
|
||||||
|
|
||||||
|
FROM base AS test
|
||||||
|
RUN pip install --no-cache-dir pytest==9.1.1
|
||||||
|
COPY tests/ tests/
|
||||||
|
COPY pytest.ini .
|
||||||
|
|
||||||
|
FROM base
|
||||||
ENTRYPOINT ["python3", "blocklist_scheduler.py"]
|
ENTRYPOINT ["python3", "blocklist_scheduler.py"]
|
||||||
|
|||||||
@@ -175,13 +175,16 @@ def schedule_job():
|
|||||||
schedule.every().day.at(f"{hour:02d}:{minute:02d}").do(update_blocklist)
|
schedule.every().day.at(f"{hour:02d}:{minute:02d}").do(update_blocklist)
|
||||||
logging.info(f"Scheduled daily update at {hour:02d}:{minute:02d}")
|
logging.info(f"Scheduled daily update at {hour:02d}:{minute:02d}")
|
||||||
elif BLOCKLIST_CRON_TYPE == "weekly":
|
elif BLOCKLIST_CRON_TYPE == "weekly":
|
||||||
valid_days = ["mon", "tue", "wed", "thu", "fri", "sat", "sun"]
|
day_names = {
|
||||||
|
"mon": "monday", "tue": "tuesday", "wed": "wednesday", "thu": "thursday",
|
||||||
|
"fri": "friday", "sat": "saturday", "sun": "sunday",
|
||||||
|
}
|
||||||
day = BLOCKLIST_CRON_DAY[:3]
|
day = BLOCKLIST_CRON_DAY[:3]
|
||||||
if day not in valid_days:
|
if day not in day_names:
|
||||||
logging.error(f"Invalid BLOCKLIST_CRON_DAY '{BLOCKLIST_CRON_DAY}', must be one of {valid_days}. Defaulting to Monday.")
|
logging.error(f"Invalid BLOCKLIST_CRON_DAY '{BLOCKLIST_CRON_DAY}', must be one of {list(day_names)}. Defaulting to Monday.")
|
||||||
day = "mon"
|
day = "mon"
|
||||||
getattr(schedule.every(), day).at(f"{hour:02d}:{minute:02d}").do(update_blocklist)
|
getattr(schedule.every(), day_names[day]).at(f"{hour:02d}:{minute:02d}").do(update_blocklist)
|
||||||
logging.info(f"Scheduled weekly update on {day.capitalize()} at {hour:02d}:{minute:02d}")
|
logging.info(f"Scheduled weekly update on {day_names[day].capitalize()} at {hour:02d}:{minute:02d}")
|
||||||
else:
|
else:
|
||||||
logging.error(f"Invalid BLOCKLIST_CRON_TYPE '{BLOCKLIST_CRON_TYPE}', must be 'daily' or 'weekly'. Defaulting to daily.")
|
logging.error(f"Invalid BLOCKLIST_CRON_TYPE '{BLOCKLIST_CRON_TYPE}', must be 'daily' or 'weekly'. Defaulting to daily.")
|
||||||
schedule.every().day.at(f"{hour:02d}:{minute:02d}").do(update_blocklist)
|
schedule.every().day.at(f"{hour:02d}:{minute:02d}").do(update_blocklist)
|
||||||
|
|||||||
@@ -0,0 +1,2 @@
|
|||||||
|
[pytest]
|
||||||
|
pythonpath = .
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
{
|
||||||
|
"$schema": "https://docs.renovatebot.com/renovate-schema.json",
|
||||||
|
"extends": ["config:recommended"],
|
||||||
|
"timezone": "Europe/Paris",
|
||||||
|
"labels": ["bot"],
|
||||||
|
"packageRules": [
|
||||||
|
{
|
||||||
|
"matchManagers": ["pip_requirements"],
|
||||||
|
"matchUpdateTypes": ["patch", "minor"],
|
||||||
|
"automerge": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"matchManagers": ["dockerfile"],
|
||||||
|
"matchUpdateTypes": ["patch"],
|
||||||
|
"automerge": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"matchUpdateTypes": ["major"],
|
||||||
|
"addLabels": ["major"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"matchUpdateTypes": ["minor"],
|
||||||
|
"addLabels": ["minor"]
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"vulnerabilityAlerts": {
|
||||||
|
"enabled": true,
|
||||||
|
"addLabels": ["bug"]
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
requests==2.34.2
|
||||||
|
pyyaml==6.0.3
|
||||||
|
schedule==1.2.2
|
||||||
@@ -0,0 +1,210 @@
|
|||||||
|
import pytest
|
||||||
|
import schedule as schedule_lib
|
||||||
|
import yaml
|
||||||
|
|
||||||
|
import blocklist_scheduler as bs
|
||||||
|
|
||||||
|
|
||||||
|
class FakeResponse:
|
||||||
|
def __init__(self, text="", status_code=200, raise_exc=None):
|
||||||
|
self.text = text
|
||||||
|
self.status_code = status_code
|
||||||
|
self._raise_exc = raise_exc
|
||||||
|
|
||||||
|
def raise_for_status(self):
|
||||||
|
if self._raise_exc:
|
||||||
|
raise self._raise_exc
|
||||||
|
|
||||||
|
|
||||||
|
def test_backup_first_start_creates_backup_when_missing(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
adguard_yaml.write_text("original: config\n")
|
||||||
|
first_backup = tmp_path / "AdGuardHome.yaml.first-start.bak"
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
monkeypatch.setattr(bs, "FIRST_BACKUP", first_backup)
|
||||||
|
|
||||||
|
bs.backup_first_start()
|
||||||
|
|
||||||
|
assert first_backup.read_text() == "original: config\n"
|
||||||
|
|
||||||
|
|
||||||
|
def test_backup_first_start_does_not_overwrite_existing_backup(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
adguard_yaml.write_text("new: config\n")
|
||||||
|
first_backup = tmp_path / "AdGuardHome.yaml.first-start.bak"
|
||||||
|
first_backup.write_text("pristine: original\n")
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
monkeypatch.setattr(bs, "FIRST_BACKUP", first_backup)
|
||||||
|
|
||||||
|
bs.backup_first_start()
|
||||||
|
|
||||||
|
assert first_backup.read_text() == "pristine: original\n"
|
||||||
|
|
||||||
|
|
||||||
|
def test_backup_first_start_raises_if_adguard_yaml_missing(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
first_backup = tmp_path / "AdGuardHome.yaml.first-start.bak"
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
monkeypatch.setattr(bs, "FIRST_BACKUP", first_backup)
|
||||||
|
|
||||||
|
with pytest.raises(FileNotFoundError):
|
||||||
|
bs.backup_first_start()
|
||||||
|
|
||||||
|
|
||||||
|
# --- update_yaml_with_ips (pyyaml) ---
|
||||||
|
|
||||||
|
def test_update_yaml_with_ips_writes_disallowed_clients(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
adguard_yaml.write_text("dns:\n bind_hosts:\n - 0.0.0.0\n")
|
||||||
|
tmp_yaml = tmp_path / "AdGuardHome.yaml.tmp"
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
monkeypatch.setattr(bs, "TMP_YAML", tmp_yaml)
|
||||||
|
|
||||||
|
result = bs.update_yaml_with_ips(["1.2.3.0/24", "5.6.7.8"])
|
||||||
|
|
||||||
|
assert result is True
|
||||||
|
data = yaml.safe_load(adguard_yaml.read_text())
|
||||||
|
assert data["dns"]["disallowed_clients"] == ["1.2.3.0/24", "5.6.7.8"]
|
||||||
|
assert not tmp_yaml.exists()
|
||||||
|
|
||||||
|
|
||||||
|
def test_update_yaml_with_ips_missing_file_returns_false(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
|
||||||
|
assert bs.update_yaml_with_ips(["1.2.3.4"]) is False
|
||||||
|
|
||||||
|
|
||||||
|
def test_update_yaml_with_ips_invalid_yaml_returns_false(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
adguard_yaml.write_text("key: [unclosed\n")
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
|
||||||
|
assert bs.update_yaml_with_ips(["1.2.3.4"]) is False
|
||||||
|
|
||||||
|
|
||||||
|
def test_update_yaml_with_ips_missing_dns_key_raises(tmp_path, monkeypatch):
|
||||||
|
adguard_yaml = tmp_path / "AdGuardHome.yaml"
|
||||||
|
adguard_yaml.write_text("some_other_key: true\n")
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs, "ADGUARD_YAML", adguard_yaml)
|
||||||
|
|
||||||
|
with pytest.raises(KeyError):
|
||||||
|
bs.update_yaml_with_ips(["1.2.3.4"])
|
||||||
|
|
||||||
|
|
||||||
|
# --- fetch_all_country_codes / download_cidr_lists / restart_adguard_container (requests) ---
|
||||||
|
|
||||||
|
def test_fetch_all_country_codes_parses_codes(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs.requests, "get", lambda *a, **k: FakeResponse(text='COUNTRIES = ["FR", "DE", "US"]\n'))
|
||||||
|
|
||||||
|
assert bs.fetch_all_country_codes() == {"fr", "de", "us"}
|
||||||
|
|
||||||
|
|
||||||
|
def test_fetch_all_country_codes_returns_empty_set_on_error(monkeypatch):
|
||||||
|
def raise_error(*a, **k):
|
||||||
|
raise bs.requests.exceptions.ConnectionError("boom")
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs.requests, "get", raise_error)
|
||||||
|
|
||||||
|
assert bs.fetch_all_country_codes() == set()
|
||||||
|
|
||||||
|
|
||||||
|
def test_download_cidr_lists_combines_successful_countries_and_skips_failures(monkeypatch):
|
||||||
|
def fake_get(url, timeout=None):
|
||||||
|
if "/fr.cidr" in url:
|
||||||
|
return FakeResponse(text="1.1.1.0/24\n1.1.2.0/24\n")
|
||||||
|
raise bs.requests.exceptions.ConnectionError("boom")
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs.requests, "get", fake_get)
|
||||||
|
|
||||||
|
result = bs.download_cidr_lists(["fr", "de"])
|
||||||
|
|
||||||
|
assert result == ["1.1.1.0/24", "1.1.2.0/24"]
|
||||||
|
|
||||||
|
|
||||||
|
def test_restart_adguard_container_success_does_not_raise(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs.requests, "post", lambda *a, **k: FakeResponse(status_code=204))
|
||||||
|
|
||||||
|
bs.restart_adguard_container()
|
||||||
|
|
||||||
|
|
||||||
|
def test_restart_adguard_container_error_status_does_not_raise(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs.requests, "post", lambda *a, **k: FakeResponse(status_code=500, text="err"))
|
||||||
|
|
||||||
|
bs.restart_adguard_container()
|
||||||
|
|
||||||
|
|
||||||
|
def test_restart_adguard_container_network_error_does_not_raise(monkeypatch):
|
||||||
|
def raise_error(*a, **k):
|
||||||
|
raise bs.requests.exceptions.ConnectionError("boom")
|
||||||
|
|
||||||
|
monkeypatch.setattr(bs.requests, "post", raise_error)
|
||||||
|
|
||||||
|
bs.restart_adguard_container()
|
||||||
|
|
||||||
|
|
||||||
|
# --- schedule_job (schedule) ---
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def clear_schedule():
|
||||||
|
yield
|
||||||
|
schedule_lib.clear()
|
||||||
|
|
||||||
|
|
||||||
|
def test_schedule_job_daily(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TYPE", "daily")
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TIME", "06:00")
|
||||||
|
|
||||||
|
bs.schedule_job()
|
||||||
|
|
||||||
|
assert len(schedule_lib.jobs) == 1
|
||||||
|
job = schedule_lib.jobs[0]
|
||||||
|
assert job.unit == "days"
|
||||||
|
assert str(job.at_time) == "06:00:00"
|
||||||
|
assert job.job_func.func is bs.update_blocklist
|
||||||
|
|
||||||
|
|
||||||
|
def test_schedule_job_weekly_valid_day(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TYPE", "weekly")
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TIME", "18:30")
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_DAY", "wed")
|
||||||
|
|
||||||
|
bs.schedule_job()
|
||||||
|
|
||||||
|
job = schedule_lib.jobs[0]
|
||||||
|
assert job.unit == "weeks"
|
||||||
|
assert job.start_day == "wednesday"
|
||||||
|
assert str(job.at_time) == "18:30:00"
|
||||||
|
|
||||||
|
|
||||||
|
def test_schedule_job_weekly_invalid_day_defaults_to_monday(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TYPE", "weekly")
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_DAY", "xxx")
|
||||||
|
|
||||||
|
bs.schedule_job()
|
||||||
|
|
||||||
|
assert schedule_lib.jobs[0].start_day == "monday"
|
||||||
|
|
||||||
|
|
||||||
|
def test_schedule_job_invalid_time_defaults_to_six_am(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TYPE", "daily")
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TIME", "not-a-time")
|
||||||
|
|
||||||
|
bs.schedule_job()
|
||||||
|
|
||||||
|
assert str(schedule_lib.jobs[0].at_time) == "06:00:00"
|
||||||
|
|
||||||
|
|
||||||
|
def test_schedule_job_invalid_type_defaults_to_daily(monkeypatch):
|
||||||
|
monkeypatch.setattr(bs, "BLOCKLIST_CRON_TYPE", "bogus")
|
||||||
|
|
||||||
|
bs.schedule_job()
|
||||||
|
|
||||||
|
assert schedule_lib.jobs[0].unit == "days"
|
||||||
Reference in New Issue
Block a user