type annotations for clients

linode · Jul 10, 2024 · 64e7b53 · 64e7b53
1 parent bdb0862
commit 64e7b53
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/integration/models/object_storage/test_obj.py b/test/integration/models/object_storage/test_obj.py
@@ -114,7 +114,9 @@ def test_get_ssl_cert(bucket: ObjectStorageBucket):
     assert not bucket.ssl_cert().ssl
 
 
-def test_get_cluster(test_linode_client, bucket: ObjectStorageBucket):
+def test_get_cluster(
+    test_linode_client: LinodeClient, bucket: ObjectStorageBucket
+):
     cluster = test_linode_client.load(ObjectStorageCluster, bucket.cluster)
 
     assert "linodeobjects.com" in cluster.domain
@@ -123,7 +125,7 @@ def test_get_cluster(test_linode_client, bucket: ObjectStorageBucket):
 
 
 def test_get_buckets_in_cluster(
-    test_linode_client, bucket: ObjectStorageBucket
+    test_linode_client: LinodeClient, bucket: ObjectStorageBucket
 ):
     cluster = test_linode_client.load(ObjectStorageCluster, bucket.cluster)
     assert any(bucket.id == b.id for b in cluster.buckets_in_cluster())