From 3022db75e484617f55ce3842da503cdda58b41fa Mon Sep 17 00:00:00 2001
From: ekremney <ekdogan@adobe.com>
Date: Thu, 1 Feb 2024 17:42:55 +0100
Subject: [PATCH] feat: fix IT tests

---
 packages/spacecat-shared-data-access/docs/schema.json   | 4 ++--
 packages/spacecat-shared-data-access/test/it/db.test.js | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/spacecat-shared-data-access/docs/schema.json b/packages/spacecat-shared-data-access/docs/schema.json
index 5c2069b4..9e283597 100644
--- a/packages/spacecat-shared-data-access/docs/schema.json
+++ b/packages/spacecat-shared-data-access/docs/schema.json
@@ -435,7 +435,7 @@
       "TableName": "spacecat-services-site-candidates",
       "KeyAttributes": {
         "PartitionKey": {
-          "AttributeName": "baseUrl",
+          "AttributeName": "baseURL",
           "AttributeType": "S"
         }
       },
@@ -515,4 +515,4 @@
       }
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/packages/spacecat-shared-data-access/test/it/db.test.js b/packages/spacecat-shared-data-access/test/it/db.test.js
index 72a80a5e..56178a8a 100644
--- a/packages/spacecat-shared-data-access/test/it/db.test.js
+++ b/packages/spacecat-shared-data-access/test/it/db.test.js
@@ -96,6 +96,7 @@ describe('DynamoDB Integration Test', async () => {
   let dataAccess;
 
   const NUMBER_OF_SITES = 10;
+  const NUMBER_OF_SITES_CANDIDATES = 100;
   const NUMBER_OF_ORGANIZATIONS = 3;
   const NUMBER_OF_AUDITS_PER_TYPE_AND_SITE = 3;
 
@@ -119,6 +120,7 @@ describe('DynamoDB Integration Test', async () => {
       TEST_DA_CONFIG,
       NUMBER_OF_ORGANIZATIONS,
       NUMBER_OF_SITES,
+      NUMBER_OF_SITES_CANDIDATES,
       NUMBER_OF_AUDITS_PER_TYPE_AND_SITE,
     );