1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
|
package persist
import (
"encoding/json"
"os"
"path/filepath"
"testing"
"github.com/crc-org/crc/v2/pkg/drivers/none"
"github.com/crc-org/crc/v2/pkg/libmachine/host"
"github.com/stretchr/testify/assert"
)
func getTestStore(t *testing.T) Filestore {
return Filestore{
MachinesDir: t.TempDir(),
}
}
func TestStoreSave(t *testing.T) {
store := getTestStore(t)
h := testHost()
assert.NoError(t, store.Save(h))
path := filepath.Join(store.MachinesDir, h.Name)
assert.DirExists(t, path)
}
func TestStoreSaveOmitRawDriver(t *testing.T) {
store := getTestStore(t)
h := testHost()
assert.NoError(t, store.Save(h))
configJSONPath := filepath.Join(store.MachinesDir, h.Name, "config.json")
configData, err := os.ReadFile(configJSONPath)
assert.NoError(t, err)
fakeHost := make(map[string]interface{})
assert.NoError(t, json.Unmarshal(configData, &fakeHost))
_, ok := fakeHost["RawDriver"]
assert.False(t, ok)
}
func TestStoreRemove(t *testing.T) {
store := getTestStore(t)
h := testHost()
assert.NoError(t, store.Save(h))
path := filepath.Join(store.MachinesDir, h.Name)
assert.DirExists(t, path)
err := store.Remove(h.Name)
assert.NoError(t, err)
assert.NoDirExists(t, path)
}
func TestStoreExists(t *testing.T) {
store := getTestStore(t)
h := testHost()
exists, err := store.Exists(h.Name)
assert.NoError(t, err)
assert.False(t, exists)
assert.NoError(t, store.Save(h))
assert.NoError(t, store.SetExists(h.Name))
exists, err = store.Exists(h.Name)
assert.NoError(t, err)
assert.True(t, exists)
assert.NoError(t, store.Remove(h.Name))
exists, err = store.Exists(h.Name)
assert.NoError(t, err)
assert.False(t, exists)
}
func TestStoreLoad(t *testing.T) {
store := getTestStore(t)
h := testHost()
assert.NoError(t, store.Save(h))
var err error
h, err = store.Load(h.Name)
assert.NoError(t, err)
rawDataDriver, ok := h.Driver.(*host.RawDataDriver)
assert.True(t, ok)
realDriver := none.NewDriver(h.Name, store.MachinesDir)
assert.NoError(t, json.Unmarshal(rawDataDriver.Data, &realDriver))
}
func testHost() *host.Host {
return &host.Host{
ConfigVersion: host.Version,
Name: "test-host",
Driver: none.NewDriver("test-host", "/tmp/artifacts"),
DriverName: "none",
}
}
|