// Copyright (c) 2020 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License-AGPL.txt in the project root for license information.

package manager

import (
	"github.com/gitpod-io/gitpod/ws-manager/api"
	fakek8s "k8s.io/client-go/kubernetes/fake"
	"testing"
)

func TestMarkWorkspace(t *testing.T) {
	tests := []struct {
		Description    string
		InitialState   map[string]string
		Operations     []*annotation
		ExpectedResult map[string]string
		ExpectedErr    string
	}{
		{"add mark", nil, []*annotation{addMark("foobar", "val")}, map[string]string{"foobar": "val"}, ""},
		{"remove mark", map[string]string{"foobar": "val"}, []*annotation{deleteMark("foobar")}, nil, ""},
		{"add/remove mark", nil, []*annotation{addMark("foobar", "val"), deleteMark("foobar")}, nil, ""},
	}

	for _, test := range tests {
		t.Run(test.Description, func(t *testing.T) {
			manager := forTestingOnlyGetManager(t)
			manager.Config.Namespace = ""
			startCtx, err := forTestingOnlyCreateStartWorkspaceContext(manager, "foo", api.WorkspaceType_REGULAR)
			if err != nil {
				t.Errorf("cannot create test pod start context; this is a bug in the unit test itself: %v", err)
				return
			}

			pod, err := manager.createDefiniteWorkspacePod(startCtx)
			if err != nil {
				t.Errorf("cannot create test pod; this is a bug in the unit test itself: %v", err)
				return
			}
			modificationPrior := make(map[string]string)
			for k, v := range pod.Annotations {
				modificationPrior[k] = v
			}
			for k, v := range test.InitialState {
				pod.Annotations[k] = v
			}
			manager.Clientset = fakek8s.NewSimpleClientset(pod)

			err = manager.markWorkspace(startCtx.Request.Id, test.Operations...)
			if err != nil && err.Error() != test.ExpectedErr {
				t.Errorf("markWorkspace failed: %v", err)
				return
			}

			pod, _ = manager.findWorkspacePod(startCtx.Request.Id)
			modificationPosterior := make(map[string]string)
			for k, v := range pod.Annotations {
				if _, wasThereAlready := modificationPrior[k]; !wasThereAlready {
					modificationPosterior[k] = v
				}
			}

			for k, v := range test.ExpectedResult {
				postVal, exists := modificationPosterior[k]
				if !exists {
					t.Errorf("annotation %s was not found post modification", k)
					return
				}
				if postVal != v {
					t.Errorf("annotation value for %s does not match: expected %s, actual %s", k, v, postVal)
					return
				}
			}
			if len(modificationPosterior) != len(test.ExpectedResult) {
				t.Errorf("annotations do not match expectation: %v != %v", test.ExpectedResult, modificationPosterior)
			}
		})
	}
}