value_comparison_test.go

package codegen

import (
	"strings"
	"testing"

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

func TestComparisonPanicUnknownType(t *testing.T) {
	assert.Panics(t, func() {
		var sb strings.Builder
		newEquals(Identifier("a"), Identifier("b"), 123).
			writeValue(&sb)
	})
}

func TestEqualsIdentifierIdentifier(t *testing.T) {
	const want = `a==b.field`

	var sb strings.Builder
	newEquals(Identifier("a"), Identifier("b").Field("field"), cmpType_Equals).
		writeValue(&sb)

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

func TestNotEqualsIdentifierIdentifier(t *testing.T) {
	const want = `a!=b.field`

	var sb strings.Builder
	newEquals(Identifier("a"), Identifier("b").Field("field"), cmpType_NotEquals).
		writeValue(&sb)

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

func TestLessThanIdentifierIdentifier(t *testing.T) {
	const want = `a<b.field`

	var sb strings.Builder
	newEquals(Identifier("a"), Identifier("b").Field("field"), cmpType_LessThan).
		writeValue(&sb)

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

func TestEqualsIdentifierFunc(t *testing.T) {
	const want = `a==myFunc()`

	var sb strings.Builder
	newEquals(Identifier("a"), FuncCall("myFunc"), cmpType_Equals).
		writeValue(&sb)

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

func TestNotEqualsIdentifierFunc(t *testing.T) {
	const want = `a!=myFunc()`

	var sb strings.Builder
	newEquals(Identifier("a"), FuncCall("myFunc"), cmpType_NotEquals).
		writeValue(&sb)

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

func TestLessThanIdentifierFunc(t *testing.T) {
	const want = `a<myFunc()`

	var sb strings.Builder
	newEquals(Identifier("a"), FuncCall("myFunc"), cmpType_LessThan).
		writeValue(&sb)

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

func TestEqualsFuncIdentifier(t *testing.T) {
	const want = `myFunc()==a`

	var sb strings.Builder
	newEquals(FuncCall("myFunc"), Identifier("a"), cmpType_Equals).
		writeValue(&sb)

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

func TestNotEqualsFuncIdentifier(t *testing.T) {
	const want = `myFunc()!=b.field`

	var sb strings.Builder
	newEquals(FuncCall("myFunc"), Identifier("b").Field("field"), cmpType_NotEquals).
		writeValue(&sb)

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

func TestLessThanFuncIdentifier(t *testing.T) {
	const want = `myFunc()<b.field`

	var sb strings.Builder
	newEquals(FuncCall("myFunc"), Identifier("b").Field("field"), cmpType_LessThan).
		writeValue(&sb)

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

func TestEqualsFuncFunc(t *testing.T) {
	const want = `alias.MyFunc()==myFunc()`

	var sb strings.Builder
	newEquals(QualFuncCall("alias", "MyFunc"), FuncCall("myFunc"), cmpType_Equals).
		writeValue(&sb)

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

func TestNotEqualsFuncFunc(t *testing.T) {
	const want = `alias.MyFunc()!=myFunc()`

	var sb strings.Builder
	newEquals(QualFuncCall("alias", "MyFunc"), FuncCall("myFunc"), cmpType_NotEquals).
		writeValue(&sb)

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

func TestLessThanFuncFunc(t *testing.T) {
	const want = `alias.MyFunc()<myFunc()`

	var sb strings.Builder
	newEquals(QualFuncCall("alias", "MyFunc"), FuncCall("myFunc"), cmpType_LessThan).
		writeValue(&sb)

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

func TestEqualsNoPointer(t *testing.T) {
	got := newEquals(Identifier("a"), Identifier("b"), cmpType_Equals).
		isPointer()

	assert.False(t, got)
}