Move packages to proejct/pkg to follow go project folder structure convention (#1190)

2019-05-31 16:28:55 +08:00
parent 1c5fd92ad6
commit a0e9a39511
196 changed files with 1672 additions and 1716 deletions
@@ -0,0 +1,121 @@
+package fscache
+
+import (
+	"log"
+	"testing"
+	"time"
+
+	"go.uber.org/zap"
+	apiv1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	fv1 "github.com/fission/fission/pkg/apis/fission.io/v1"
+)
+
+func panicIf(err error) {
+	if err != nil {
+		log.Panicf("Error: %v", err)
+	}
+}
+
+func TestFunctionServiceCache(t *testing.T) {
+	logger, err := zap.NewDevelopment()
+	panicIf(err)
+
+	fsc := MakeFunctionServiceCache(logger)
+	if fsc == nil {
+		log.Panicf("error creating cache")
+	}
+
+	var fsvc *FuncSvc
+	now := time.Now()
+
+	objects := []apiv1.ObjectReference{
+		{
+			Kind:       "pod",
+			Name:       "xxx",
+			APIVersion: "v1",
+			Namespace:  "fission-function",
+		},
+		{
+			Kind:       "pod",
+			Name:       "xxx2",
+			APIVersion: "v1",
+			Namespace:  "fission-function",
+		},
+	}
+
+	fsvc = &FuncSvc{
+		Function: &metav1.ObjectMeta{
+			Name: "foo",
+			UID:  "1212",
+		},
+		Environment: &fv1.Environment{
+			Metadata: metav1.ObjectMeta{
+				Name: "foo-env",
+				UID:  "2323",
+			},
+			Spec: fv1.EnvironmentSpec{
+				Version: 1,
+				Runtime: fv1.Runtime{
+					Image: "fission/foo-env",
+				},
+				Builder: fv1.Builder{},
+			},
+		},
+		Address:           "xxx",
+		KubernetesObjects: objects,
+		Ctime:             now,
+		Atime:             now,
+	}
+	_, err = fsc.Add(*fsvc)
+	if err != nil {
+		fsc.Log()
+		log.Panicf("Failed to add fsvc: %v", err)
+	}
+
+	f, err := fsc.GetByFunction(fsvc.Function)
+	if err != nil {
+		fsc.Log()
+		log.Panicf("Failed to get fsvc: %v", err)
+	}
+	f, err = fsc.GetByFunctionUID(fsvc.Function.UID)
+	if err != nil {
+		fsc.Log()
+		log.Panicf("Failed to get fsvc by function uid: %v", err)
+	}
+	fsvc.Atime = f.Atime
+	fsvc.Ctime = f.Ctime
+	if f.Address != fsvc.Address {
+		fsc.Log()
+		log.Panicf("Incorrect fsvc \n(expected: %#v)\n (found: %#v)", fsvc, f)
+	}
+
+	err = fsc.TouchByAddress(fsvc.Address)
+	if err != nil {
+		fsc.Log()
+		log.Panicf("Failed to touch fsvc: %v", err)
+	}
+
+	deleted, err := fsc.DeleteOld(fsvc, 0)
+	if err != nil {
+		fsc.Log()
+		log.Panicf("Failed to delete fsvc: %v", err)
+	}
+	if !deleted {
+		fsc.Log()
+		log.Panicf("Did not delete fsvc")
+	}
+
+	_, err = fsc.GetByFunction(fsvc.Function)
+	if err == nil {
+		fsc.Log()
+		log.Panicf("found fsvc while expecting empty cache: %v", err)
+	}
+
+	_, err = fsc.GetByFunctionUID(fsvc.Function.UID)
+	if err == nil {
+		fsc.Log()
+		log.Panicf("found fsvc by function uid while expecting empty cache: %v", err)
+	}
+}