mocked getHours for withdraw bounty

stakwork · Oct 18, 2024 · fb21888 · fb21888
1 parent f0bf975
commit fb21888
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/db/test_config.go b/db/test_config.go
@@ -12,10 +12,10 @@ var TestDB database
 
 func InitTestDB() {
 	rdsHost := "localhost"
-	rdsPort := fmt.Sprintf("%d", 5532)
+	rdsPort := fmt.Sprintf("%d", 5432)
 	rdsDbName := "test_db"
-	rdsUsername := "test_user"
-	rdsPassword := "test_password"
+	rdsUsername := "raph"
+	rdsPassword := "raph"
 	dbURL := fmt.Sprintf("postgres://%s:%s@%s:%s/%s", rdsUsername, rdsPassword, rdsHost, rdsPort, rdsDbName)
 
 	if dbURL == "" {

diff --git a/handlers/bounty.go b/handlers/bounty.go
@@ -34,7 +34,6 @@ type bountyHandler struct {
 func NewBountyHandler(httpClient HttpClient, database db.Database) *bountyHandler {
 	dbConf := db.NewDatabaseConfig(&gorm.DB{})
 	return &bountyHandler{
-
 		httpClient:               httpClient,
 		db:                       database,
 		getSocketConnections:     db.Store.GetSocketConnections,
@@ -936,7 +935,7 @@ func (h *bountyHandler) BountyBudgetWithdraw(w http.ResponseWriter, r *http.Requ
 		withdrawCreated := lastWithdrawal.Created
 		withdrawTime := utils.ConvertTimeToTimestamp(withdrawCreated.String())
 
-		hoursDiff := utils.GetHoursDifference(int64(withdrawTime), &now)
+		hoursDiff := h.getHoursDifference(int64(withdrawTime), &now)
 
 		// Check that last withdraw time is greater than 1
 		if hoursDiff < 1 {