mirror of
https://github.com/moonD4rk/HackBrowserData.git
synced 2026-05-19 18:58:03 +02:00
Roger
130 lines
4.3 KiB
Go
130 lines
4.3 KiB
Go
package safari
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/moond4rk/plist"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// testExtensionEntry mirrors the shape of one entry in Safari's Extensions.plist:
|
|
// an untyped dictionary keyed by string. Using a map (instead of safariExtension)
|
|
// lets tests omit keys like Enabled for AppExtension-style fixtures, matching
|
|
// what Safari actually writes.
|
|
type testExtensionEntry map[string]any
|
|
|
|
// writeTestExtensionsPlist writes an Extensions.plist under
|
|
// <container>/Safari/<subdir>/Extensions.plist. subdir is either
|
|
// "AppExtensions" or "WebExtensions".
|
|
func writeTestExtensionsPlist(t *testing.T, container, subdir string, entries map[string]testExtensionEntry) {
|
|
t.Helper()
|
|
dir := filepath.Join(container, safariExtensionsSubdir, subdir)
|
|
require.NoError(t, os.MkdirAll(dir, 0o755))
|
|
|
|
f, err := os.Create(filepath.Join(dir, safariExtensionsPlistFile))
|
|
require.NoError(t, err)
|
|
defer f.Close()
|
|
require.NoError(t, plist.NewBinaryEncoder(f).Encode(entries))
|
|
}
|
|
|
|
func TestExtractExtensions_AppAndWebMerged(t *testing.T) {
|
|
container := t.TempDir()
|
|
writeTestExtensionsPlist(t, container, safariAppExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.colliderli.iina.OpenInIINA (67CQ77V27R)": {},
|
|
})
|
|
writeTestExtensionsPlist(t, container, safariWebExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.1password.safari.extension (2BUA8C4S2C)": {"Enabled": true},
|
|
})
|
|
|
|
got, err := extractExtensions(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, got, 2)
|
|
|
|
// Results are sorted by key, so 1Password (com.1…) comes before iina (com.c…).
|
|
assert.Equal(t, "com.1password.safari.extension", got[0].Name)
|
|
assert.Equal(t, "com.1password.safari.extension (2BUA8C4S2C)", got[0].ID)
|
|
assert.True(t, got[0].Enabled)
|
|
|
|
assert.Equal(t, "com.colliderli.iina.OpenInIINA", got[1].Name)
|
|
assert.Equal(t, "com.colliderli.iina.OpenInIINA (67CQ77V27R)", got[1].ID)
|
|
// AppExtensions omit the Enabled field — defaults to true (present == enabled).
|
|
assert.True(t, got[1].Enabled)
|
|
}
|
|
|
|
func TestExtractExtensions_EnabledFlag(t *testing.T) {
|
|
container := t.TempDir()
|
|
writeTestExtensionsPlist(t, container, safariWebExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.example.a (AAAAAAAAAA)": {"Enabled": true},
|
|
"com.example.b (BBBBBBBBBB)": {"Enabled": false},
|
|
})
|
|
|
|
got, err := extractExtensions(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, got, 2)
|
|
assert.True(t, got[0].Enabled)
|
|
assert.False(t, got[1].Enabled)
|
|
}
|
|
|
|
func TestExtractExtensions_BundleIDFallbackOnUnexpectedKey(t *testing.T) {
|
|
container := t.TempDir()
|
|
writeTestExtensionsPlist(t, container, safariAppExtensionsSubdir, map[string]testExtensionEntry{
|
|
"legacy-key-without-team-id": {},
|
|
})
|
|
|
|
got, err := extractExtensions(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, got, 1)
|
|
// Regex miss → fall back to the full trimmed key.
|
|
assert.Equal(t, "legacy-key-without-team-id", got[0].Name)
|
|
assert.Equal(t, "legacy-key-without-team-id", got[0].ID)
|
|
}
|
|
|
|
func TestExtractExtensions_OnlyAppExt(t *testing.T) {
|
|
container := t.TempDir()
|
|
writeTestExtensionsPlist(t, container, safariAppExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.example.only (XXXXXXXXX1)": {},
|
|
})
|
|
|
|
got, err := extractExtensions(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, got, 1)
|
|
assert.Equal(t, "com.example.only", got[0].Name)
|
|
}
|
|
|
|
func TestExtractExtensions_OnlyWebExt(t *testing.T) {
|
|
container := t.TempDir()
|
|
writeTestExtensionsPlist(t, container, safariWebExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.example.web (XXXXXXXXX2)": {"Enabled": true},
|
|
})
|
|
|
|
got, err := extractExtensions(container)
|
|
require.NoError(t, err)
|
|
require.Len(t, got, 1)
|
|
assert.Equal(t, "com.example.web", got[0].Name)
|
|
}
|
|
|
|
func TestExtractExtensions_NoPlists(t *testing.T) {
|
|
container := t.TempDir()
|
|
got, err := extractExtensions(container)
|
|
require.NoError(t, err)
|
|
assert.Empty(t, got)
|
|
}
|
|
|
|
func TestCountExtensions(t *testing.T) {
|
|
container := t.TempDir()
|
|
writeTestExtensionsPlist(t, container, safariAppExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.example.a (AAAAAAAAAA)": {},
|
|
"com.example.b (BBBBBBBBBB)": {},
|
|
})
|
|
writeTestExtensionsPlist(t, container, safariWebExtensionsSubdir, map[string]testExtensionEntry{
|
|
"com.example.c (CCCCCCCCCC)": {"Enabled": true},
|
|
})
|
|
|
|
count, err := countExtensions(container)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 3, count)
|
|
}
|