onepassword-operator/pkg/onepassword/connect_setup_test.go

package onepassword

import (
	"context"
	"testing"

	appsv1 "k8s.io/api/apps/v1"
	corev1 "k8s.io/api/core/v1"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/types"
	"k8s.io/kubectl/pkg/scheme"
	"sigs.k8s.io/controller-runtime/pkg/client/fake"
)

var defaultNamespacedName = types.NamespacedName{Name: "onepassword-connect", Namespace: "default"}

func TestServiceSetup(t *testing.T) {
	ctx := context.Background()

	// Register operator types with the runtime scheme.
	s := scheme.Scheme

	// Objects to track in the fake client.
	objs := []runtime.Object{}

	// Create a fake client to mock API calls.
	client := fake.NewClientBuilder().WithScheme(s).WithRuntimeObjects(objs...).Build()

	err := setupService(ctx, client, "../../config/connect/service.yaml", defaultNamespacedName.Namespace)

	if err != nil {
		t.Errorf("Error Setting Up Connect: %v", err)
	}

	// check that service was created
	service := &corev1.Service{}
	err = client.Get(ctx, defaultNamespacedName, service)
	if err != nil {
		t.Errorf("Error Setting Up Connect service: %v", err)
	}
}

func TestDeploymentSetup(t *testing.T) {
	ctx := context.Background()

	// Register operator types with the runtime scheme.
	s := scheme.Scheme

	// Objects to track in the fake client.
	objs := []runtime.Object{}

	// Create a fake client to mock API calls.
	client := fake.NewClientBuilder().WithScheme(s).WithRuntimeObjects(objs...).Build()

	err := setupDeployment(ctx, client, "../../config/connect/deployment.yaml", defaultNamespacedName.Namespace)

	if err != nil {
		t.Errorf("Error Setting Up Connect: %v", err)
	}

	// check that deployment was created
	deployment := &appsv1.Deployment{}
	err = client.Get(ctx, defaultNamespacedName, deployment)
	if err != nil {
		t.Errorf("Error Setting Up Connect deployment: %v", err)
	}
}