add default 1s timemouts

Author: ramonpetgrave64

kmsauth/__init__.py

@@ -57,7 +57,9 @@ def __init__(
             endpoint_url=None,
             token_cache_size=4096,
             stats=None,
-            max_pool_connections=100
+            max_pool_connections=100,
+            connect_timeout=1,
+            read_timeout=1,
             ):
         """Create a KMSTokenValidator object.
 
@@ -109,14 +111,18 @@ def __init__(
                 aws_secret_access_key=self.aws_creds['SecretAccessKey'],
                 aws_session_token=self.aws_creds['SessionToken'],
                 endpoint_url=endpoint_url,
-                max_pool_connections=max_pool_connections
+                max_pool_connections=max_pool_connections,
+                connect_timeout=connect_timeout,
+                read_timeout=read_timeout,
             )
         else:
             self.kms_client = kmsauth.services.get_boto_client(
                 'kms',
                 region=self.region,
                 endpoint_url=endpoint_url,
-                max_pool_connections=max_pool_connections
+                max_pool_connections=max_pool_connections,
+                connect_timeout=connect_timeout,
+                read_timeout=read_timeout,
             )
         if extra_context is None:
             self.extra_context = {}

kmsauth/services.py

@@ -15,7 +15,9 @@ def get_boto_client(
         aws_secret_access_key=None,
         aws_session_token=None,
         endpoint_url=None,
-        max_pool_connections=None
+        max_pool_connections=None,
+        connect_timeout=None,
+        read_timeout=None,
         ):
     """Get a boto3 client connection."""
     cache_key = '{0}:{1}:{2}:{3}'.format(
@@ -41,7 +43,9 @@ def get_boto_client(
         client,
         endpoint_url=endpoint_url,
         config=botocore.config.Config(
-            max_pool_connections=max_pool_connections
+            max_pool_connections=max_pool_connections,
+            connect_timeout=connect_timeout,
+            read_timeout=read_timeout,
         )
     )
     return CLIENT_CACHE[cache_key]

setup.py

@@ -13,7 +13,7 @@
 
 from setuptools import setup, find_packages
 
-VERSION = "0.6.2.dev1"
+VERSION = "0.6.2.dev2"
 
 requirements = [
     # Boto3 is the Amazon Web Services (AWS) Software Development Kit (SDK)