Skip to content
Merged
Show file tree
Hide file tree
Changes from 11 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 46 additions & 0 deletions internal/repository/manifest/manifest_repo.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
package manifest

import (
"context"
"fmt"

apimetav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"sigs.k8s.io/controller-runtime/pkg/client"

"github.com/kyma-project/lifecycle-manager/api/shared"
"github.com/kyma-project/lifecycle-manager/api/v1beta2"
)

type Repository struct {
clnt client.Client
namespace string
}

func NewRepository(clnt client.Client, namespace string) *Repository {
return &Repository{
clnt: clnt,
namespace: namespace,
}
}

func (r *Repository) DeleteAllForModule(ctx context.Context, moduleName string) error {
err := r.clnt.DeleteAllOf(ctx, &v1beta2.Manifest{}, client.InNamespace(r.namespace),
client.MatchingLabels{shared.ModuleName: moduleName})
if err != nil {
return fmt.Errorf("failed to delete all manifests for module %s: %w", moduleName, err)
}
return nil
}

func (r *Repository) ListAllForModule(ctx context.Context, moduleName string) (
[]apimetav1.PartialObjectMetadata, error,
) {
var manifestList apimetav1.PartialObjectMetadataList
manifestList.SetGroupVersionKind(v1beta2.GroupVersion.WithKind("ManifestList"))

if err := r.clnt.List(ctx, &manifestList, client.InNamespace(r.namespace),
client.MatchingLabels{shared.ModuleName: moduleName}); err != nil {
return nil, fmt.Errorf("failed to list Manifests for module %s: %w", moduleName, err)
}
return manifestList.Items, nil
}
170 changes: 170 additions & 0 deletions internal/repository/manifest/manifest_repo_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,170 @@
package manifest_test

import (
"context"
"errors"
"testing"

"github.com/kyma-project/lifecycle-manager/pkg/testutils/random"

Check failure on line 8 in internal/repository/manifest/manifest_repo_test.go

View workflow job for this annotation

GitHub Actions / lint

File is not properly formatted (gci)
"github.com/stretchr/testify/require"
apimetav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"sigs.k8s.io/controller-runtime/pkg/client"

"github.com/kyma-project/lifecycle-manager/api/shared"
"github.com/kyma-project/lifecycle-manager/api/v1beta2"
"github.com/kyma-project/lifecycle-manager/internal/repository/manifest"
)

type clientStub struct {
client.Client

deleteAllOfCalled bool
listCalled bool
deleteAllOfErr error
listErr error

capturedNamespace string
capturedLabels map[string]string
capturedObjectType client.Object

partialObjectMetadata []apimetav1.PartialObjectMetadata
}

func (c *clientStub) DeleteAllOf(_ context.Context, obj client.Object, opts ...client.DeleteAllOfOption) error {
c.deleteAllOfCalled = true
c.capturedObjectType = obj

for _, opt := range opts {
if nsOpt, ok := opt.(client.InNamespace); ok {
c.capturedNamespace = string(nsOpt)
}
if labelOpt, ok := opt.(client.MatchingLabels); ok {
c.capturedLabels = labelOpt
}
}

return c.deleteAllOfErr
}

func (c *clientStub) List(_ context.Context, list client.ObjectList, opts ...client.ListOption) error {
c.listCalled = true

for _, opt := range opts {
if nsOpt, ok := opt.(client.InNamespace); ok {
c.capturedNamespace = string(nsOpt)
}
if labelOpt, ok := opt.(client.MatchingLabels); ok {
c.capturedLabels = labelOpt
}
}

if c.listErr != nil {
return c.listErr
}

if partialList, ok := list.(*apimetav1.PartialObjectMetadataList); ok {
partialList.Items = c.partialObjectMetadata
}

return nil
}

func TestRepository_DeleteAllForModule(t *testing.T) {
ctx := context.Background()
testNamespace := random.Name()
testModuleName := random.Name()

t.Run("successfully deletes all manifests for module", func(t *testing.T) {
stub := &clientStub{}
repo := manifest.NewRepository(stub, testNamespace)

err := repo.DeleteAllForModule(ctx, testModuleName)

require.NoError(t, err)
require.True(t, stub.deleteAllOfCalled)
require.Equal(t, testNamespace, stub.capturedNamespace)
require.Equal(t, testModuleName, stub.capturedLabels[shared.ModuleName])
require.IsType(t, &v1beta2.Manifest{}, stub.capturedObjectType)
})

t.Run("returns error when deleteAllOf fails", func(t *testing.T) {
expectedErr := errors.New("delete error")
stub := &clientStub{deleteAllOfErr: expectedErr}
repo := manifest.NewRepository(stub, testNamespace)

err := repo.DeleteAllForModule(ctx, testModuleName)

require.Error(t, err)
require.Contains(t, err.Error(), "failed to delete all manifests for module")
require.Contains(t, err.Error(), testModuleName)
require.True(t, stub.deleteAllOfCalled)
require.Equal(t, testNamespace, stub.capturedNamespace)
require.Equal(t, testModuleName, stub.capturedLabels[shared.ModuleName])
})
}

