Merge pull request #316 from aclist/chore/geofile-stripping

chore: add geofile test
This commit is contained in:
aclist 2026-05-28 02:09:45 +09:00 committed by GitHub
commit 7e5e326ebe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

27
tests/test_geo_file.py Normal file
View File

@ -0,0 +1,27 @@
from dzgui.config.ipdb import get_ipdb
from pathlib import Path
import os
import pytest
import tempfile
@pytest.mark.slow
def test_stripping():
d = tempfile.TemporaryDirectory()
path = Path(d.name).joinpath("ips.csv")
get_ipdb(path)
with open(path, "rb") as f:
first = f.readline().decode().rstrip()
try:
f.seek(-2, os.SEEK_END)
while f.read(1) != b"\n":
f.seek(-2, os.SEEK_CUR)
except OSError:
f.seek(0)
last = f.read().decode().rstrip()
first_els = first.split(",")
last_els = last.split(",")
assert first_els[0] == "0.0.0.0"
assert last_els[0] == "224.0.0.0"