From 0013810bb8821e182c9701c096d78703bb4ac917 Mon Sep 17 00:00:00 2001
From: Pavel <pashamullin202@gmail.com>
Date: Tue, 11 Jun 2024 19:08:22 +0300
Subject: [PATCH] added error handling

---
 repository/account/account.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/repository/account/account.go b/repository/account/account.go
index 8c9eac3..2bedd34 100644
--- a/repository/account/account.go
+++ b/repository/account/account.go
@@ -398,6 +398,9 @@ func (r *AccountRepository) GetLeadTarget(ctx context.Context, accountID string,
 		Quizid:    quizID,
 	})
 	if err != nil {
+		if err == sql.ErrNoRows {
+			return nil, pj_errors.ErrNotFound
+		}
 		return nil, err
 	}
 	var leadTargets []model.LeadTarget
@@ -422,7 +425,10 @@ func (r *AccountRepository) UpdateLeadTarget(ctx context.Context, req model.Lead
 		ID:     req.ID,
 	})
 	if err != nil {
-		return model.LeadTarget{}, nil
+		if err == sql.ErrNoRows {
+			return model.LeadTarget{}, pj_errors.ErrNotFound
+		}
+		return model.LeadTarget{}, err
 	}
 	return model.LeadTarget{
 		ID:        row.ID,