Implement application manager unit tests (#4)

Author: alexmt

Gopkg.lock

@@ -30,3 +30,7 @@ required = [
 [[constraint]]
   branch = "release-6.0"
   name = "k8s.io/client-go"
+
+[[constraint]]
+  name = "github.com/stretchr/testify"
+  version = "1.2.1"

Gopkg.toml

@@ -0,0 +1,32 @@
+package application
+
+import (
+	"time"
+
+	"github.com/argoproj/argo-cd/pkg/apis/application/v1alpha1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+// AppComparator defines methods which allow to compare application spec and actual application state.
+type AppComparator interface {
+	CompareAppState(appRepoPath string, app *v1alpha1.Application) (*v1alpha1.ComparisonResult, error)
+}
+
+// KsonnetAppComparator allows to compare application using KSonnet CLI
+type KsonnetAppComparator struct {
+}
+
+// CompareAppState compares application spec and real app state using KSonnet
+func (ks *KsonnetAppComparator) CompareAppState(appRepoPath string, app *v1alpha1.Application) (*v1alpha1.ComparisonResult, error) {
+	// TODO (amatyushentsev): Implement actual comparison
+	return &v1alpha1.ComparisonResult{
+		Status:     v1alpha1.ComparisonStatusEqual,
+		ComparedTo: app.Spec.Source,
+		ComparedAt: metav1.Time{Time: time.Now().UTC()},
+	}, nil
+}
+
+// NewKsonnetAppComparator creates new instance of Ksonnet app comparator
+func NewKsonnetAppComparator() AppComparator {
+	return &KsonnetAppComparator{}
+}

application/comparator.go

@@ -21,6 +21,7 @@ type Manager struct {
 	gitClient            git.Client
 	repoService          repository.RepositoryServiceServer
 	statusRefreshTimeout time.Duration
+	appComparator        AppComparator
 }
 
 // NeedRefreshAppStatus answers if application status needs to be refreshed. Returns true if application never been compared, has changed or comparison result has expired.
@@ -66,7 +67,7 @@ func (m *Manager) tryRefreshAppStatus(app *v1alpha1.Application) (*v1alpha1.Appl
 	if err != nil {
 		return nil, err
 	}
-	comparisonResult, err := m.compareAppState(appRepoPath, app)
+	comparisonResult, err := m.appComparator.CompareAppState(appRepoPath, app)
 	if err != nil {
 		return nil, err
 	}
@@ -75,20 +76,16 @@ func (m *Manager) tryRefreshAppStatus(app *v1alpha1.Application) (*v1alpha1.Appl
 	}, nil
 }
 
-func (m *Manager) compareAppState(appRepoPath string, app *v1alpha1.Application) (*v1alpha1.ComparisonResult, error) {
-	// TODO (amatyushentsev): Implement actual comparison
-	return &v1alpha1.ComparisonResult{
-		Status:     v1alpha1.ComparisonStatusEqual,
-		ComparedTo: app.Spec.Source,
-		ComparedAt: metav1.Time{Time: time.Now().UTC()},
-	}, nil
-}
-
-// NewAppManager creates new instance of app.Manager
-func NewAppManager(gitClient git.Client, repoService repository.RepositoryServiceServer, statusRefreshTimeout time.Duration) *Manager {
+// NewAppManager creates new instance of app manager.
+func NewAppManager(
+	gitClient git.Client,
+	repoService repository.RepositoryServiceServer,
+	appComparator AppComparator,
+	statusRefreshTimeout time.Duration) *Manager {
 	return &Manager{
 		gitClient:            gitClient,
 		repoService:          repoService,
 		statusRefreshTimeout: statusRefreshTimeout,
+		appComparator:        appComparator,
 	}
 }

application/manager.go

@@ -0,0 +1,73 @@
+package application_test
+
+import (
+	"testing"
+
+	"time"
+
+	"github.com/argoproj/argo-cd/application"
+	"github.com/argoproj/argo-cd/pkg/apis/application/v1alpha1"
+	"github.com/stretchr/testify/assert"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+func TestManager(t *testing.T) {
+
+	refreshTimeout := time.Second * 10
+	appSource := v1alpha1.ApplicationSource{
+		Environment:    "prod/us-west-2",
+		Path:           "apps/elk",
+		TargetRevision: "master",
+		RepoURL:        "http://my-git-repo.git",
+	}
+
+	t.Run("NeedRefreshAppStatus", func(t *testing.T) {
+
+		manager := application.NewAppManager(nil, nil, nil, refreshTimeout)
+		t.Run("TestReturnsTrueIfAppWasNotCompared", func(t *testing.T) {
+			needRefresh := manager.NeedRefreshAppStatus(&v1alpha1.Application{
+				Spec: v1alpha1.ApplicationSpec{Source: appSource},
+				Status: v1alpha1.ApplicationStatus{
+					ComparisonResult: v1alpha1.ComparisonResult{Status: v1alpha1.ComparisonStatusUnknown},
+				},
+			})
+			assert.True(t, needRefresh)
+		})
+
+		t.Run("TestReturnsFalseIfAppWasComparedBeforeRefreshTimeoutExpires", func(t *testing.T) {
+			needRefresh := manager.NeedRefreshAppStatus(&v1alpha1.Application{
+				Spec: v1alpha1.ApplicationSpec{Source: appSource},
+				Status: v1alpha1.ApplicationStatus{
+					ComparisonResult: v1alpha1.ComparisonResult{Status: v1alpha1.ComparisonStatusEqual, ComparedAt: metav1.Time{Time: time.Now()}, ComparedTo: appSource},
+				},
+			})
+			assert.False(t, needRefresh)
+		})
+
+		t.Run("TestReturnsTrueIfAppWasComparedAfterRefreshTimeoutExpires", func(t *testing.T) {
+			needRefresh := manager.NeedRefreshAppStatus(&v1alpha1.Application{
+				Spec: v1alpha1.ApplicationSpec{Source: appSource},
+				Status: v1alpha1.ApplicationStatus{
+					ComparisonResult: v1alpha1.ComparisonResult{
+						Status:     v1alpha1.ComparisonStatusEqual,
+						ComparedAt: metav1.Time{Time: time.Now().Add(-(refreshTimeout + time.Second))},
+						ComparedTo: appSource,
+					},
+				},
+			})
+			assert.True(t, needRefresh)
+		})
+
+		t.Run("TestReturnsTrueApplicationSourceHasChanged", func(t *testing.T) {
+			updatedSource := *appSource.DeepCopy()
+			updatedSource.TargetRevision = "abc"
+			needRefresh := manager.NeedRefreshAppStatus(&v1alpha1.Application{
+				Spec: v1alpha1.ApplicationSpec{Source: appSource},
+				Status: v1alpha1.ApplicationStatus{
+					ComparisonResult: v1alpha1.ComparisonResult{Status: v1alpha1.ComparisonStatusEqual, ComparedAt: metav1.Time{Time: time.Now()}, ComparedTo: updatedSource},
+				},
+			})
+			assert.True(t, needRefresh)
+		})
+	})
+}

application/manager_test.go

@@ -44,7 +44,11 @@ func newCommand() *cobra.Command {
 			namespace := "default"
 			appResyncPeriod := time.Minute * 10
 
-			appManager := application.NewAppManager(nativeGitClient, repository.NewServer(namespace, kubeClient, appClient), appResyncPeriod)
+			appManager := application.NewAppManager(
+				nativeGitClient,
+				repository.NewServer(namespace, kubeClient, appClient),
+				application.NewKsonnetAppComparator(),
+				appResyncPeriod)
 
 			appController := controller.NewApplicationController(kubeClient, appClient, appManager, appResyncPeriod)