Skip to content

Commit c094afa

Browse files
committed
refactor: rename db.disable_agent() to db.revoke_agent() for consistency
1 parent e64f8c8 commit c094afa

4 files changed

Lines changed: 5 additions & 5 deletions

File tree

src/where_the_plow/admin_routes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ async def revoke_agent(
163163
return JSONResponse({"error": "Unauthorized"}, status_code=401)
164164

165165
db: Database = request.app.state.db
166-
db.disable_agent(agent_id)
166+
db.revoke_agent(agent_id)
167167
return {"ok": True}
168168

169169

src/where_the_plow/db.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -563,7 +563,7 @@ def approve_agent(self, agent_id: str) -> None:
563563
[agent_id],
564564
)
565565

566-
def disable_agent(self, agent_id: str) -> None:
566+
def revoke_agent(self, agent_id: str) -> None:
567567
"""Revoke an agent by setting status='revoked'."""
568568
self._cursor().execute(
569569
"UPDATE agents SET status = 'revoked' WHERE agent_id = ?",

tests/test_agent_routes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def test_checkin_unknown_agent(agent_client):
9090
def test_checkin_revoked_agent(agent_client):
9191
db = app.state.db
9292
private_pem, _, agent_id, _ = _register_agent(db)
93-
db.disable_agent(agent_id)
93+
db.revoke_agent(agent_id)
9494

9595
body = b""
9696
headers = _sign_request(private_pem, agent_id, body)

tests/test_agents_db.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,9 +61,9 @@ def test_list_agents(db):
6161
assert agents[1]["agent_id"] == "agent-2"
6262

6363

64-
def test_disable_agent(db):
64+
def test_revoke_agent(db):
6565
db.create_agent("agent-1", "My Agent", "pk_abc123")
66-
db.disable_agent("agent-1")
66+
db.revoke_agent("agent-1")
6767
agent = db.get_agent("agent-1")
6868
assert agent is not None
6969
assert agent["status"] == "revoked"

0 commit comments

Comments
 (0)