// Vikunja is a to-do list application to facilitate your life. // Copyright 2018-present Vikunja and contributors. All rights reserved. // // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as published by // the Free Software Foundation, either version 3 of the License, or // (at your option) any later version. // // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU Affero General Public License for more details. // // You should have received a copy of the GNU Affero General Public License // along with this program. If not, see . package db import ( "errors" "testing" "github.com/stretchr/testify/assert" "xorm.io/builder" ) func TestMultiFieldSearchLogic(t *testing.T) { // Test the logic without requiring database initialization fields := []string{"title", "description"} search := "test" // Test with ParadeDB enabled originalParadeDB := paradedbInstalled paradedbInstalled = true defer func() { paradedbInstalled = originalParadeDB }() // We'll test the logic by checking if the right type of condition is created // without relying on the Type() function that requires DB initialization // Create conditions manually for each database type conditions := make([]builder.Cond, len(fields)) for i, field := range fields { conditions[i] = &builder.Like{field, "%" + search + "%"} } fallbackCond := builder.Or(conditions...) // Test ParadeDB query string generation fieldQueries := make([]string, len(fields)) for i, field := range fields { fieldQueries[i] = field + ":" + search } expectedParadeDBQuery := "title:test OR description:test" actualQuery := fieldQueries[0] + " OR " + fieldQueries[1] if actualQuery != expectedParadeDBQuery { t.Errorf("Expected ParadeDB query '%s', got '%s'", expectedParadeDBQuery, actualQuery) } // Test that fallback condition is created correctly if fallbackCond == nil { t.Fatal("Expected non-nil fallback condition") } t.Logf("ParadeDB query would be: %s", expectedParadeDBQuery) t.Logf("Fallback condition created successfully") } func TestIsMySQLDuplicateEntryError(t *testing.T) { tests := []struct { name string err error constraintName string expected bool }{ { name: "nil error", err: nil, constraintName: "UQE_task_buckets_task_project_view", expected: false, }, { name: "matching MySQL duplicate entry error", err: errors.New("Error 1062 (23000): Duplicate entry '424-557' for key 'UQE_task_buckets_task_project_view'"), constraintName: "UQE_task_buckets_task_project_view", expected: true, }, { name: "MySQL duplicate entry error with different constraint", err: errors.New("Error 1062 (23000): Duplicate entry '424-557' for key 'some_other_constraint'"), constraintName: "UQE_task_buckets_task_project_view", expected: false, }, { name: "non-MySQL error", err: errors.New("some other database error"), constraintName: "UQE_task_buckets_task_project_view", expected: false, }, { name: "case insensitive matching", err: errors.New("ERROR 1062 (23000): DUPLICATE ENTRY '424-557' FOR KEY 'UQE_TASK_BUCKETS_TASK_PROJECT_VIEW'"), constraintName: "uqe_task_buckets_task_project_view", expected: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { result := IsMySQLDuplicateEntryError(tt.err, tt.constraintName) assert.Equal(t, tt.expected, result) }) } } func TestIsUniqueConstraintError(t *testing.T) { tests := []struct { name string err error constraintName string expected bool }{ { name: "nil error", err: nil, constraintName: "UQE_task_buckets_task_project_view", expected: false, }, // MySQL tests { name: "MySQL duplicate entry error", err: errors.New("Error 1062 (23000): Duplicate entry '424-557' for key 'UQE_task_buckets_task_project_view'"), constraintName: "UQE_task_buckets_task_project_view", expected: true, }, { name: "MySQL duplicate entry error with different constraint", err: errors.New("Error 1062 (23000): Duplicate entry '424-557' for key 'some_other_constraint'"), constraintName: "UQE_task_buckets_task_project_view", expected: false, }, // PostgreSQL tests { name: "PostgreSQL duplicate key error", err: errors.New(`duplicate key value violates unique constraint "UQE_task_buckets_task_project_view"`), constraintName: "UQE_task_buckets_task_project_view", expected: true, }, { name: "PostgreSQL duplicate key error with different constraint", err: errors.New(`duplicate key value violates unique constraint "some_other_constraint"`), constraintName: "UQE_task_buckets_task_project_view", expected: false, }, // SQLite tests { name: "SQLite unique constraint failed", err: errors.New("UNIQUE constraint failed: task_buckets.task_id, task_buckets.project_view_id"), constraintName: "UQE_task_buckets_task_project_view", expected: true, }, { name: "SQLite constraint failed with unique", err: errors.New("constraint failed: UNIQUE constraint failed: task_buckets.task_id"), constraintName: "UQE_task_buckets_task_project_view", expected: true, // Should match on task_buckets table pattern }, // General tests { name: "non-constraint error", err: errors.New("some other database error"), constraintName: "UQE_task_buckets_task_project_view", expected: false, }, { name: "case insensitive matching - MySQL", err: errors.New("ERROR 1062 (23000): DUPLICATE ENTRY '424-557' FOR KEY 'UQE_TASK_BUCKETS_TASK_PROJECT_VIEW'"), constraintName: "uqe_task_buckets_task_project_view", expected: true, }, { name: "case insensitive matching - PostgreSQL", err: errors.New(`DUPLICATE KEY VALUE VIOLATES UNIQUE CONSTRAINT "UQE_TASK_BUCKETS_TASK_PROJECT_VIEW"`), constraintName: "uqe_task_buckets_task_project_view", expected: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { result := IsUniqueConstraintError(tt.err, tt.constraintName) assert.Equal(t, tt.expected, result) }) } }