httpbin.go

// Copyright 2022 The Amesh Authors
//
// Licensed 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 framework

import (
	"fmt"
	"strings"
	"time"

	"github.com/api7/gopkg/pkg/log"

	"github.com/api7/amesh/e2e/framework/utils"
)

const (
	_httpbinManifest = `
apiVersion: apps/v1
kind: Deployment
metadata:
  name: {{ .Name }}
  labels:
    appKind: httpbin
    app: {{ .Name }}
    version: {{ .Version }}
spec:
  replicas: {{ .HttpBinReplicas }}
  selector:
    matchLabels:
      appKind: httpbin
      app: {{ .Name }}
      version: {{ .Version }}
  template:
    metadata:
      labels:
        appKind: httpbin
        app: {{ .Name }}
        version: {{ .Version }}
      annotations:
        sidecar.istio.io/inject: "{{ .InMesh }}"
    spec:
      containers:
      - name: httpbin
        image: {{ .LocalRegistry }}/kennethreitz/httpbin
        imagePullPolicy: IfNotPresent
        command: ["gunicorn", "--access-logfile", "-", "-b", "0.0.0.0:80", "httpbin:app"]
        ports:
        - containerPort: 80
          protocol: TCP
          name: http
---
apiVersion: v1
kind: Service
metadata:
  name: {{ .Name }}
spec:
  selector:
    app: {{ .Name }}
  ports:
  - name: http
    targetPort: 80
    port: 80
    protocol: TCP
`
	httpbinService = `
apiVersion: v1
kind: Service
metadata:
  name: httpbin-kind
spec:
  selector:
    appKind: httpbin
  ports:
  - name: http
    targetPort: 80
    port: 80
    protocol: TCP
`
)

func (f *Framework) CreateHttpbinService() {
	utils.AssertNil(f.ApplyResourceFromString(httpbinService), "create httpbin service")
}

type httpbinRenderArgs struct {
	*ManifestArgs
	HttpBinReplicas int
	Name            string
	Version         string
	InMesh          bool
}

func (f *Framework) newHttpBin(name string, inMesh bool) {
	str := "inside mesh"
	if !inMesh {
		str = "outside mesh"
	}
	log.Infof("Creating HTTPBIN " + str)

	version := "v1"
	if strings.HasSuffix(name, "-v2") {
		version = "v2"
	} else if strings.HasSuffix(name, "-v3") {
		version = "v3"
	}

	artifact, err := utils.RenderManifest(_httpbinManifest, &httpbinRenderArgs{
		ManifestArgs:    f.args,
		HttpBinReplicas: 1,
		Name:            name,
		Version:         version,
		InMesh:          inMesh,
	})
	utils.AssertNil(err, "render httpbin %s template", name)

	log.Infof("creating httpbin %s", name)
	err = f.ApplyResourceFromString(artifact)
	utils.AssertNil(err, "apply httpbin %s", name)
}

func (f *Framework) CreateHttpbinInMesh(nameOpt ...string) {
	name := "httpbin"
	if len(nameOpt) > 0 {
		name = nameOpt[0]
	}

	f.newHttpBin(name, true)
}

func (f *Framework) CreateHttpbinOutsideMesh(nameOpt ...string) {
	name := "httpbin"
	if len(nameOpt) > 0 {
		name = nameOpt[0]
	}

	f.newHttpBin(name, false)
}

func (f *Framework) WaitForHttpbinReady(nameOpt ...string) {
	name := "httpbin"
	if len(nameOpt) > 0 {
		name = nameOpt[0]
	}

	log.Infof("wait for httpbin (%v) ready", name)
	defer utils.LogTimeTrack(time.Now(), name+" ready (%v)")
	utils.AssertNil(f.WaitForDeploymentPodsReady(name), "wait for httpbin ready")
}

// GetHttpBinServiceFQDN returns the FQDN description for HttpBin service.
func (f *Framework) GetHttpBinServiceFQDN(nameOpt ...string) string {
	name := "httpbin"
	if len(nameOpt) > 0 {
		name = nameOpt[0]
	}

	return fmt.Sprintf("%s.%s.svc", name, f.namespace)
}