Improve code, add functionality to decorators, fix bugs

Author: adhorn

LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2018 Adrian Hornsby
+Copyright (c) 2019 Adrian Hornsby
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

README.md

@@ -9,7 +9,7 @@
 * Support for HTTP Error status code injection using ```error_code```
 * Support for disk space failure injection using ```file_size```
 * Using for SSM Parameter Store to control the experiment using ```isEnabled```
-* Per Lambda function injection control using Environment variable (```FAILURE_INJECTION_PARAM```) (thanks to Gunnar Grosch)
+* Per Lambda function injection control using Environment variable (```FAILURE_INJECTION_PARAM```)
 * Support for Serverless Framework using ```sls deploy``` (thanks to Gunnar Grosch)
 * Support for adding rate of failure using ```rate```. (Default rate = 1)
 

example/lambda_function.py

@@ -0,0 +1,74 @@
+import os
+from chaos_lib import (
+    corrupt_delay, corrupt_exception, corrupt_statuscode, SessionWithDelay)
+
+os.environ['CHAOS_PARAM'] = 'chaoslambda.config'
+
+def session_request_with_delay():
+    session = SessionWithDelay(delay=300)
+    session.get('https://stackoverflow.com/')
+    pass
+
+
+@corrupt_exception
+def handler_with_exception(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_exception(exception_type=ValueError)
+def handler_with_exception_arg(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_exception(exception_type=TypeError, exception_msg='foobar')
+def handler_with_exception_arg2(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_statuscode
+def handler_with_statuscode(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_statuscode(error_code=500)
+def handler_with_statuscode_arg(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_delay
+def handler_with_delay(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_delay(delay=1000)
+def handler_with_delay_arg(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }
+
+
+@corrupt_delay(delay=0)
+def handler_with_delay_zero(event, context):
+    return {
+        'statusCode': 200,
+        'body': 'Hello from Lambda!'
+    }