func TestRepository_ListAllForModule(t *testing.T) {
ctx := context.Background()
testNamespace := random.Name()
testModuleName := random.Name()

t.Run("successfully lists all manifests for module", func(t *testing.T) {
expectedMetadata := []apimetav1.PartialObjectMetadata{
{
ObjectMeta: apimetav1.ObjectMeta{
Name: "manifest1",
Namespace: testNamespace,
Labels: map[string]string{shared.ModuleName: testModuleName},
},
},
{
ObjectMeta: apimetav1.ObjectMeta{
Name: "manifest2",
Namespace: testNamespace,
Labels: map[string]string{shared.ModuleName: testModuleName},
},
},
}

stub := &clientStub{partialObjectMetadata: expectedMetadata}
repo := manifest.NewRepository(stub, testNamespace)

result, err := repo.ListAllForModule(ctx, testModuleName)

require.NoError(t, err)
require.Len(t, result, 2)
require.Equal(t, expectedMetadata, result)
require.True(t, stub.listCalled)
require.Equal(t, testNamespace, stub.capturedNamespace)
require.Equal(t, testModuleName, stub.capturedLabels[shared.ModuleName])
})

t.Run("returns empty list when no manifests found", func(t *testing.T) {
stub := &clientStub{partialObjectMetadata: []apimetav1.PartialObjectMetadata{}}
repo := manifest.NewRepository(stub, testNamespace)

result, err := repo.ListAllForModule(ctx, testModuleName)

require.NoError(t, err)
require.Empty(t, result)
require.True(t, stub.listCalled)
require.Equal(t, testNamespace, stub.capturedNamespace)
require.Equal(t, testModuleName, stub.capturedLabels[shared.ModuleName])
})

t.Run("returns error when list fails", func(t *testing.T) {
expectedErr := errors.New("list error")
stub := &clientStub{listErr: expectedErr}
repo := manifest.NewRepository(stub, testNamespace)

result, err := repo.ListAllForModule(ctx, testModuleName)

require.Error(t, err)
require.Nil(t, result)
require.Contains(t, err.Error(), "failed to list Manifests for module")
require.Contains(t, err.Error(), testModuleName)
require.True(t, stub.listCalled)
require.Equal(t, testNamespace, stub.capturedNamespace)
require.Equal(t, testModuleName, stub.capturedLabels[shared.ModuleName])
})
}
58 changes: 58 additions & 0 deletions internal/repository/modulereleasemeta/mrm_repo.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
package modulereleasemeta

import (
"context"
"fmt"

"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"

"github.com/kyma-project/lifecycle-manager/api/v1beta2"
)

type Repository struct {
clnt client.Client
namespace string
}

func NewRepository(clnt client.Client, namespace string) *Repository {
return &Repository{
clnt: clnt,
namespace: namespace,
}
}

func (r *Repository) EnsureFinalizer(ctx context.Context, mrmName string, finalizer string) error {
mrm, err := r.Get(ctx, mrmName)
if err != nil {
return err
}
if updated := controllerutil.AddFinalizer(mrm, finalizer); updated {
if err := r.clnt.Update(ctx, mrm); err != nil {
return fmt.Errorf("failed to add finalizer to ModuleReleaseMeta %s: %w", mrmName, err)
}
}
return nil
}

func (r *Repository) RemoveFinalizer(ctx context.Context, mrmName string, finalizer string) error {
mrm, err := r.Get(ctx, mrmName)
if err != nil {
return err
}
if updated := controllerutil.RemoveFinalizer(mrm, finalizer); updated {
if err := r.clnt.Update(ctx, mrm); err != nil {
return fmt.Errorf("failed to remove finalizer from ModuleReleaseMeta %s: %w", mrmName, err)
}
}
return nil
}

func (r *Repository) Get(ctx context.Context, mrmName string) (*v1beta2.ModuleReleaseMeta, error) {
mrm := &v1beta2.ModuleReleaseMeta{}
err := r.clnt.Get(ctx, client.ObjectKey{Name: mrmName, Namespace: r.namespace}, mrm)
if err != nil {
return nil, fmt.Errorf("failed to get ModuleReleaseMeta %s in namespace %s: %w", mrmName, r.namespace, err)
}
return mrm, nil
}
Loading
Loading