package tests

import (
	"bytes"
	"encoding/json"
	"github.com/fnproject/fn_go/models"
	"net/url"
	"path"
	"strconv"
	"strings"
	"testing"
)

type JSONResponse struct {
	Message string `json:"message"`
}

func TestFnJSONFormats(t *testing.T) {
	t.Parallel()
	s := SetupHarness()
	defer s.Cleanup()

	// TODO(treeder): put image in fnproject @ dockerhub

	s.GivenAppExists(t, &models.App{Name: s.AppName})
	rt := s.BasicRoute()
	rt.Image = "denismakogon/test-hot-json-go:0.0.1"
	rt.Format = "json"
	s.GivenRouteExists(t, s.AppName, rt)

	u := url.URL{
		Scheme: "http",
		Host:   Host(),
	}
	u.Path = path.Join(u.Path, "r", s.AppName, s.RoutePath)

	b, _ := json.Marshal(&struct {
		Name string `json:"name"`
	}{
		Name: "Jimmy",
	})
	content := bytes.NewBuffer(b)
	output := &bytes.Buffer{}
	resp, err := CallFN(s.Context, u.String(), content, output, "POST", []string{})
	if err != nil {
		t.Errorf("Got unexpected error: %v", err)
	}

	msg := &JSONResponse{}
	json.Unmarshal(output.Bytes(), msg)
	expectedOutput := "Hello Jimmy"
	if !strings.Contains(expectedOutput, msg.Message) {
		t.Errorf("Assertion error.\n\tExpected: %v\n\tActual: %v", expectedOutput, output.String())
	}

	expectedHeaderNames := []string{"Content-Type", "Content-Length"}
	expectedHeaderValues := []string{"application/json; charset=utf-8", strconv.Itoa(output.Len())}
	for i, name := range expectedHeaderNames {
		actual := resp.Header.Get(name)
		expected := expectedHeaderValues[i]
		if !strings.Contains(expected, actual) {
			t.Errorf("HTTP header assertion error for %v."+
				"\n\tExpected: %v\n\tActual: %v", name, expected, actual)
		}
	}

}