mirror of
https://github.com/henrygd/beszel.git
synced 2025-12-17 02:36:17 +01:00
402a1584d7
- Add version exchange between hub and agent. - Introduce ConnectionManager for managing WebSocket and SSH connections. - Implement fingerprint generation and storage in agent. - Create expiry map package to store universal tokens. - Update config.yml configuration to include tokens. - Enhance system management with new methods for handling system states and alerts. - Update front-end components to support token / fingerprint management features. - Introduce utility functions for token generation and hub URL retrieval. Co-authored-by: nhas <jordanatararimu@gmail.com>
89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
//go:build testing
|
|
// +build testing
|
|
|
|
// Package tests provides helpers for testing the application.
|
|
package tests
|
|
|
|
import (
|
|
"beszel/internal/hub"
|
|
|
|
"github.com/pocketbase/pocketbase/core"
|
|
"github.com/pocketbase/pocketbase/tests"
|
|
|
|
_ "github.com/pocketbase/pocketbase/migrations"
|
|
)
|
|
|
|
// TestHub is a wrapper hub instance used for testing.
|
|
type TestHub struct {
|
|
core.App
|
|
*tests.TestApp
|
|
*hub.Hub
|
|
}
|
|
|
|
// NewTestHub creates and initializes a test application instance.
|
|
//
|
|
// It is the caller's responsibility to call app.Cleanup() when the app is no longer needed.
|
|
func NewTestHub(optTestDataDir ...string) (*TestHub, error) {
|
|
var testDataDir string
|
|
if len(optTestDataDir) > 0 {
|
|
testDataDir = optTestDataDir[0]
|
|
}
|
|
|
|
return NewTestHubWithConfig(core.BaseAppConfig{
|
|
DataDir: testDataDir,
|
|
EncryptionEnv: "pb_test_env",
|
|
})
|
|
}
|
|
|
|
// NewTestHubWithConfig creates and initializes a test application instance
|
|
// from the provided config.
|
|
//
|
|
// If config.DataDir is not set it fallbacks to the default internal test data directory.
|
|
//
|
|
// config.DataDir is cloned for each new test application instance.
|
|
//
|
|
// It is the caller's responsibility to call app.Cleanup() when the app is no longer needed.
|
|
func NewTestHubWithConfig(config core.BaseAppConfig) (*TestHub, error) {
|
|
testApp, err := tests.NewTestAppWithConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
hub := hub.NewHub(testApp)
|
|
|
|
t := &TestHub{
|
|
App: testApp,
|
|
TestApp: testApp,
|
|
Hub: hub,
|
|
}
|
|
|
|
return t, nil
|
|
}
|
|
|
|
// Helper function to create a test user for config tests
|
|
func CreateUser(app core.App, email string, password string) (*core.Record, error) {
|
|
userCollection, err := app.FindCachedCollectionByNameOrId("users")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
user := core.NewRecord(userCollection)
|
|
user.Set("email", email)
|
|
user.Set("password", password)
|
|
|
|
return user, app.Save(user)
|
|
}
|
|
|
|
// Helper function to create a test record
|
|
func CreateRecord(app core.App, collectionName string, fields map[string]any) (*core.Record, error) {
|
|
collection, err := app.FindCachedCollectionByNameOrId(collectionName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
record := core.NewRecord(collection)
|
|
record.Load(fields)
|
|
|
|
return record, app.Save(record)
|
|
}
|