mirror of
https://github.com/kairos-io/provider-kairos.git
synced 2025-04-27 19:35:36 +00:00
72c01e2657
The path in `go.mod` should be ended in `/v2` suffix, as per [go.mod module version numbers](https://go.dev/doc/modules/version-numbers). --------- Signed-off-by: Mateusz Urbanek <mateusz.urbanek.98@gmail.com>
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package cli_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/kairos-io/kairos/v2/pkg/config"
|
|
. "github.com/kairos-io/provider-kairos/v2/internal/cli"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type TConfig struct {
|
|
Kairos struct {
|
|
NetworkToken string `yaml:"network_token"`
|
|
} `yaml:"p2p"`
|
|
}
|
|
|
|
var _ = Describe("Get config", func() {
|
|
Context("directory", func() {
|
|
|
|
It("replace token in config files", func() {
|
|
|
|
var cc string = `#node-config
|
|
p2p:
|
|
network_token: "foo"
|
|
|
|
bb:
|
|
nothing: "foo"
|
|
`
|
|
d, _ := ioutil.TempDir("", "xxxx")
|
|
defer os.RemoveAll(d)
|
|
|
|
err := ioutil.WriteFile(filepath.Join(d, "test"), []byte(cc), os.ModePerm)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
err = ioutil.WriteFile(filepath.Join(d, "b"), []byte(`
|
|
fooz: "bar"
|
|
`), os.ModePerm)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = ReplaceToken([]string{d, "/doesnotexist"}, "baz")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
content, err := ioutil.ReadFile(filepath.Join(d, "test"))
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
res := map[interface{}]interface{}{}
|
|
|
|
err = yaml.Unmarshal(content, &res)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(res).To(Equal(map[interface{}]interface{}{
|
|
"p2p": map[interface{}]interface{}{"network_token": "baz"},
|
|
"bb": map[interface{}]interface{}{"nothing": "foo"},
|
|
}))
|
|
|
|
hasHeader, _ := config.HasHeader(string(content), "#node-config")
|
|
Expect(hasHeader).To(BeTrue(), string(content))
|
|
})
|
|
})
|
|
})
|