From ddcd417cf479a7b4bae150cc0998eda5ad52aaa3 Mon Sep 17 00:00:00 2001
From: schultztimothy <schultz.timothy52@gmail.com>
Date: Wed, 19 Jun 2024 13:02:09 -0600
Subject: [PATCH] chore: add api key

---
 api/account/api.py                  |  6 +++++-
 api/account/test/test_allow_list.py | 11 +++++++++--
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/api/account/api.py b/api/account/api.py
index a2841836d..1519808ad 100644
--- a/api/account/api.py
+++ b/api/account/api.py
@@ -32,6 +32,10 @@
 
 from .deduplication import Rules
 
+from trusta_labs.api import CgrantsApiKey
+
+secret_key = CgrantsApiKey()
+
 log = logging.getLogger(__name__)
 
 api = NinjaExtraAPI(urls_namespace="account")
@@ -646,7 +650,7 @@ def get_account_customization(request, dashboard_path: str):
         raise APIException("Customization not found", status.HTTP_404_NOT_FOUND)
 
 
-@api.get("/allow-list/{str:list}/{str:address}", auth=None)
+@api.get("/allow-list/{str:list}/{str:address}", auth=secret_key)
 def check_on_allow_list(request, list: str, address: str):
     """
     Check if an address is on the allow list for a specific round
diff --git a/api/account/test/test_allow_list.py b/api/account/test/test_allow_list.py
index 07b33f046..209b46550 100644
--- a/api/account/test/test_allow_list.py
+++ b/api/account/test/test_allow_list.py
@@ -1,3 +1,4 @@
+from django.conf import settings
 import pytest
 from django.test import Client
 from account.models import AddressListMember, AddressList
@@ -19,13 +20,19 @@ def test_successful_get_allow_list(self):
         )
 
         client = Client()
-        response = client.get(f"/account/allow-list/{list_name}/{user_address}")
+        response = client.get(
+            f"/account/allow-list/{list_name}/{user_address}",
+            HTTP_AUTHORIZATION=settings.CGRANTS_API_TOKEN,
+        )
         assert response.status_code == 200
         assert response.json()["is_member"]
 
     def test_unsuccessful_get_allow_list(self):
         list_name = "test"
         client = Client()
-        response = client.get(f"/account/allow-list/{list_name}/0x123")
+        response = client.get(
+            f"/account/allow-list/{list_name}/0x123",
+            HTTP_AUTHORIZATION=settings.CGRANTS_API_TOKEN,
+        )
         assert response.status_code == 200
         assert not response.json()["is_member"]