block_funcParam_test.go

package codegen

import (
	"strings"
	"testing"

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

func TestParamName(t *testing.T) {
	const want = `name`
	got := Param("name", "typeName").GetName()

	assert.Equal(t, want, got)
}

func TestQualParamFullName(t *testing.T) {
	const want = `name`
	got := QualParam("name", "alias", "typeName").GetName()

	assert.Equal(t, want, got)
}

func TestParamFullType(t *testing.T) {
	const want = `typeName`
	got := Param("name", "typeName").GetFullType()

	assert.Equal(t, want, got)
}

func TestQualParamFullType(t *testing.T) {
	const want = `alias.typeName`
	got := QualParam("name", "alias", "typeName").GetFullType()

	assert.Equal(t, want, got)
}

func TestParamFullTypePointer(t *testing.T) {
	const want = `*typeName`
	got := Param("name", "typeName").Pointer().GetFullType()

	assert.Equal(t, want, got)
}

func TestQualParamTypePointer(t *testing.T) {
	const want = `*alias.typeName`
	got := QualParam("name", "alias", "typeName").Pointer().GetFullType()

	assert.Equal(t, want, got)
}

func TestFuncParam(t *testing.T) {
	const want = `name type`

	var sb strings.Builder
	Param("name", "type").wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamPointer(t *testing.T) {
	const want = `name *type`

	var sb strings.Builder
	Param("name", "type").Pointer().
		wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncQualParam(t *testing.T) {
	const want = `name alias.type`

	var sb strings.Builder
	QualParam("name", "alias", "type").wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncQualParamPointer(t *testing.T) {
	const want = `name *alias.type`

	var sb strings.Builder
	QualParam("name", "alias", "type").Pointer().
		wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamsEmpty(t *testing.T) {
	const want = `()`

	var sb strings.Builder
	params := make([]*ParamDecl, 0)
	writeParams(&sb, params)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamsOne(t *testing.T) {
	const want = `(name1 type)`

	var sb strings.Builder
	params := []*ParamDecl{
		Param("name1", "type"),
	}
	writeParams(&sb, params)

	assert.Equal(t, want, sb.String())
}

func TestFuncParams(t *testing.T) {
	const want = `(name1 type,name2 alias.type,name3 *type,name4 *alias.type)`

	var sb strings.Builder
	params := []*ParamDecl{
		Param("name1", "type"),
		QualParam("name2", "alias", "type"),
		Param("name3", "type").Pointer(),
		QualParam("name4", "alias", "type").Pointer(),
	}
	writeParams(&sb, params)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamSetIsPointerTrue(t *testing.T) {
	const want = `(name1 *type)`

	var sb strings.Builder
	params := []*ParamDecl{
		Param("name1", "type").SetIsPointer(true),
	}
	writeParams(&sb, params)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamSetIsPointerFalse(t *testing.T) {
	const want = `(name1 type)`

	var sb strings.Builder
	params := []*ParamDecl{
		Param("name1", "type").SetIsPointer(false),
	}
	writeParams(&sb, params)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamArray(t *testing.T) {
	const want = `param []string`

	var sb strings.Builder
	Param("param", "string").Array().
		wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncQualParamArray(t *testing.T) {
	const want = `param []alias.MyType`

	var sb strings.Builder
	QualParam("param", "alias", "MyType").Array().
		wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamSetIsArrayTrue(t *testing.T) {
	const want = `param []string`

	var sb strings.Builder
	Param("param", "string").SetIsArray(true).
		wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncParamSetIsArrayFalse(t *testing.T) {
	const want = `param string`

	var sb strings.Builder
	Param("param", "string").SetIsArray(false).
		wr(&sb)

	assert.Equal(t, want, sb.String())
}

func TestFuncGetters(t *testing.T) {
	fixture := QualParam("param", "alias", "MyType").Pointer().Array()

	assert.Equal(t, "param", fixture.GetName())
	assert.Equal(t, "alias", fixture.GetTypeAlias())
	assert.Equal(t, "MyType", fixture.GetTypeName())
	assert.True(t, fixture.GetIsPointer())
	assert.True(t, fixture.GetIsArray())
}