From 2f693ad1fd857280ef30c6cc0297fb926f0e79e8 Mon Sep 17 00:00:00 2001
From: Apoorv Dixit <64925866+apoorvdixit88@users.noreply.github.com>
Date: Thu, 18 Jan 2024 12:51:47 +0530
Subject: [PATCH] fix(user): fetch profile_id for sample data (#3358)

---
 crates/router/src/utils/user/sample_data.rs | 23 ++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/crates/router/src/utils/user/sample_data.rs b/crates/router/src/utils/user/sample_data.rs
index dcf635595e0f..3fa2a10629e5 100644
--- a/crates/router/src/utils/user/sample_data.rs
+++ b/crates/router/src/utils/user/sample_data.rs
@@ -52,7 +52,7 @@ pub async fn generate_sample_data(
 
     let business_label_default = merchant_parsed_details.first().map(|x| x.business.clone());
 
-    let profile_id = crate::core::utils::get_profile_id_from_business_details(
+    let profile_id = match crate::core::utils::get_profile_id_from_business_details(
         business_country_default,
         business_label_default.as_ref(),
         &merchant_from_db,
@@ -61,8 +61,25 @@ pub async fn generate_sample_data(
         false,
     )
     .await
-    .change_context(SampleDataError::InternalServerError)
-    .attach_printable("Failed to get business profile")?;
+    {
+        Ok(id) => id.clone(),
+        Err(error) => {
+            router_env::logger::error!(
+                "Profile ID not found in business details. Attempting to fetch from the database {error:?}"
+            );
+
+            state
+                .store
+                .list_business_profile_by_merchant_id(&merchant_id)
+                .await
+                .change_context(SampleDataError::InternalServerError)
+                .attach_printable("Failed to get business profile")?
+                .first()
+                .ok_or(SampleDataError::InternalServerError)?
+                .profile_id
+                .clone()
+        }
+    };
 
     // 10 percent payments should be failed
     #[allow(clippy::as_conversions)]