Merge pull request stakwork#1604 from MirzaHanan/tests_BountyBudgetWi…

…thdraw

Backend [Integration Test] : `bounty.go` handlers `BountyBudgetWithdraw`

handlers/bounty_test.go

@@ -1567,6 +1567,93 @@ func TestBountyBudgetWithdraw(t *testing.T) {
 		assert.Equal(t, "Payment error", response["error"].(string))
 		mockHttpClient.AssertCalled(t, "Do", mock.AnythingOfType("*http.Request"))
 	})
+
+	t.Run("Should test that an Organization's Budget Total Amount is accurate after three (3) successful 'Budget Withdrawal Requests'", func(t *testing.T) {
+		ctxs := context.WithValue(context.Background(), auth.ContextKey, "valid-key")
+		mockDb := dbMocks.NewDatabase(t)
+		mockHttpClient := mocks.NewHttpClient(t)
+		bHandler := NewBountyHandler(mockHttpClient, mockDb)
+
+		paymentAmount := uint(1500)
+		initialBudget := uint(5000)
+		invoice := "lnbc15u1p3xnhl2pp5jptserfk3zk4qy42tlucycrfwxhydvlemu9pqr93tuzlv9cc7g3sdqsvfhkcap3xyhx7un8cqzpgxqzjcsp5f8c52y2stc300gl6s4xswtjpc37hrnnr3c9wvtgjfuvqmpm35evq9qyyssqy4lgd8tj637qcjp05rdpxxykjenthxftej7a2zzmwrmrl70fyj9hvj0rewhzj7jfyuwkwcg9g2jpwtk3wkjtwnkdks84hsnu8xps5vsq4gj5hs"
+
+		for i := 0; i < 3; i++ {
+			expectedFinalBudget := initialBudget - (paymentAmount * uint(i))
+
+			mockDb.ExpectedCalls = nil
+			mockDb.Calls = nil
+			mockHttpClient.ExpectedCalls = nil
+			mockHttpClient.Calls = nil
+
+			mockDb.On("UserHasAccess", "valid-key", "org-1", db.WithdrawBudget).Return(true)
+			mockDb.On("GetOrganizationBudget", "org-1").Return(db.BountyBudget{
+				TotalBudget: expectedFinalBudget,
+			}, nil)
+			mockDb.On("WithdrawBudget", "valid-key", "org-1", paymentAmount).Return(nil)
+			mockHttpClient.On("Do", mock.AnythingOfType("*http.Request")).Return(&http.Response{
+				StatusCode: 200,
+				Body:       io.NopCloser(bytes.NewBufferString(`{"success": true}`)),
+			}, nil)
+
+			withdrawRequest := db.WithdrawBudgetRequest{
+				PaymentRequest: invoice,
+				OrgUuid:        "org-1",
+			}
+			requestBody, _ := json.Marshal(withdrawRequest)
+			req, _ := http.NewRequestWithContext(ctxs, http.MethodPost, "/budget/withdraw", bytes.NewReader(requestBody))
+
+			rr := httptest.NewRecorder()
+
+			bHandler.BountyBudgetWithdraw(rr, req)
+			assert.Equal(t, http.StatusOK, rr.Code)
+			var response db.InvoicePaySuccess
+			err := json.Unmarshal(rr.Body.Bytes(), &response)
+			assert.NoError(t, err)
+			assert.True(t, response.Success, "Expected invoice payment to succeed")
+			finalBudget := mockDb.GetOrganizationBudget("org-1")
+			assert.Equal(t, expectedFinalBudget, finalBudget.TotalBudget, "The organization's final budget should reflect the deductions from the successful withdrawals")
+
+		}
+	})
+
+	t.Run("Should test that the BountyBudgetWithdraw handler gets locked by go mutex when it is called i.e. the handler has to be fully executed before it processes another request.", func(t *testing.T) {
+		mockDb := dbMocks.NewDatabase(t)
+		mockHttpClient := mocks.NewHttpClient(t)
+		bHandler := NewBountyHandler(mockHttpClient, mockDb)
+
+		var processingTimes []time.Time
+		var mutex sync.Mutex
+
+		server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			mutex.Lock()
+			processingTimes = append(processingTimes, time.Now())
+			time.Sleep(10 * time.Millisecond)
+			mutex.Unlock()
+
+			bHandler.BountyBudgetWithdraw(w, r)
+		}))
+		defer server.Close()
+
+		var wg sync.WaitGroup
+		for i := 0; i < 3; i++ {
+			wg.Add(1)
+			go func() {
+				defer wg.Done()
+				_, err := http.Get(server.URL)
+				if err != nil {
+					t.Errorf("Failed to send request: %v", err)
+				}
+			}()
+		}
+		wg.Wait()
+
+		for i := 1; i < len(processingTimes); i++ {
+			assert.True(t, processingTimes[i].After(processingTimes[i-1]),
+				"Expected processing times to be sequential, indicating mutex is locking effectively.")
+		}
+	})
+
 }
 
 func TestPollInvoice(t *testing.T) {