2023-05-31 12:49:45 +00:00
|
|
|
package database
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
2023-11-03 15:55:57 +00:00
|
|
|
kv "github.com/strimertul/kilovolt/v11"
|
2023-05-31 12:49:45 +00:00
|
|
|
"go.uber.org/zap/zaptest"
|
|
|
|
)
|
|
|
|
|
|
|
|
func CreateInMemoryLocalClient(t *testing.T) (*LocalDBClient, kv.Driver) {
|
|
|
|
logger := zaptest.NewLogger(t)
|
|
|
|
|
|
|
|
// Create in-memory store and hub
|
|
|
|
inMemoryStore := kv.MakeBackend()
|
|
|
|
hub, err := kv.NewHub(inMemoryStore, kv.HubOptions{}, logger)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
go hub.Run()
|
|
|
|
|
|
|
|
// Create local client
|
|
|
|
client, err := NewLocalClient(hub, logger)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return client, inMemoryStore
|
|
|
|
}
|
|
|
|
|
|
|
|
func CleanupLocalClient(client *LocalDBClient) {
|
|
|
|
if client.hub != nil {
|
|
|
|
_ = client.Close()
|
|
|
|
client.hub.Close()
|
|
|
|
}
|
|
|
|
}
|