kubevpn/pkg/handler/remote_test.go

package handler

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"k8s.io/utils/pointer"
	"net"
	"os/exec"
	"path/filepath"
	"testing"
	"time"

	log "github.com/sirupsen/logrus"
	corev1 "k8s.io/api/core/v1"
	k8serrors "k8s.io/apimachinery/pkg/api/errors"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	k8sruntime "k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/types"
	"k8s.io/apimachinery/pkg/util/wait"
	"k8s.io/cli-runtime/pkg/genericclioptions"
	pkgresource "k8s.io/cli-runtime/pkg/resource"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/tools/clientcmd"
	"k8s.io/client-go/util/homedir"
	"k8s.io/client-go/util/retry"
	cmdutil "k8s.io/kubectl/pkg/cmd/util"

	"github.com/wencaiwulue/kubevpn/pkg/config"
	"github.com/wencaiwulue/kubevpn/pkg/util"
)

//func TestCreateServer(t *testing.T) {
//	clientConfig := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
//		&clientcmd.ClientConfigLoadingRules{ExplicitPath: clientcmd.RecommendedHomeFile}, nil,
//	)
//	config, err := clientConfig.ClientConfig()
//	if err != nil {
//		log.Fatal(err)
//	}
//	clientset, err := kubernetes.NewForConfig(config)
//	if err != nil {
//		log.Fatal(err)
//	}
//
//	i := &net.IPNet{
//		IP:   net.ParseIP("192.168.254.100"),
//		Mask: net.IPv4Mask(255, 255, 255, 0),
//	}
//
//	j := &net.IPNet{
//		IP:   net.ParseIP("172.20.0.0"),
//		Mask: net.IPv4Mask(255, 255, 0, 0),
//	}
//
//	server, err := pkg.CreateOutboundPod(clientset, "test", i, []*net.IPNet{j})
//	fmt.Println(server)
//}

func TestGetIp(t *testing.T) {
	ip := &net.IPNet{
		IP:   net.IPv4(192, 168, 254, 100),
		Mask: net.IPv4Mask(255, 255, 255, 0),
	}
	fmt.Println(ip.String())
}

func TestGetIPFromDHCP(t *testing.T) {
	clientConfig := clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
		&clientcmd.ClientConfigLoadingRules{ExplicitPath: clientcmd.RecommendedHomeFile}, nil,
	)
	config, err := clientConfig.ClientConfig()
	if err != nil {
		log.Fatal(err)
	}
	clientset, err := kubernetes.NewForConfig(config)
	if err != nil {
		log.Fatal(err)
	}
	_, ipNet, err := net.ParseCIDR("192.168.1.100/24")
	manager := NewDHCPManager(clientset.CoreV1().ConfigMaps("test"), "test", ipNet)
	manager.InitDHCP()
	for i := 0; i < 10; i++ {
		ipNet, err := manager.RentIPRandom()
		ipNet2, err := manager.RentIPRandom()
		if err != nil {
			fmt.Println(err)
			continue
		} else {
			fmt.Printf("%s->%s\n", ipNet.String(), ipNet2.String())
		}
		time.Sleep(time.Millisecond * 10)
		err = manager.ReleaseIpToDHCP(ipNet)
		err = manager.ReleaseIpToDHCP(ipNet2)
		if err != nil {
			fmt.Println(err)
		}
		time.Sleep(time.Millisecond * 10)
	}
}

func TestGetTopController(t *testing.T) {
	s := "/Users/naison/.kube/devpool"
	configFlags := genericclioptions.NewConfigFlags(true).WithDeprecatedPasswordFlag()
	configFlags.KubeConfig = &s
	factory := cmdutil.NewFactory(cmdutil.NewMatchVersionFlags(configFlags))
	controller, err := util.GetTopOwnerReference(factory, "nh90bwck", "pods/services-authors-shadow")
	fmt.Println(controller.Name)
	fmt.Println(controller.Mapping.Resource.Resource)
	fmt.Println(err)
}

func TestGetTopControllerByLabel(t *testing.T) {
	s := "/Users/naison/.kube/mesh"
	configFlags := genericclioptions.NewConfigFlags(true).WithDeprecatedPasswordFlag()
	configFlags.KubeConfig = &s
	factory := cmdutil.NewFactory(cmdutil.NewMatchVersionFlags(configFlags))
	controller, err := util.GetTopOwnerReferenceBySelector(factory, "default", "app=productpage")
	fmt.Println(controller)
	fmt.Println(err)
}

func TestPreCheck(t *testing.T) {
	options := ConnectOptions{
		KubeconfigPath: filepath.Join(homedir.HomeDir(), ".kube", "mesh"),
		Namespace:      "naison-test",
		Workloads:      []string{"services/authors"},
	}
	options.InitClient()
	options.PreCheckResource()
	fmt.Println(options.Workloads)
}

func init() {
	util.InitLogger(config.Debug)
}

