style

Author: ceciliastevens

src/_incydr_cli/cmds/users.py

@@ -139,7 +139,7 @@ def list_devices(
     List devices associated with a particular user.
     """
     client = Client()
-    if '@' in user:
+    if "@" in user:
         user = client.users.v1.get_user(user).user_id
     devices = list(client.agents.v1.iter_all(user_id=user))
 

tests/test_users.py

@@ -15,15 +15,12 @@
 from _incydr_sdk.users.models import User
 from _incydr_sdk.users.models import UserRole
 from _incydr_sdk.users.models import UsersPage
-from incydr.models import Agent
-from incydr.models import AgentsPage
 from incydr import Client
 
 TEST_USER_ID = "user-1"
 TEST_ORG_GUID = "orgGuid-1"
 
 
-
 TEST_AGENT_ID = "agent-1"
 
 TEST_AGENT_1 = {
@@ -274,14 +271,15 @@ def mock_get_devices(httpserver_auth: HTTPServer):
         f"/v1/users/{TEST_USER_ID}/devices", method="GET", query_string=urlencode(query)
     ).respond_with_json(devices_data)
 
+
 @pytest.fixture
 def mock_get_agents(httpserver_auth: HTTPServer):
     query = {
         "srtKey": "NAME",
         "srtDir": "ASC",
         "pageSize": 500,
         "page": 1,
-        "userId": TEST_USER_ID
+        "userId": TEST_USER_ID,
     }
     agents_data = {
         "agents": [TEST_AGENT_1, TEST_AGENT_2],
@@ -293,6 +291,7 @@ def mock_get_agents(httpserver_auth: HTTPServer):
         "/v1/agents", method="GET", query_string=urlencode(query)
     ).respond_with_json(agents_data)
 
+
 @pytest.fixture
 def mock_get_user_roles(httpserver_auth: HTTPServer):
     roles_data = {