Add basic test coverage
Rob Watson
10 months ago
parent
d38f31c3d9
commit
4f6b2d1550
@ -0,0 +1,26 @@
|
||||
package logs_test
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"git.netflux.io/rob/kubectl-persistent-logger/logs"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
testclient "k8s.io/client-go/kubernetes/fake"
|
||||
)
|
||||
|
||||
func TestStream(t *testing.T) {
|
||||
client := logs.KubernetesClient{Interface: testclient.NewSimpleClientset()}
|
||||
|
||||
var buf bytes.Buffer
|
||||
pod := corev1.Pod{ObjectMeta: metav1.ObjectMeta{Name: "foo", Namespace: "default"}}
|
||||
stream := logs.NewStream(client, &pod, &buf)
|
||||
|
||||
err := stream.Copy(context.Background())
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, "[foo] fake logs\n", buf.String())
|
||||
}
|
||||
@ -0,0 +1,70 @@
|
||||
package logs_test
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"git.netflux.io/rob/kubectl-persistent-logger/logs"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
appsv1 "k8s.io/api/apps/v1"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/watch"
|
||||
testclient "k8s.io/client-go/kubernetes/fake"
|
||||
k8stest "k8s.io/client-go/testing"
|
||||
)
|
||||
|
||||
func TestWatcher(t *testing.T) {
|
||||
clientset := testclient.NewSimpleClientset(
|
||||
&appsv1.Deployment{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: "mydeployment",
|
||||
Namespace: "default",
|
||||
},
|
||||
Spec: appsv1.DeploymentSpec{
|
||||
Selector: &metav1.LabelSelector{
|
||||
MatchLabels: map[string]string{"mylabelname": "mylabelvalue"},
|
||||
},
|
||||
},
|
||||
},
|
||||
)
|
||||
|
||||
k8swatcher := watch.NewFake()
|
||||
clientset.PrependWatchReactor("pods", k8stest.DefaultWatchReactor(k8swatcher, nil))
|
||||
pods := []*corev1.Pod{
|
||||
{
|
||||
ObjectMeta: metav1.ObjectMeta{Name: "foo", Namespace: "default"},
|
||||
Status: corev1.PodStatus{Phase: corev1.PodRunning},
|
||||
},
|
||||
{
|
||||
ObjectMeta: metav1.ObjectMeta{Name: "bar", Namespace: "default"},
|
||||
Status: corev1.PodStatus{Phase: corev1.PodRunning},
|
||||
},
|
||||
{
|
||||
ObjectMeta: metav1.ObjectMeta{Name: "baz", Namespace: "default"},
|
||||
Status: corev1.PodStatus{Phase: corev1.PodPending},
|
||||
},
|
||||
}
|
||||
|
||||
go func() {
|
||||
defer k8swatcher.Stop()
|
||||
for _, pod := range pods {
|
||||
time.Sleep(time.Millisecond * 300)
|
||||
k8swatcher.Add(pod)
|
||||
}
|
||||
}()
|
||||
|
||||
client := logs.KubernetesClient{Interface: clientset}
|
||||
|
||||
var buf bytes.Buffer
|
||||
watcher := logs.NewWatcher("mydeployment", client, &buf)
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
||||
defer cancel()
|
||||
err := watcher.Watch(ctx)
|
||||
require.EqualError(t, err, context.DeadlineExceeded.Error())
|
||||
assert.Equal(t, "[foo] fake logs\n[bar] fake logs\n", buf.String())
|
||||
}
|
||||
Loading…
Reference in New Issue