| /* |
| * 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 stream_route_test |
| |
| import ( |
| "encoding/json" |
| "io/ioutil" |
| "net" |
| "net/http" |
| "time" |
| |
| . "github.com/onsi/ginkgo" |
| . "github.com/onsi/ginkgo/extensions/table" |
| . "github.com/onsi/gomega" |
| |
| "github.com/apisix/manager-api/test/e2e/base" |
| ) |
| |
| var _ = Describe("Stream Route", func() { |
| DescribeTable("test stream route data CURD", |
| func(tc base.HttpTestCase) { |
| base.RunTestCase(tc) |
| }, |
| // base case |
| Entry("create stream route", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/stream_routes", |
| Body: `{ |
| "id": "sr1", |
| "remote_addr": "127.0.0.1", |
| "server_addr": "127.0.0.1", |
| "server_port": 10090, |
| "sni": "test.com", |
| "upstream": { |
| "nodes": { |
| "` + base.UpstreamIp + `:1980": 1 |
| }, |
| "type": "roundrobin" |
| } |
| }`, |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }), |
| Entry("get stream route #1", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodGet, |
| Path: "/apisix/admin/stream_routes/sr1", |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| ExpectBody: `"server_port":10090`, |
| }), |
| Entry("update stream route", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPut, |
| Path: "/apisix/admin/stream_routes/sr1", |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| Body: `{ |
| "id": "sr1", |
| "server_port": 10091, |
| "upstream": { |
| "nodes": { |
| "` + base.UpstreamIp + `:1980": 1 |
| }, |
| "type": "roundrobin" |
| } |
| }`, |
| ExpectStatus: http.StatusOK, |
| ExpectBody: `"server_port":10091`, |
| }), |
| Entry("get stream route #2", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodGet, |
| Path: "/apisix/admin/stream_routes/sr1", |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| ExpectBody: `"server_port":10091`, |
| }), |
| Entry("hit stream route", base.HttpTestCase{ |
| Object: base.APISIXStreamProxyExpect(10091, ""), |
| Method: http.MethodGet, |
| Path: "/hello", |
| ExpectStatus: http.StatusOK, |
| ExpectBody: "hello world", |
| }), |
| Entry("delete stream route", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodDelete, |
| Path: "/apisix/admin/stream_routes/sr1", |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }), |
| ) |
| |
| DescribeTable("test stream route with HTTP upstream", |
| func(tc base.HttpTestCase) { |
| base.RunTestCase(tc) |
| }, |
| Entry("create stream route", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/stream_routes", |
| Body: `{ |
| "id": "sr1", |
| "server_port": 10090, |
| "upstream": { |
| "nodes": { |
| "` + base.UpstreamIp + `:1980": 1 |
| }, |
| "type": "roundrobin" |
| } |
| }`, |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }), |
| Entry("hit stream route", base.HttpTestCase{ |
| Object: base.APISIXStreamProxyExpect(10090, ""), |
| Method: http.MethodGet, |
| Path: "/hello", |
| ExpectStatus: http.StatusOK, |
| ExpectBody: "hello world", |
| }), |
| Entry("delete stream route", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodDelete, |
| Path: "/apisix/admin/stream_routes/sr1", |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }), |
| ) |
| |
| // prepare ssl certificate |
| apisixCert, err := ioutil.ReadFile("../../certs/apisix.crt") |
| Expect(err).To(BeNil()) |
| apisixKey, err := ioutil.ReadFile("../../certs/apisix.key") |
| Expect(err).To(BeNil()) |
| apisixSSLBody, err := json.Marshal(map[string]string{"cert": string(apisixCert), "key": string(apisixKey)}) |
| Expect(err).To(BeNil()) |
| DescribeTable("test stream route with HTTPS upstream", |
| func(tc base.HttpTestCase) { |
| base.RunTestCase(tc) |
| }, |
| Entry("create ssl cert", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/ssl", |
| Body: string(apisixSSLBody), |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }), |
| Entry("create stream route", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/stream_routes", |
| Body: `{ |
| "id": "sr1", |
| "server_port": 10093, |
| "sni": "test.com", |
| "upstream": { |
| "nodes": { |
| "` + base.UpstreamIp + `:1980": 1 |
| }, |
| "type": "roundrobin" |
| } |
| }`, |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }), |
| Entry("hit stream route through https", base.HttpTestCase{ |
| Object: base.APISIXStreamProxyExpect(10093, "test.com"), |
| Method: http.MethodGet, |
| Path: "/hello", |
| ExpectStatus: http.StatusOK, |
| ExpectBody: "hello world", |
| }), |
| ) |
| |
| Describe("test stream route with TCP upstream", func() { |
| It("create stream route", func() { |
| base.RunTestCase(base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/stream_routes", |
| Body: `{ |
| "id": "sr1tcp", |
| "server_port": 10090, |
| "upstream": { |
| "nodes": { |
| "` + base.UpstreamIp + `:1991": 1 |
| }, |
| "type": "roundrobin" |
| } |
| }`, |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }) |
| }) |
| It("hit stream route through tcp", func() { |
| time.Sleep(base.SleepTime) |
| conn, err := net.Dial("tcp", "127.0.0.1:1991") |
| Expect(err).To(BeNil()) |
| |
| _ = conn.SetDeadline(time.Now().Add(time.Second * 3)) |
| |
| _, err = conn.Write([]byte("world")) |
| Expect(err).To(BeNil()) |
| |
| result := make([]byte, 11) |
| n, err := conn.Read(result) |
| Expect(n).Should(BeNumerically("==", 11)) |
| Expect(err).To(BeNil()) |
| Expect(string(result)).To(ContainSubstring("hello world")) |
| |
| err = conn.Close() |
| Expect(err).To(BeNil()) |
| }) |
| }) |
| |
| Describe("test stream route with UDP upstream", func() { |
| It("create stream route", func() { |
| base.RunTestCase(base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/stream_routes", |
| Body: `{ |
| "id": "sr1udp", |
| "server_port": 10095, |
| "upstream": { |
| "nodes": { |
| "` + base.UpstreamIp + `:1992": 1 |
| }, |
| "type": "roundrobin" |
| } |
| }`, |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusOK, |
| }) |
| }) |
| It("hit stream route through udp", func() { |
| time.Sleep(base.SleepTime) |
| conn, err := net.Dial("udp", "127.0.0.1:10095") |
| Expect(err).To(BeNil()) |
| |
| _ = conn.SetDeadline(time.Now().Add(time.Second * 3)) |
| |
| _, err = conn.Write([]byte("world")) |
| Expect(err).To(BeNil()) |
| |
| result := make([]byte, 11) |
| n, err := conn.Read(result) |
| Expect(n).Should(BeNumerically("==", 11)) |
| Expect(err).To(BeNil()) |
| Expect(string(result)).To(ContainSubstring("hello world")) |
| |
| err = conn.Close() |
| Expect(err).To(BeNil()) |
| }) |
| }) |
| |
| DescribeTable("test stream route data CURD exception", |
| func(tc base.HttpTestCase) { |
| base.RunTestCase(tc) |
| }, |
| Entry("create stream route with upstream id not found", base.HttpTestCase{ |
| Object: base.ManagerApiExpect(), |
| Method: http.MethodPost, |
| Path: "/apisix/admin/stream_routes", |
| Body: `{ |
| "id": "sr1", |
| "server_port": 10090, |
| "upstream_id": "u1" |
| }`, |
| Headers: map[string]string{"Authorization": base.GetToken()}, |
| ExpectStatus: http.StatusBadRequest, |
| }), |
| ) |
| }) |