Vardhaman SuranaandGitHub
3fabf64b3c
* used manager's Add function in more places * exit when ctx.Done is received in archivePruner go routines * fix manager tests * fix data race * added more gpm function in manager and removed manager from a util function * closed unused channel and stopped ticker after context is done * added log statements * used context.Done inside function instead of stopper channel
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"regexp"
|
|
"testing"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
|
|
"github.com/fission/fission/pkg/utils/loggerfactory"
|
|
"github.com/fission/fission/pkg/utils/manager"
|
|
)
|
|
|
|
func TestServiceAccountCheck(t *testing.T) {
|
|
mgr := manager.New()
|
|
defer mgr.Wait()
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
kubernetesClient := fake.NewSimpleClientset()
|
|
logger := loggerfactory.GetLogger()
|
|
os.Setenv(ENV_CREATE_SA, "true")
|
|
CreateMissingPermissionForSA(ctx, kubernetesClient, logger)
|
|
|
|
// Get rolebinding for a service account
|
|
rolebindings, err := kubernetesClient.RbacV1().RoleBindings("default").List(ctx, metav1.ListOptions{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(rolebindings.Items) != 2 {
|
|
t.Fatal("Rolebinding not created", len(rolebindings.Items))
|
|
}
|
|
regexp := regexp.MustCompile(`fission\-(fetcher|builder)\-rolebinding\-[a-z0-9]{6}`)
|
|
for _, rolebinding := range rolebindings.Items {
|
|
if !regexp.Match([]byte(rolebinding.Name)) {
|
|
t.Fatal("Rolebinding not created for fission-builder or fission-fetcher", rolebinding.Name)
|
|
}
|
|
}
|
|
}
|