main_test.go

package main

import (
	"crypto/tls"
	"net/http"
	"testing"
)

func TestServeTLS(t *testing.T) {
	*configFile = "testdata/tls.conf.yml"
	cfg, err := reloadConfig()
	if err != nil {
		t.Fatalf("unexpected error while loading config: %s", err)
	}

	done := make(chan struct{})
	ln := newTLSListener(cfg.ListenAddr, &cfg.TLSConfig)
	go func() {
		listenAndServe(ln)
		close(done)
	}()

	tr := &http.Transport{
		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
	}
	client := &http.Client{Transport: tr}

	resp, err := client.Get("https://127.0.0.1:9090/metrics")
	if err != nil {
		t.Fatalf("unexpected error: %s", err)
	}

	if resp.StatusCode != http.StatusOK {
		t.Fatalf("unexpected status code: %d; expected: %d", resp.StatusCode, http.StatusOK)
	}

	if err := ln.Close(); err != nil {
		t.Fatalf("unexpected error while closing listener: %s", err)
	}
	<-done
}

func TestServe(t *testing.T) {
	*configFile = "testdata/http.conf.yml"
	cfg, err := reloadConfig()
	if err != nil {
		t.Fatalf("unexpected error while loading config: %s", err)
	}

	done := make(chan struct{})
	ln := newListener(cfg.ListenAddr)
	go func() {
		listenAndServe(ln)
		close(done)
	}()

	resp, err := http.Get("http://127.0.0.1:8080/metrics")
	if err != nil {
		t.Fatalf("unexpected error: %s", err)
	}

	if resp.StatusCode != http.StatusOK {
		t.Fatalf("unexpected status code: %d; expected: %d", resp.StatusCode, http.StatusOK)
	}

	if err := ln.Close(); err != nil {
		t.Fatalf("unexpected error while closing listener: %s", err)
	}
	<-done
}