| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package kubernetes |
| |
| import ( |
| "encoding/json" |
| "os" |
| "strconv" |
| "testing" |
| "time" |
| ) |
| |
| import ( |
| "github.com/stretchr/testify/assert" |
| "github.com/stretchr/testify/suite" |
| "k8s.io/api/core/v1" |
| "k8s.io/client-go/kubernetes" |
| "k8s.io/client-go/kubernetes/fake" |
| ) |
| |
| import ( |
| "github.com/apache/dubbo-go/common" |
| "github.com/apache/dubbo-go/common/constant" |
| "github.com/apache/dubbo-go/config_center" |
| "github.com/apache/dubbo-go/remoting" |
| ) |
| |
| var clientPodJsonData = `{ |
| "apiVersion": "v1", |
| "kind": "Pod", |
| "metadata": { |
| "annotations": { |
| "dubbo.io/annotation": "W3siayI6Ii9kdWJibyIsInYiOiIifSx7ImsiOiIvZHViYm8vY29tLmlrdXJlbnRvLnVzZXIuVXNlclByb3ZpZGVyIiwidiI6IiJ9LHsiayI6Ii9kdWJiby9jb20uaWt1cmVudG8udXNlci5Vc2VyUHJvdmlkZXIvY29uc3VtZXJzIiwidiI6IiJ9LHsiayI6Ii9kdWJibyIsInYiOiIifSx7ImsiOiIvZHViYm8vY29tLmlrdXJlbnRvLnVzZXIuVXNlclByb3ZpZGVyIiwidiI6IiJ9LHsiayI6Ii9kdWJiby9jb20uaWt1cmVudG8udXNlci5Vc2VyUHJvdmlkZXIvcHJvdmlkZXJzIiwidiI6IiJ9LHsiayI6Ii9kdWJiby9jb20uaWt1cmVudG8udXNlci5Vc2VyUHJvdmlkZXIvY29uc3VtZXJzL2NvbnN1bWVyJTNBJTJGJTJGMTcyLjE3LjAuOCUyRlVzZXJQcm92aWRlciUzRmNhdGVnb3J5JTNEY29uc3VtZXJzJTI2ZHViYm8lM0RkdWJib2dvLWNvbnN1bWVyLTIuNi4wJTI2cHJvdG9jb2wlM0RkdWJibyIsInYiOiIifV0=" |
| }, |
| "creationTimestamp": "2020-03-13T03:38:57Z", |
| "labels": { |
| "dubbo.io/label": "dubbo.io-value" |
| }, |
| "name": "client", |
| "namespace": "default", |
| "resourceVersion": "2449700", |
| "selfLink": "/api/v1/namespaces/default/pods/client", |
| "uid": "3ec394f5-dcc6-49c3-8061-57b4b2b41344" |
| }, |
| "spec": { |
| "containers": [ |
| { |
| "env": [ |
| { |
| "name": "NAMESPACE", |
| "valueFrom": { |
| "fieldRef": { |
| "apiVersion": "v1", |
| "fieldPath": "metadata.namespace" |
| } |
| } |
| } |
| ], |
| "image": "registry.cn-hangzhou.aliyuncs.com/scottwang/dubbogo-client", |
| "imagePullPolicy": "Always", |
| "name": "client", |
| "resources": {}, |
| "terminationMessagePath": "/dev/termination-log", |
| "terminationMessagePolicy": "File", |
| "volumeMounts": [ |
| { |
| "mountPath": "/var/run/secrets/kubernetes.io/serviceaccount", |
| "name": "dubbo-sa-token-l2lzh", |
| "readOnly": true |
| } |
| ] |
| } |
| ], |
| "dnsPolicy": "ClusterFirst", |
| "enableServiceLinks": true, |
| "nodeName": "minikube", |
| "priority": 0, |
| "restartPolicy": "Never", |
| "schedulerName": "default-scheduler", |
| "securityContext": {}, |
| "serviceAccount": "dubbo-sa", |
| "serviceAccountName": "dubbo-sa", |
| "terminationGracePeriodSeconds": 30, |
| "tolerations": [ |
| { |
| "effect": "NoExecute", |
| "key": "node.kubernetes.io/not-ready", |
| "operator": "Exists", |
| "tolerationSeconds": 300 |
| }, |
| { |
| "effect": "NoExecute", |
| "key": "node.kubernetes.io/unreachable", |
| "operator": "Exists", |
| "tolerationSeconds": 300 |
| } |
| ], |
| "volumes": [ |
| { |
| "name": "dubbo-sa-token-l2lzh", |
| "secret": { |
| "defaultMode": 420, |
| "secretName": "dubbo-sa-token-l2lzh" |
| } |
| } |
| ] |
| }, |
| "status": { |
| "conditions": [ |
| { |
| "lastProbeTime": null, |
| "lastTransitionTime": "2020-03-13T03:38:57Z", |
| "status": "True", |
| "type": "Initialized" |
| }, |
| { |
| "lastProbeTime": null, |
| "lastTransitionTime": "2020-03-13T03:40:18Z", |
| "status": "True", |
| "type": "Ready" |
| }, |
| { |
| "lastProbeTime": null, |
| "lastTransitionTime": "2020-03-13T03:40:18Z", |
| "status": "True", |
| "type": "ContainersReady" |
| }, |
| { |
| "lastProbeTime": null, |
| "lastTransitionTime": "2020-03-13T03:38:57Z", |
| "status": "True", |
| "type": "PodScheduled" |
| } |
| ], |
| "containerStatuses": [ |
| { |
| "containerID": "docker://2870d6abc19ca7fe22ca635ebcfac5d48c6d5550a659bafd74fb48104f6dfe3c", |
| "image": "registry.cn-hangzhou.aliyuncs.com/scottwang/dubbogo-client:latest", |
| "imageID": "docker-pullable://registry.cn-hangzhou.aliyuncs.com/scottwang/dubbogo-client@sha256:1f075131f708a0d400339e81549d7c4d4ed917ab0b6bd38ef458dd06ad25a559", |
| "lastState": {}, |
| "name": "client", |
| "ready": true, |
| "restartCount": 0, |
| "state": { |
| "running": { |
| "startedAt": "2020-03-13T03:40:17Z" |
| } |
| } |
| } |
| ], |
| "hostIP": "10.0.2.15", |
| "phase": "Running", |
| "podIP": "172.17.0.8", |
| "qosClass": "BestEffort", |
| "startTime": "2020-03-13T03:38:57Z" |
| } |
| } |
| ` |
| |
| func Test_DataChange(t *testing.T) { |
| listener := NewRegistryDataListener(&MockDataListener{}) |
| url, _ := common.NewURL("jsonrpc%3A%2F%2F127.0.0.1%3A20001%2Fcom.ikurento.user.UserProvider%3Fanyhost%3Dtrue%26app.version%3D0.0.1%26application%3DBDTService%26category%3Dproviders%26cluster%3Dfailover%26dubbo%3Ddubbo-provider-golang-2.6.0%26environment%3Ddev%26group%3D%26interface%3Dcom.ikurento.user.UserProvider%26ip%3D10.32.20.124%26loadbalance%3Drandom%26methods.GetUser.loadbalance%3Drandom%26methods.GetUser.retries%3D1%26methods.GetUser.weight%3D0%26module%3Ddubbogo%2Buser-info%2Bserver%26name%3DBDTService%26organization%3Dikurento.com%26owner%3DZX%26pid%3D74500%26retries%3D0%26service.filter%3Decho%26side%3Dprovider%26timestamp%3D1560155407%26version%3D%26warmup%3D100") |
| listener.AddInterestedURL(&url) |
| int := listener.DataChange(remoting.Event{Path: "/dubbo/com.ikurento.user.UserProvider/providers/jsonrpc%3A%2F%2F127.0.0.1%3A20001%2Fcom.ikurento.user.UserProvider%3Fanyhost%3Dtrue%26app.version%3D0.0.1%26application%3DBDTService%26category%3Dproviders%26cluster%3Dfailover%26dubbo%3Ddubbo-provider-golang-2.6.0%26environment%3Ddev%26group%3D%26interface%3Dcom.ikurento.user.UserProvider%26ip%3D10.32.20.124%26loadbalance%3Drandom%26methods.GetUser.loadbalance%3Drandom%26methods.GetUser.retries%3D1%26methods.GetUser.weight%3D0%26module%3Ddubbogo%2Buser-info%2Bserver%26name%3DBDTService%26organization%3Dikurento.com%26owner%3DZX%26pid%3D74500%26retries%3D0%26service.filter%3Decho%26side%3Dprovider%26timestamp%3D1560155407%26version%3D%26warmup%3D100"}) |
| assert.Equal(t, true, int) |
| } |
| |
| type MockDataListener struct{} |
| |
| func (*MockDataListener) Process(configType *config_center.ConfigChangeEvent) {} |
| |
| type KubernetesRegistryTestSuite struct { |
| suite.Suite |
| |
| currentPod v1.Pod |
| } |
| |
| func (s *KubernetesRegistryTestSuite) initRegistry() *kubernetesRegistry { |
| |
| t := s.T() |
| |
| regurl, err := common.NewURL("registry://127.0.0.1:443", common.WithParamsValue(constant.ROLE_KEY, strconv.Itoa(common.PROVIDER))) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| mock, err := newMockKubernetesRegistry(®url, s.currentPod.GetNamespace(), func() (kubernetes.Interface, error) { |
| |
| out := fake.NewSimpleClientset() |
| |
| // mock current pod |
| if _, err := out.CoreV1().Pods(s.currentPod.GetNamespace()).Create(&s.currentPod); err != nil { |
| t.Fatal(err) |
| } |
| return out, nil |
| }) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| time.Sleep(time.Second) |
| return mock.(*kubernetesRegistry) |
| } |
| |
| func (s *KubernetesRegistryTestSuite) SetupSuite() { |
| |
| t := s.T() |
| |
| const ( |
| // kubernetes inject the var |
| podNameKey = "HOSTNAME" |
| nameSpaceKey = "NAMESPACE" |
| ) |
| |
| // 1. install test data |
| if err := json.Unmarshal([]byte(clientPodJsonData), &s.currentPod); err != nil { |
| t.Fatal(err) |
| } |
| |
| // 2. set downward-api inject env |
| if err := os.Setenv(podNameKey, s.currentPod.GetName()); err != nil { |
| t.Fatal(err) |
| } |
| if err := os.Setenv(nameSpaceKey, s.currentPod.GetNamespace()); err != nil { |
| t.Fatal(err) |
| } |
| |
| } |
| |
| func (s *KubernetesRegistryTestSuite) TestDataChange() { |
| |
| t := s.T() |
| |
| listener := NewRegistryDataListener(&MockDataListener{}) |
| url, _ := common.NewURL("jsonrpc%3A%2F%2F127.0.0.1%3A20001%2Fcom.ikurento.user.UserProvider%3Fanyhost%3Dtrue%26app.version%3D0.0.1%26application%3DBDTService%26category%3Dproviders%26cluster%3Dfailover%26dubbo%3Ddubbo-provider-golang-2.6.0%26environment%3Ddev%26group%3D%26interface%3Dcom.ikurento.user.UserProvider%26ip%3D10.32.20.124%26loadbalance%3Drandom%26methods.GetUser.loadbalance%3Drandom%26methods.GetUser.retries%3D1%26methods.GetUser.weight%3D0%26module%3Ddubbogo%2Buser-info%2Bserver%26name%3DBDTService%26organization%3Dikurento.com%26owner%3DZX%26pid%3D74500%26retries%3D0%26service.filter%3Decho%26side%3Dprovider%26timestamp%3D1560155407%26version%3D%26warmup%3D100") |
| listener.AddInterestedURL(&url) |
| if !listener.DataChange(remoting.Event{Path: "/dubbo/com.ikurento.user.UserProvider/providers/jsonrpc%3A%2F%2F127.0.0.1%3A20001%2Fcom.ikurento.user.UserProvider%3Fanyhost%3Dtrue%26app.version%3D0.0.1%26application%3DBDTService%26category%3Dproviders%26cluster%3Dfailover%26dubbo%3Ddubbo-provider-golang-2.6.0%26environment%3Ddev%26group%3D%26interface%3Dcom.ikurento.user.UserProvider%26ip%3D10.32.20.124%26loadbalance%3Drandom%26methods.GetUser.loadbalance%3Drandom%26methods.GetUser.retries%3D1%26methods.GetUser.weight%3D0%26module%3Ddubbogo%2Buser-info%2Bserver%26name%3DBDTService%26organization%3Dikurento.com%26owner%3DZX%26pid%3D74500%26retries%3D0%26service.filter%3Decho%26side%3Dprovider%26timestamp%3D1560155407%26version%3D%26warmup%3D100"}) { |
| t.Fatal("data change not ok") |
| } |
| } |
| |
| func TestKubernetesRegistrySuite(t *testing.T) { |
| suite.Run(t, &KubernetesRegistryTestSuite{}) |
| } |
