From de9d1f61a984b2a14de46382ffc5dabac841bddb Mon Sep 17 00:00:00 2001 From: Tim Wundenberg Date: Fri, 16 May 2025 11:00:23 +0200 Subject: [PATCH] feat(transaction): #66 fix tests --- db/auth.go | 7 +++++++ main_test.go | 17 +++++++++++++++++ service/treasure_chest.go | 10 ---------- 3 files changed, 24 insertions(+), 10 deletions(-) diff --git a/db/auth.go b/db/auth.go index ab2106d..8923c38 100644 --- a/db/auth.go +++ b/db/auth.go @@ -170,6 +170,13 @@ func (db AuthSqlite) DeleteUser(userId uuid.UUID) error { return types.ErrInternal } + _, err = tx.Exec("DELETE FROM \"transaction\" WHERE user_id = ?", userId) + if err != nil { + _ = tx.Rollback() + log.Error("Could not delete user: %v", err) + return types.ErrInternal + } + err = tx.Commit() if err != nil { log.Error("Could not commit transaction: %v", err) diff --git a/main_test.go b/main_test.go index 3dcbb38..83919c8 100644 --- a/main_test.go +++ b/main_test.go @@ -910,6 +910,20 @@ func TestIntegrationAuth(t *testing.T) { assert.Nil(t, err) assert.Equal(t, http.StatusOK, resp.StatusCode) + formData = url.Values{ + "timestamp": {"2006-01-02T15:04"}, + "value": {"100.00"}, + "csrf-token": {csrfToken}, + } + req, err = http.NewRequestWithContext(ctx, "POST", basePath+"/transaction/new", strings.NewReader(formData.Encode())) + assert.Nil(t, err) + req.Header.Set("Content-Type", "application/x-www-form-urlencoded") + req.Header.Set("Cookie", "id="+sessionId) + req.Header.Set("HX-Request", "true") + resp, err = httpClient.Do(req) + assert.Nil(t, err) + assert.Equal(t, http.StatusOK, resp.StatusCode) + formData = url.Values{ "password": {"password"}, "csrf-token": {csrfToken}, @@ -940,6 +954,9 @@ func TestIntegrationAuth(t *testing.T) { err = db.QueryRow("SELECT COUNT(*) FROM treasure_chest WHERE user_id = ?", userId).Scan(&rows) assert.Nil(t, err) assert.Equal(t, 0, rows) + err = db.QueryRow("SELECT COUNT(*) FROM \"transaction\" WHERE user_id = ?", userId).Scan(&rows) + assert.Nil(t, err) + assert.Equal(t, 0, rows) }) }) diff --git a/service/treasure_chest.go b/service/treasure_chest.go index 445d2a4..316828b 100644 --- a/service/treasure_chest.go +++ b/service/treasure_chest.go @@ -261,13 +261,3 @@ func sortTree(nodes []*types.TreasureChest) []*types.TreasureChest { return result } - -func compareStrings(a, b string) int { - if a < b { - return -1 - } - if a > b { - return 1 - } - return 0 -}