add tests

Signed-off-by: Matthias Bertschy <[email protected]>
kubescape · Nov 20, 2023 · 99a4cbc · 99a4cbc
1 parent 95df542
commit 99a4cbc
Show file tree

Hide file tree

Showing 4 changed files with 169 additions and 17 deletions.
diff --git a/adapters/mock.go b/adapters/mock.go
@@ -13,27 +13,28 @@ import (
 
 type MockAdapter struct {
 	callbacks     domain.Callbacks
-	patchStrategy bool
-	resources     map[string][]byte
+	patchStrategy bool // true for client, false for server
+	Resources     map[string][]byte
 	shadowObjects map[string][]byte
 }
 
-func NewMockAdapter() *MockAdapter {
+func NewMockAdapter(patchStrategy bool) *MockAdapter {
 	return &MockAdapter{
-		resources:     map[string][]byte{},
+		patchStrategy: patchStrategy,
+		Resources:     map[string][]byte{},
 		shadowObjects: map[string][]byte{},
 	}
 }
 
 var _ Adapter = (*MockAdapter)(nil)
 
 func (m *MockAdapter) DeleteObject(_ context.Context, id domain.KindName) error {
-	delete(m.resources, id.String())
+	delete(m.Resources, id.String())
 	return nil
 }
 
 func (m *MockAdapter) GetObject(ctx context.Context, id domain.KindName, baseObject []byte) error {
-	object, ok := m.resources[id.String()]
+	object, ok := m.Resources[id.String()]
 	if !ok {
 		return fmt.Errorf("object not found")
 	}
@@ -72,7 +73,7 @@ func (m *MockAdapter) PatchObject(ctx context.Context, id domain.KindName, check
 }
 
 func (m *MockAdapter) patchObject(id domain.KindName, checksum string, patch []byte) ([]byte, error) {
-	object, ok := m.resources[id.String()]
+	object, ok := m.Resources[id.String()]
 	if !ok {
 		return nil, fmt.Errorf("object not found")
 	}
@@ -87,13 +88,13 @@ func (m *MockAdapter) patchObject(id domain.KindName, checksum string, patch []b
 	if newChecksum != checksum {
 		return object, fmt.Errorf("checksum mismatch: %s != %s", newChecksum, checksum)
 	}
-	m.resources[id.String()] = modified
+	m.Resources[id.String()] = modified
 	m.shadowObjects[id.String()] = modified
 	return object, nil
 }
 
 func (m *MockAdapter) PutObject(_ context.Context, id domain.KindName, object []byte) error {
-	m.resources[id.String()] = object
+	m.Resources[id.String()] = object
 	return nil
 }
 
@@ -119,7 +120,7 @@ func (m *MockAdapter) VerifyObject(ctx context.Context, id domain.KindName, newC
 }
 
 func (m *MockAdapter) verifyObject(id domain.KindName, newChecksum string) ([]byte, error) {
-	object, ok := m.resources[id.String()]
+	object, ok := m.Resources[id.String()]
 	if !ok {
 		return nil, fmt.Errorf("object not found")
 	}
@@ -132,3 +133,33 @@ func (m *MockAdapter) verifyObject(id domain.KindName, newChecksum string) ([]by
 	}
 	return object, nil
 }
+
+func (m *MockAdapter) TestCallDeleteObject(ctx context.Context, id domain.KindName) error {
+	ctx = utils.ContextFromGeneric(ctx, domain.Generic{})
+	err := m.callbacks.DeleteObject(ctx, id)
+	if err != nil {
+		return fmt.Errorf("send delete: %w", err)
+	}
+	if m.patchStrategy {
+		// remove from known resources
+		delete(m.shadowObjects, id.String())
+	}
+	return nil
+}
+
+// TestCallVerifyObject is used for testing purposes only, it is similar to incluster.client.callVerifyObject
+func (m *MockAdapter) TestCallVerifyObject(ctx context.Context, id domain.KindName, object []byte) error {
+	// store object
+	m.Resources[id.String()] = object
+	// calculate checksum
+	checksum, err := utils.CanonicalHash(object)
+	if err != nil {
+		return fmt.Errorf("calculate checksum: %w", err)
+	}
+	ctx = utils.ContextFromGeneric(ctx, domain.Generic{})
+	err = m.callbacks.VerifyObject(ctx, id, checksum)
+	if err != nil {
+		return fmt.Errorf("send checksum: %w", err)
+	}
+	return nil
+}
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -60,7 +60,7 @@ func main() {
 	} else {
 		// mock adapter
 		logger.L().Info("initializing mock adapter")
-		adapter = adapters.NewMockAdapter()
+		adapter = adapters.NewMockAdapter(false)
 	}
 
 	// websocket server

diff --git a/core/synchronizer_server_test.go b/core/synchronizer_server_test.go
@@ -0,0 +1,50 @@
+package core
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSynchronizer_ObjectAddedOnServer(t *testing.T) {
+	ctx, clientAdapter, serverAdapter := initTest(t)
+	// add object
+	err := serverAdapter.TestCallVerifyObject(ctx, kindKnownServers, object)
+	assert.NoError(t, err)
+	time.Sleep(1 * time.Second)
+	// check object added
+	serverObj, ok := clientAdapter.Resources[kindKnownServers.String()]
+	assert.True(t, ok)
+	assert.Equal(t, object, serverObj)
+}
+
+func TestSynchronizer_ObjectDeletedOnServer(t *testing.T) {
+	ctx, clientAdapter, serverAdapter := initTest(t)
+	// pre: add object
+	clientAdapter.Resources[kindKnownServers.String()] = object
+	serverAdapter.Resources[kindKnownServers.String()] = object
+	// delete object
+	err := serverAdapter.TestCallDeleteObject(ctx, kindKnownServers)
+	assert.NoError(t, err)
+	time.Sleep(1 * time.Second)
+	// check object deleted
+	_, ok := clientAdapter.Resources[kindKnownServers.String()]
+	assert.False(t, ok)
+}
+
+func TestSynchronizer_ObjectModifiedOnServer(t *testing.T) {
+	ctx, clientAdapter, serverAdapter := initTest(t)
+	// pre: add object
+	clientAdapter.Resources[kindKnownServers.String()] = object
+	serverAdapter.Resources[kindKnownServers.String()] = object
+	// modify object
+	objectV2 := []byte(`{"kind":"kind","metadata":{"name":"name","version":"2"}}`)
+	err := serverAdapter.TestCallVerifyObject(ctx, kindKnownServers, objectV2)
+	assert.NoError(t, err)
+	time.Sleep(1 * time.Second)
+	// check object modified
+	serverObj, ok := clientAdapter.Resources[kindKnownServers.String()]
+	assert.True(t, ok)
+	assert.Equal(t, objectV2, serverObj)
+}
diff --git a/core/synchronizer_test.go b/core/synchronizer_test.go
@@ -2,19 +2,90 @@ package core
 
 import (
 	"context"
+	"net"
 	"testing"
+	"time"
 
+	"github.com/kubescape/go-logger"
+	"github.com/kubescape/go-logger/helpers"
 	"github.com/kubescape/synchronizer/adapters"
 	"github.com/kubescape/synchronizer/domain"
 	"github.com/stretchr/testify/assert"
 )
 
-func TestSynchronizer_HandleSyncGetObject(t *testing.T) {
-	synchronizer := &Synchronizer{
-		adapter: adapters.NewMockAdapter(),
-		outPool: nil,
+var (
+	kindDeployment = domain.KindName{
+		Kind:      domain.KindFromString("apps/v1/Deployment"),
+		Name:      "name",
+		Namespace: "namespace",
 	}
+	kindKnownServers = domain.KindName{
+		Kind:      domain.KindFromString("spdx.softwarecomposition.kubescape.io/v1beta1/KnownServers"),
+		Name:      "name",
+		Namespace: "namespace",
+	}
+	object = []byte(`{"kind":"kind","metadata":{"name":"name"}}`)
+)
+
+func initTest(t *testing.T) (context.Context, *adapters.MockAdapter, *adapters.MockAdapter) {
+	ctx := context.WithValue(context.TODO(), domain.ContextKeyClientIdentifier, domain.ClientIdentifier{
+		Account: "11111111-2222-3333-4444-555555555555",
+		Cluster: "cluster",
+	})
+	err := logger.L().SetLevel(helpers.DebugLevel.String())
+	assert.NoError(t, err)
+	clientAdapter := adapters.NewMockAdapter(true)
+	serverAdapter := adapters.NewMockAdapter(false)
+	clientConn, serverConn := net.Pipe()
+	client := NewSynchronizerClient(ctx, clientAdapter, clientConn)
+	server := NewSynchronizerServer(ctx, serverAdapter, serverConn)
+	go func() {
+		_ = client.Start(ctx)
+	}()
+	go func() {
+		_ = server.Start(ctx)
+	}()
+	return ctx, clientAdapter, serverAdapter
+}
+
+func TestSynchronizer_ObjectAdded(t *testing.T) {
+	ctx, clientAdapter, serverAdapter := initTest(t)
+	// add object
+	err := clientAdapter.TestCallVerifyObject(ctx, kindDeployment, object)
+	assert.NoError(t, err)
+	time.Sleep(1 * time.Second)
+	// check object added
+	serverObj, ok := serverAdapter.Resources[kindDeployment.String()]
+	assert.True(t, ok)
+	assert.Equal(t, object, serverObj)
+}
+
+func TestSynchronizer_ObjectDeleted(t *testing.T) {
+	ctx, clientAdapter, serverAdapter := initTest(t)
+	// pre: add object
+	clientAdapter.Resources[kindDeployment.String()] = object
+	serverAdapter.Resources[kindDeployment.String()] = object
+	// delete object
+	err := clientAdapter.TestCallDeleteObject(ctx, kindDeployment)
+	assert.NoError(t, err)
+	time.Sleep(1 * time.Second)
+	// check object deleted
+	_, ok := serverAdapter.Resources[kindDeployment.String()]
+	assert.False(t, ok)
+}
 
-	err := synchronizer.handleSyncGetObject(context.TODO(), domain.KindName{}, []byte("baseObject"))
-	assert.ErrorContains(t, err, "object not found")
+func TestSynchronizer_ObjectModified(t *testing.T) {
+	ctx, clientAdapter, serverAdapter := initTest(t)
+	// pre: add object
+	clientAdapter.Resources[kindDeployment.String()] = object
+	serverAdapter.Resources[kindDeployment.String()] = object
+	// modify object
+	objectV2 := []byte(`{"kind":"kind","metadata":{"name":"name","version":"2"}}`)
+	err := clientAdapter.TestCallVerifyObject(ctx, kindDeployment, objectV2)
+	assert.NoError(t, err)
+	time.Sleep(1 * time.Second)
+	// check object modified
+	serverObj, ok := serverAdapter.Resources[kindDeployment.String()]
+	assert.True(t, ok)
+	assert.Equal(t, objectV2, serverObj)
 }