From c6f941adb37dd008e6602c78a259db9b8aec9131 Mon Sep 17 00:00:00 2001
From: Eva Ho <hoyyeva@gmail.com>
Date: Thu, 18 Dec 2025 10:29:07 -0500
Subject: [PATCH] fix test

---
 app/store/database.go       | 14 +++-----------
 app/updater/updater_test.go | 10 ++++++++++
 app/wintray/tray.go         |  1 +
 3 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/app/store/database.go b/app/store/database.go
index 3443f974b..b0b1155c2 100644
--- a/app/store/database.go
+++ b/app/store/database.go
@@ -9,7 +9,7 @@ import (
 	"strings"
 	"time"
 
-	sqlite3 "github.com/mattn/go-sqlite3"
+	_ "github.com/mattn/go-sqlite3"
 )
 
 // currentSchemaVersion defines the current database schema version.
@@ -504,19 +504,11 @@ func (db *database) cleanupOrphanedData() error {
 }
 
 func duplicateColumnError(err error) bool {
-	if sqlite3Err, ok := err.(sqlite3.Error); ok {
-		return sqlite3Err.Code == sqlite3.ErrError &&
-			strings.Contains(sqlite3Err.Error(), "duplicate column name")
-	}
-	return false
+	return err != nil && strings.Contains(err.Error(), "duplicate column name")
 }
 
 func columnNotExists(err error) bool {
-	if sqlite3Err, ok := err.(sqlite3.Error); ok {
-		return sqlite3Err.Code == sqlite3.ErrError &&
-			strings.Contains(sqlite3Err.Error(), "no such column")
-	}
-	return false
+	return err != nil && strings.Contains(err.Error(), "no such column")
 }
 
 func (db *database) getAllChats() ([]Chat, error) {
diff --git a/app/updater/updater_test.go b/app/updater/updater_test.go
index dea820c28..b4e8bea6b 100644
--- a/app/updater/updater_test.go
+++ b/app/updater/updater_test.go
@@ -86,6 +86,16 @@ func TestBackgoundChecker(t *testing.T) {
 
 	updater := &Updater{Store: &store.Store{}}
 	defer updater.Store.Close() // Ensure database is closed
+	
+	settings, err := updater.Store.Settings()
+	if err != nil {
+		t.Fatal(err)
+	}
+	settings.AutoUpdateEnabled = true
+	if err := updater.Store.SetSettings(settings); err != nil {
+		t.Fatal(err)
+	}
+	
 	updater.StartBackgroundUpdaterChecker(ctx, cb)
 	select {
 	case <-stallTimer.C:
diff --git a/app/wintray/tray.go b/app/wintray/tray.go
index 0c1a2ed0f..0271f47d4 100644
--- a/app/wintray/tray.go
+++ b/app/wintray/tray.go
@@ -41,6 +41,7 @@ type TrayCallbacks interface {
 	Quit()
 	TrayRun()
 	UpdateAvailable(ver string) error
+	ClearUpdateAvailable() error
 	GetIconHandle() windows.Handle
 }