package integration_test

import (
	"fmt"
	"net"
	"os"
	"path/filepath"
	"time"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	"k8s.io/kubectl/pkg/framework/test"
)

var _ = Describe("Integration", func() {
	It("Successfully manages the fixtures lifecycle", func() {
		assetsDir, ok := os.LookupEnv("KUBE_ASSETS_DIR")
		Expect(ok).To(BeTrue(), "Expected $KUBE_ASSETS_DIR to be set")

		pathToEtcd := filepath.Join(assetsDir, "etcd")
		pathToApiserver := filepath.Join(assetsDir, "kube-apiserver")

		fixtures := test.NewFixtures(pathToEtcd, pathToApiserver)

		err := fixtures.Start()
		Expect(err).NotTo(HaveOccurred(), "Expected fixtures to start successfully")

		isEtcdListening := isSomethingListeningOnPort(2379)
		isAPIServerListening := isSomethingListeningOnPort(8080)

		Expect(isEtcdListening()).To(BeTrue(), "Expected Etcd to listen on 2379")

		Expect(isAPIServerListening()).To(BeTrue(), "Expected APIServer to listen on 8080")

		err = fixtures.Stop()
		Expect(err).NotTo(HaveOccurred(), "Expected fixtures to stop successfully")

		Expect(isEtcdListening()).To(BeFalse(), "Expected Etcd not to listen anymore")

		By("Ensuring APIServer is not listening anymore")
		Expect(isAPIServerListening()).To(BeFalse(), "Expected APIServer not to listen anymore")
	})
})

type portChecker func() bool

func isSomethingListeningOnPort(port int) portChecker {
	return func() bool {
		conn, err := net.DialTimeout("tcp", net.JoinHostPort("", fmt.Sprintf("%d", port)), 1*time.Second)

		if err != nil {
			return false
		}
		conn.Close()
		return true
	}
}