
Object Cluster will be a entry point to intractions with kubernetes It will provide: - kubectl abstraction which will emulate kubectl apply and other commands (in the future) - kubernets client-go kubernetes interface which has all operations, including CRUD to kubernetes apps, core, extentions, etc... objects Change-Id: Ie2961f68a160e720c264c622c1124f283bded161
110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
package kubectl_test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"sigs.k8s.io/kustomize/v3/pkg/fs"
|
|
|
|
"opendev.org/airship/airshipctl/pkg/k8s/kubectl"
|
|
k8sutils "opendev.org/airship/airshipctl/pkg/k8s/utils"
|
|
"opendev.org/airship/airshipctl/testutil"
|
|
k8stest "opendev.org/airship/airshipctl/testutil/k8sutils"
|
|
)
|
|
|
|
var (
|
|
kubeconfigPath = "testdata/kubeconfig.yaml"
|
|
fixtureDir = "testdata/"
|
|
|
|
writeOutError = errors.New("writeOutError")
|
|
TempFileError = errors.New("TempFileError")
|
|
)
|
|
|
|
type MockFileSystem struct {
|
|
MockRemoveAll func() error
|
|
MockTempFile func() (kubectl.File, error)
|
|
fs.FileSystem
|
|
}
|
|
|
|
func (fsys MockFileSystem) RemoveAll(name string) error { return fsys.MockRemoveAll() }
|
|
func (fsys MockFileSystem) TempFile(bufferDir string, prefix string) (kubectl.File, error) {
|
|
return fsys.MockTempFile()
|
|
}
|
|
|
|
type TestFile struct {
|
|
kubectl.File
|
|
MockName func() string
|
|
MockWrite func() (int, error)
|
|
MockClose func() error
|
|
}
|
|
|
|
func (f TestFile) Name() string { return f.MockName() }
|
|
func (f TestFile) Write([]byte) (int, error) { return f.MockWrite() }
|
|
func (f TestFile) Close() error { return f.MockClose() }
|
|
|
|
func TestNewKubectlFromKubeconfigPath(t *testing.T) {
|
|
f := k8sutils.FactoryFromKubeconfigPath(kubeconfigPath)
|
|
kctl := kubectl.NewKubectl(f).WithBufferDir("/tmp/.airship")
|
|
|
|
assert.NotNil(t, kctl.Factory)
|
|
assert.NotNil(t, kctl.FileSystem)
|
|
assert.NotNil(t, kctl.IOStreams)
|
|
}
|
|
|
|
func TestApply(t *testing.T) {
|
|
f := k8stest.NewFakeFactoryForRC(t, filenameRC)
|
|
defer f.Cleanup()
|
|
kctl := kubectl.NewKubectl(f).WithBufferDir("/tmp/.airship")
|
|
kctl.Factory = f
|
|
ao, err := kctl.ApplyOptions()
|
|
require.NoError(t, err, "failed to get documents from bundle")
|
|
ao.SetDryRun(true)
|
|
|
|
b := testutil.NewTestBundle(t, fixtureDir)
|
|
docs, err := b.GetByAnnotation("airshipit.org/initinfra")
|
|
require.NoError(t, err, "failed to get documents from bundle")
|
|
|
|
tests := []struct {
|
|
name string
|
|
expectedErr error
|
|
fs kubectl.FileSystem
|
|
}{
|
|
{
|
|
expectedErr: nil,
|
|
fs: MockFileSystem{
|
|
MockRemoveAll: func() error { return nil },
|
|
MockTempFile: func() (kubectl.File, error) {
|
|
return TestFile{
|
|
MockName: func() string { return filenameRC },
|
|
MockWrite: func() (int, error) { return 0, nil },
|
|
MockClose: func() error { return nil },
|
|
}, nil
|
|
},
|
|
},
|
|
},
|
|
{
|
|
expectedErr: writeOutError,
|
|
fs: MockFileSystem{
|
|
MockTempFile: func() (kubectl.File, error) { return nil, writeOutError }},
|
|
},
|
|
{
|
|
expectedErr: TempFileError,
|
|
fs: MockFileSystem{
|
|
MockRemoveAll: func() error { return nil },
|
|
MockTempFile: func() (kubectl.File, error) {
|
|
return TestFile{
|
|
MockWrite: func() (int, error) { return 0, TempFileError },
|
|
MockName: func() string { return filenameRC },
|
|
MockClose: func() error { return nil },
|
|
}, nil
|
|
}},
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
kctl.FileSystem = test.fs
|
|
assert.Equal(t, kctl.Apply(docs, ao), test.expectedErr)
|
|
}
|
|
}
|