mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-17 22:14:25 +01:00
Add feature to search users by IP address (#1059)
* Allows admins to search users by IP address * Closes #1054
This commit is contained in:
committed by
Kevin Chung
parent
cf7959ab16
commit
34bab12a99
@@ -0,0 +1,49 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from tests.helpers import (
|
||||
create_ctfd,
|
||||
destroy_ctfd,
|
||||
login_as_user,
|
||||
gen_user,
|
||||
gen_tracking,
|
||||
)
|
||||
|
||||
|
||||
def test_admin_user_ip_search():
|
||||
"""Can an admin search user IPs"""
|
||||
app = create_ctfd()
|
||||
with app.app_context():
|
||||
u1 = gen_user(app.db, name="user1", email="user1@ctfd.io")
|
||||
gen_tracking(app.db, user_id=u1.id, ip="1.1.1.1")
|
||||
|
||||
u2 = gen_user(app.db, name="user2", email="user2@ctfd.io")
|
||||
gen_tracking(app.db, user_id=u2.id, ip="2.2.2.2")
|
||||
|
||||
u3 = gen_user(app.db, name="user3", email="user3@ctfd.io")
|
||||
gen_tracking(app.db, user_id=u3.id, ip="3.3.3.3")
|
||||
|
||||
u4 = gen_user(app.db, name="user4", email="user4@ctfd.io")
|
||||
gen_tracking(app.db, user_id=u4.id, ip="3.3.3.3")
|
||||
gen_tracking(app.db, user_id=u4.id, ip="4.4.4.4")
|
||||
|
||||
with login_as_user(app, name="admin", password="password") as admin:
|
||||
r = admin.get("/admin/users?field=ip&q=1.1.1.1")
|
||||
resp = r.get_data(as_text=True)
|
||||
assert "user1" in resp
|
||||
assert "user2" not in resp
|
||||
assert "user3" not in resp
|
||||
|
||||
r = admin.get("/admin/users?field=ip&q=2.2.2.2")
|
||||
resp = r.get_data(as_text=True)
|
||||
assert "user1" not in resp
|
||||
assert "user2" in resp
|
||||
assert "user3" not in resp
|
||||
|
||||
r = admin.get("/admin/users?field=ip&q=3.3.3.3")
|
||||
resp = r.get_data(as_text=True)
|
||||
assert "user1" not in resp
|
||||
assert "user2" not in resp
|
||||
assert "user3" in resp
|
||||
assert "user4" in resp
|
||||
destroy_ctfd(app)
|
||||
|
||||
Reference in New Issue
Block a user