mirror of https://github.com/k3s-io/k3s
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
|
package tests
|
||
|
|
||
|
import (
|
||
|
"net"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
|
||
|
"github.com/k3s-io/k3s/pkg/daemons/config"
|
||
|
"github.com/k3s-io/k3s/pkg/daemons/control/deps"
|
||
|
)
|
||
|
|
||
|
// GenerateDataDir creates a temporary directory at "/tmp/k3s/<RANDOM_STRING>/".
|
||
|
// The latest directory created with this function is soft linked to "/tmp/k3s/latest/".
|
||
|
// This allows tests to replicate the "/var/lib/rancher/k3s" directory structure.
|
||
|
func GenerateDataDir(cnf *config.Control) error {
|
||
|
if err := os.MkdirAll(cnf.DataDir, 0700); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
testDir, err := os.MkdirTemp(cnf.DataDir, "*")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
// Remove old symlink and add new one
|
||
|
os.Remove(filepath.Join(cnf.DataDir, "latest"))
|
||
|
if err = os.Symlink(testDir, filepath.Join(cnf.DataDir, "latest")); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
cnf.DataDir = testDir
|
||
|
cnf.DataDir, err = filepath.Abs(cnf.DataDir)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// CleanupDataDir removes the associated "/tmp/k3s/<RANDOM_STRING>"
|
||
|
// directory along with the 'latest' symlink that points at it.
|
||
|
func CleanupDataDir(cnf *config.Control) {
|
||
|
os.Remove(filepath.Join(cnf.DataDir, "..", "latest"))
|
||
|
os.RemoveAll(cnf.DataDir)
|
||
|
}
|
||
|
|
||
|
// GenerateRuntime creates a temporary data dir and configures
|
||
|
// config.ControlRuntime with all the appropriate certificate keys.
|
||
|
func GenerateRuntime(cnf *config.Control) error {
|
||
|
cnf.Runtime = &config.ControlRuntime{}
|
||
|
if err := GenerateDataDir(cnf); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
os.MkdirAll(filepath.Join(cnf.DataDir, "tls"), 0700)
|
||
|
os.MkdirAll(filepath.Join(cnf.DataDir, "cred"), 0700)
|
||
|
|
||
|
deps.CreateRuntimeCertFiles(cnf)
|
||
|
|
||
|
return deps.GenServerDeps(cnf)
|
||
|
}
|
||
|
|
||
|
func ClusterIPNet() *net.IPNet {
|
||
|
_, clusterIPNet, _ := net.ParseCIDR("10.42.0.0/16")
|
||
|
return clusterIPNet
|
||
|
}
|
||
|
|
||
|
func ServiceIPNet() *net.IPNet {
|
||
|
_, serviceIPNet, _ := net.ParseCIDR("10.43.0.0/16")
|
||
|
return serviceIPNet
|
||
|
}
|