data_test.go

// Copyright 2021-2024 Nokia
// Licensed under the BSD 3-Clause License.
// SPDX-License-Identifier: BSD-3-Clause

package restful

import (
	"net/http"
	"net/http/httptest"
	"net/url"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

type abcType struct {
	A      string
	B      []string
	CField int `schema:"c-field"`
}

func TestDataGetQuery(t *testing.T) {
	assert := assert.New(t)

	// Server
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		var abc abcType
		GetRequestData(r, 0, &abc)
		assert.Equal("a", abc.A)
		assert.Equal("b", abc.B[0])
		assert.Equal("B", abc.B[1])
		assert.Equal(42, abc.CField)
		w.WriteHeader(http.StatusOK)
	}))
	defer srv.Close()

	// Client
	{
		req, err := http.NewRequest("GET", srv.URL, nil)
		assert.Nil(err)
		q := url.Values{"a": {"a"}, "b": {"b", "B"}, "c-field": {"42"}}
		req.URL.RawQuery = q.Encode()
		_, err = NewClient().Do(req)
		assert.Nil(err)
	}
}

func TestDataGetQueryLambda(t *testing.T) {
	assert := assert.New(t)

	// Server
	// A real listening one, so that one can make a capture.
	mux := NewRouter()
	mux.HandleFunc("/", func(ab abcType) error {
		assert.Equal("a", ab.A)
		assert.Equal("b", ab.B[0])
		assert.Equal("B", ab.B[1])
		return nil
	})

	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		mux.ServeHTTP(w, r)
	}))
	defer srv.Close()

	// Client
	{
		req, err := http.NewRequest("GET", srv.URL, nil)
		assert.NoError(err)
		q := url.Values{"a": {"a"}, "b": {"b", "B"}}
		req.URL.RawQuery = q.Encode()
		_, err = NewClient().Do(req)
		assert.NoError(err)
	}
}

func TestDataPostForm(t *testing.T) {
	assert := assert.New(t)

	// Server
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		var ab abcType
		GetRequestData(r, 0, &ab)
		assert.Equal("a", ab.A)
		assert.Equal("b", ab.B[0])
		assert.Equal("B", ab.B[1])
		w.WriteHeader(http.StatusOK)
	}))
	defer srv.Close()

	// Client
	{
		f := url.Values{"a": {"a"}, "b": {"b", "B"}}
		req, err := http.NewRequest("POST", srv.URL, strings.NewReader(f.Encode()))
		assert.Nil(err)
		req.Header.Set("content-type", "application/x-www-form-urlencoded")
		_, err = NewClient().Do(req)
		assert.Nil(err)
	}
}