func TestBackoff(t *testing.T) {
	var last = time.Now()
	retry.OnError(wait.Backoff{
		Steps:    10,
		Duration: 40 * time.Millisecond,
		Factor:   2.0,
		Jitter:   0.5,
	}, func(err error) bool {
		return true
	}, func() error {
		now := time.Now()
		fmt.Printf("%vs\n", now.Sub(last).Seconds())
		last = now
		return errors.New("")
	})
}

func TestGetCRD(t *testing.T) {
	join := filepath.Join(homedir.HomeDir(), ".kube", "nocalhost.large")
	configFlags := genericclioptions.NewConfigFlags(true).WithDeprecatedPasswordFlag()
	configFlags.KubeConfig = &join
	factory := cmdutil.NewFactory(cmdutil.NewMatchVersionFlags(configFlags))
	Namespace, _, _ := factory.ToRawKubeConfigLoader().Namespace()
	object, err := util.GetUnstructuredObject(factory, Namespace, "statefulsets.apps.kruise.io/sample-beta1")
	fmt.Println(object)
	fmt.Println(err)
}

func TestDeleteAndCreate(t *testing.T) {
	file := clientcmd.RecommendedHomeFile
	file = filepath.Join(homedir.HomeDir(), ".kube", "config")
	configFlags := genericclioptions.NewConfigFlags(true).WithDeprecatedPasswordFlag()
	configFlags.KubeConfig = &file
	factory := cmdutil.NewFactory(cmdutil.NewMatchVersionFlags(configFlags))

	Namespace, _, err := factory.ToRawKubeConfigLoader().Namespace()
	object, err := util.GetUnstructuredObject(factory, Namespace, "pods/nginx")

	u := object.Object.(*unstructured.Unstructured)
	var pp corev1.Pod
	marshal, err := json.Marshal(u)
	err = json.Unmarshal(marshal, &pp)

	helper := pkgresource.NewHelper(object.Client, object.Mapping)
	if _, err = helper.DeleteWithOptions(object.Namespace, object.Name, &metav1.DeleteOptions{
		GracePeriodSeconds: pointer.Int64(0),
	}); err != nil {
		log.Fatal(err)
	}
	_ = exec.Command("kubectl", "wait", "pods/nginx", "--for=delete").Run()

	p := &corev1.Pod{ObjectMeta: pp.ObjectMeta, Spec: pp.Spec}
	CleanupUselessInfo(p)
	if err = retry.OnError(wait.Backoff{
		Steps:    10,
		Duration: 50 * time.Millisecond,
		Factor:   5.0,
		Jitter:   1,
	}, func(err error) bool {
		if !k8serrors.IsAlreadyExists(err) {
			return true
		}
		clientset, err := factory.KubernetesClientSet()
		get, err := clientset.CoreV1().Pods(p.Namespace).Get(context.Background(), p.Name, metav1.GetOptions{})
		if err != nil || get.Status.Phase != corev1.PodRunning {
			return true
		}
		return false
	}, func() error {
		_, err = helper.Create(object.Namespace, true, p)
		if err != nil {
			return err
		}
		return errors.New("")
	}); !k8serrors.IsAlreadyExists(err) {
		log.Fatal(err)
	}
}

func TestReadiness(t *testing.T) {
	configFlags := genericclioptions.NewConfigFlags(true).WithDeprecatedPasswordFlag()
	configFlags.KubeConfig = &clientcmd.RecommendedHomeFile
	factory := cmdutil.NewFactory(cmdutil.NewMatchVersionFlags(configFlags))
	object, err := util.GetUnstructuredObject(factory, "default", "deployment/authors")
	if err != nil {
		panic(err)
	}
	podTemplateSpec, path, err := util.GetPodTemplateSpecPath(object.Object.(*unstructured.Unstructured))
	if err != nil {
		panic(err)
	}
	helper := pkgresource.NewHelper(object.Client, object.Mapping)
	removePatch, restorePatch := patch(*podTemplateSpec, path)
	marshal, _ := json.Marshal(removePatch)
	bytes, _ := json.Marshal(restorePatch)
	_, err = patchs(helper, object.Namespace, object.Name, marshal)
	if err != nil {
		panic(err)
	}
	_, err = patchs(helper, object.Namespace, object.Name, bytes)
	if err != nil {
		panic(err)
	}
}

func patchs(helper *pkgresource.Helper, namespace, name string, p []byte) (k8sruntime.Object, error) {
	return helper.Patch(
		namespace,
		name,
		types.JSONPatchType,
		p,
		&metav1.PatchOptions{},
	)
}

func TestSliceRemove(t *testing.T) {
	a := []string{"a", "a", "b", "c"}
	for i := 0; i < len(a); i++ {
		if a[i] == "a" {
			a = append(a[:i], a[i+1:]...)
			//i--
		}
	}
	fmt.Println(a)
}