Add country-specific companion table filtering for map page
This commit is contained in:
@@ -80,6 +80,32 @@ async def test_get_currently_banned_filters_and_pagination(tmp_path: Path) -> No
|
||||
assert records[0].ip == "3.3.3.3"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_currently_banned_filters_by_ip_list(tmp_path: Path) -> None:
|
||||
db_path = str(tmp_path / "fail2ban.db")
|
||||
async with aiosqlite.connect(db_path) as db:
|
||||
await _create_bans_table(db)
|
||||
await db.executemany(
|
||||
"INSERT INTO bans (jail, ip, timeofban, bancount, data) VALUES (?, ?, ?, ?, ?)",
|
||||
[
|
||||
("jail1", "1.1.1.1", 10, 1, "{}"),
|
||||
("jail1", "2.2.2.2", 20, 1, "{}"),
|
||||
("jail1", "3.3.3.3", 30, 1, "{}"),
|
||||
],
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
records, total = await fail2ban_db_repo.get_currently_banned(
|
||||
db_path=db_path,
|
||||
since=0,
|
||||
ip_filter=["2.2.2.2", "3.3.3.3"],
|
||||
)
|
||||
|
||||
assert total == 2
|
||||
assert len(records) == 2
|
||||
assert {record.ip for record in records} == {"2.2.2.2", "3.3.3.3"}
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_ban_counts_by_bucket_ignores_out_of_range_buckets(tmp_path: Path) -> None:
|
||||
db_path = str(tmp_path / "fail2ban.db")
|
||||
|
||||
@@ -47,6 +47,10 @@ async def test_get_archived_history_filtering_and_pagination(app_db: str) -> Non
|
||||
assert total == 2
|
||||
assert len(rows) == 1
|
||||
|
||||
rows, total = await get_archived_history(db, ip_filter=["2.2.2.2"])
|
||||
assert total == 1
|
||||
assert rows[0]["ip"] == "2.2.2.2"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_purge_archived_history(app_db: str) -> None:
|
||||
|
||||
Reference in New Issue
Block a user