mirror of https://github.com/fluxcd/cli-utils.git
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
// Copyright 2019 The Kubernetes Authors.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package task
|
|
|
|
import (
|
|
"bytes"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
appsv1 "k8s.io/api/apps/v1"
|
|
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"sigs.k8s.io/cli-utils/pkg/apply/event"
|
|
)
|
|
|
|
func TestKubectlPrinterAdapter(t *testing.T) {
|
|
ch := make(chan event.Event)
|
|
buffer := bytes.Buffer{}
|
|
operation := "serverside-applied"
|
|
|
|
adapter := KubectlPrinterAdapter{
|
|
ch: ch,
|
|
}
|
|
|
|
toPrinterFunc := adapter.toPrinterFunc()
|
|
resourcePrinter, err := toPrinterFunc(operation)
|
|
assert.NoError(t, err)
|
|
|
|
deployment := appsv1.Deployment{
|
|
TypeMeta: v1.TypeMeta{
|
|
APIVersion: "apps/v1",
|
|
Kind: "Deployment",
|
|
},
|
|
ObjectMeta: v1.ObjectMeta{
|
|
Name: "name",
|
|
Namespace: "namespace",
|
|
},
|
|
}
|
|
|
|
// Need to run this in a separate gorutine since go channels
|
|
// are blocking.
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
err = resourcePrinter.PrintObj(&deployment, &buffer)
|
|
}()
|
|
msg := <-ch
|
|
wg.Wait()
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, event.ServersideApplied, msg.ApplyEvent.Operation)
|
|
assert.Equal(t, &deployment, msg.ApplyEvent.Object)
|
|
}
|