From cc101695ea03fb2c60f52755b9d4b4be490d39b6 Mon Sep 17 00:00:00 2001
From: Lee ByeongJun <lbj199874@gmail.com>
Date: Thu, 16 Jan 2025 17:18:12 +0900
Subject: [PATCH] update reward test

---
 launchpad/reward_test.gno | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/launchpad/reward_test.gno b/launchpad/reward_test.gno
index d7fb90a6c..4f3e38aa3 100644
--- a/launchpad/reward_test.gno
+++ b/launchpad/reward_test.gno
@@ -1,8 +1,10 @@
 package launchpad
 
 import (
+	"std"
 	"testing"
 
+	"gno.land/p/demo/testutils"
 	"gno.land/p/demo/uassert"
 	"gno.land/p/demo/ufmt"
 	u256 "gno.land/p/gnoswap/uint256"
@@ -225,3 +227,39 @@ func TestCalculateTierRewards(t *testing.T) {
 		})
 	}
 }
+
+func TestCollectRewardByDepositId_FailCases(t *testing.T) {
+	setup := func() {
+        projects = make(map[string]Project)
+        deposits = make(map[string]Deposit)
+        depositsByUserByProject = make(map[std.Address]map[string][]string)
+    }
+
+	t.Run("ID not found", func(t *testing.T) {
+        setup()
+        
+        defer func() {
+            if r := recover(); r == nil {
+                t.Error("expected panic but not")
+            }
+        }()
+        
+        CollectRewardByDepositId("non_existent_id")
+    })
+
+	t.Run("Claimable height before", func(t *testing.T) {
+        setup()
+
+        projectId := "test_project"
+        depositId := "test_deposit"
+
+        deposits[depositId] = Deposit{
+            id: depositId,
+            projectId: projectId,
+            claimableHeight: uint64(std.GetHeight()) + 1000,
+        }
+
+		reward := CollectRewardByDepositId(depositId)
+		uassert.Equal(t, reward, uint64(0))
+	})
+}