From 1faf547225f75446bb8687f111de14d659f79024 Mon Sep 17 00:00:00 2001
From: Stephen Muss <stephen.muss@qantas.com.au>
Date: Thu, 19 Oct 2023 15:13:48 +1100
Subject: [PATCH] Add support for KMS

---
 pytest_localstack/constants.py      | 1 +
 pytest_localstack/service_checks.py | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/pytest_localstack/constants.py b/pytest_localstack/constants.py
index 3a8f175..c5f981f 100644
--- a/pytest_localstack/constants.py
+++ b/pytest_localstack/constants.py
@@ -33,6 +33,7 @@
     "firehose": 4573,
     "iam": 4593,
     "kinesis": 4568,
+    "kms": 4599,
     "lambda": 4574,
     "logs": 4586,
     "redshift": 4577,
diff --git a/pytest_localstack/service_checks.py b/pytest_localstack/service_checks.py
index 3eeacce..32becad 100644
--- a/pytest_localstack/service_checks.py
+++ b/pytest_localstack/service_checks.py
@@ -126,6 +126,9 @@ def _f(client_response):
     "kinesis": botocore_check_response_type(
         "kinesis", "list_streams", list, "StreamNames"
     ),
+    "kms": botocore_check_response_type(
+        "kms", "list_keys", list, "Keys"
+    ),
     "lambda": botocore_check_response_type(
         "lambda", "list_functions", list, "Functions"
     ),