116 lines
4.2 KiB
Go
116 lines
4.2 KiB
Go
package display
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
core "dappco.re/go/core"
|
|
"forge.lthn.ai/core/gui/pkg/chat"
|
|
"forge.lthn.ai/core/gui/pkg/window"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDisplay_Good_WindowOpenIncludesPreload(t *testing.T) {
|
|
platform := window.NewMockPlatform()
|
|
c := core.New(
|
|
core.WithService(Register(nil)),
|
|
core.WithService(window.Register(platform)),
|
|
core.WithServiceLock(),
|
|
)
|
|
require.True(t, c.ServiceStartup(context.Background(), nil).OK)
|
|
|
|
result := c.Action("window.open").Run(context.Background(), core.NewOptions(
|
|
core.Option{Key: "task", Value: window.TaskOpenWindow{
|
|
Options: []window.WindowOption{
|
|
window.WithName("preload"),
|
|
window.WithURL("https://example.com"),
|
|
},
|
|
}},
|
|
))
|
|
require.True(t, result.OK)
|
|
require.Len(t, platform.Windows, 1)
|
|
assert.NotEmpty(t, platform.Windows[0].ExecJSCalls())
|
|
assert.Contains(t, platform.Windows[0].ExecJSCalls()[0], "globalThis.core.ml")
|
|
assert.Contains(t, platform.Windows[0].ExecJSCalls()[0], "globalThis.core.storage.cookies")
|
|
assert.Contains(t, platform.Windows[0].ExecJSCalls()[0], "Document.prototype, 'cookie'")
|
|
assert.NotContains(t, platform.Windows[0].ExecJSCalls()[0], "globalThis.electron")
|
|
assert.NotContains(t, platform.Windows[0].ExecJSCalls()[0], "core.background.serviceWorker.register")
|
|
}
|
|
|
|
func TestDisplay_Good_WindowOpenTrustedOriginIncludesPrivilegedBridge(t *testing.T) {
|
|
platform := window.NewMockPlatform()
|
|
c := core.New(
|
|
core.WithService(Register(nil)),
|
|
core.WithService(window.Register(platform)),
|
|
core.WithServiceLock(),
|
|
)
|
|
require.True(t, c.ServiceStartup(context.Background(), nil).OK)
|
|
|
|
result := c.Action("window.open").Run(context.Background(), core.NewOptions(
|
|
core.Option{Key: "task", Value: window.TaskOpenWindow{
|
|
Options: []window.WindowOption{
|
|
window.WithName("preload"),
|
|
window.WithURL("http://localhost:3000"),
|
|
},
|
|
}},
|
|
))
|
|
require.True(t, result.OK)
|
|
require.Len(t, platform.Windows, 1)
|
|
script := platform.Windows[0].ExecJSCalls()[0]
|
|
assert.Contains(t, script, "globalThis.electron")
|
|
assert.Contains(t, script, "core.background.serviceWorker.register")
|
|
assert.Contains(t, script, "globalThis.core.ml")
|
|
}
|
|
|
|
func TestDisplay_Good_CoreSchemeRoutesThroughBackend(t *testing.T) {
|
|
platform := window.NewMockPlatform()
|
|
c := core.New(
|
|
core.WithService(Register(nil)),
|
|
core.WithService(chat.Register(func(o *chat.Options) { o.StorePath = filepath.Join(t.TempDir(), "chat.db") })),
|
|
core.WithService(window.Register(platform)),
|
|
core.WithServiceLock(),
|
|
)
|
|
require.True(t, c.ServiceStartup(context.Background(), nil).OK)
|
|
|
|
require.True(t, c.Action("window.open").Run(context.Background(), core.NewOptions(
|
|
core.Option{Key: "task", Value: window.TaskOpenWindow{
|
|
Options: []window.WindowOption{window.WithName("settings")},
|
|
}},
|
|
)).OK)
|
|
|
|
require.True(t, c.Action("window.setURL").Run(context.Background(), core.NewOptions(
|
|
core.Option{Key: "task", Value: window.TaskSetURL{Name: "settings", URL: "core://settings"}},
|
|
)).OK)
|
|
|
|
require.Len(t, platform.Windows, 1)
|
|
assert.True(t, strings.Contains(platform.Windows[0].HTMLContent(), "core://settings"))
|
|
}
|
|
|
|
func TestPreload_ValidatedLocalMLAPIURL_Good(t *testing.T) {
|
|
assert.Equal(t, "http://localhost:8090", validatedLocalMLAPIURL("http://localhost:8090/"))
|
|
assert.Equal(t, "https://127.0.0.1:9443", validatedLocalMLAPIURL("https://127.0.0.1:9443/"))
|
|
}
|
|
|
|
func TestPreload_ValidatedLocalMLAPIURL_Bad(t *testing.T) {
|
|
assert.Equal(t, "http://localhost:8090", validatedLocalMLAPIURL("https://example.com"))
|
|
assert.Equal(t, "http://localhost:8090", validatedLocalMLAPIURL("ftp://localhost:8090"))
|
|
}
|
|
|
|
func TestPreload_ValidatedLocalMLAPIURL_Ugly(t *testing.T) {
|
|
assert.Equal(t, "http://localhost:8090", validatedLocalMLAPIURL(""))
|
|
assert.Equal(t, "http://localhost:8090", validatedLocalMLAPIURL("not a url"))
|
|
}
|
|
|
|
func TestPreload_TrustedPreloadOrigin_Good(t *testing.T) {
|
|
assert.True(t, trustedPreloadOrigin("core://store"))
|
|
assert.True(t, trustedPreloadOrigin("http://localhost:3000"))
|
|
assert.True(t, trustedPreloadOrigin("https://127.0.0.1:8443"))
|
|
}
|
|
|
|
func TestPreload_TrustedPreloadOrigin_Bad(t *testing.T) {
|
|
assert.False(t, trustedPreloadOrigin("https://example.com"))
|
|
assert.False(t, trustedPreloadOrigin("http://10.0.0.1:3000"))
|
|
}
|