Files
HackBrowserData/browser/firefox/extract_bookmark_test.go
T
Roger b3bbc0dadf feat: add CountEntries to skip decryption for list --detail (#562)
* feat: add CountEntries to skip decryption for list --detail (#549)
* test: add CountEntries and countCategory tests at browser level
* fix: address review feedback on CountRows and countLocalStorage
* test: add CountRows unit tests
2026-04-07 22:28:39 +08:00

52 lines
1.3 KiB
Go

package firefox
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func setupMozBookmarkDB(t *testing.T) string {
t.Helper()
return createTestDB(t, "places.sqlite", []string{mozPlacesSchema, mozBookmarksSchema},
insertMozPlace(1, "https://go.dev", "Go", 0, 0),
insertMozPlace(2, "https://github.com", "GitHub", 0, 0),
insertMozBookmark(1, 1, 1, "Go Website", 1700000000000000),
insertMozBookmark(2, 2, 1, "GitHub", 1710000000000000),
)
}
func TestExtractBookmarks(t *testing.T) {
path := setupMozBookmarkDB(t)
got, err := extractBookmarks(path)
require.NoError(t, err)
require.Len(t, got, 2)
// Verify sort order: dateAdded descending
assert.Equal(t, "GitHub", got[0].Name)
assert.Equal(t, "Go Website", got[1].Name)
// Verify field mapping
assert.Equal(t, "https://github.com", got[0].URL)
assert.Equal(t, "url", got[0].Folder) // type=1 → "url"
assert.False(t, got[0].CreatedAt.IsZero())
}
func TestCountBookmarks(t *testing.T) {
path := setupMozBookmarkDB(t)
count, err := countBookmarks(path)
require.NoError(t, err)
assert.Equal(t, 2, count)
}
func TestCountBookmarks_Empty(t *testing.T) {
path := createTestDB(t, "places.sqlite", []string{mozPlacesSchema, mozBookmarksSchema})
count, err := countBookmarks(path)
require.NoError(t, err)
assert.Equal(t, 0, count)
}