wrap_test.go

package postgres

import (
	"testing"

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

type WrapTestSuite struct {
	suite.Suite
	wrap *Wrap
}

func TestWrapSuite(t *testing.T) {
	suite.Run(t, new(WrapTestSuite))
}

func (s *WrapTestSuite) SetupTest() {
	s.wrap = NewWrap("prefix_")
}

func (s *WrapTestSuite) TestColumn() {
	// With alias
	result := s.wrap.Column("column as alias")
	s.Equal(`"column" as "prefix_alias"`, result)

	// Without alias
	result = s.wrap.Column("column")
	s.Equal(`"column"`, result)
}

func (s *WrapTestSuite) TestColumnize() {
	result := s.wrap.Columnize([]string{"column1", "column2 as alias2"})
	s.Equal(`"column1", "column2" as "prefix_alias2"`, result)
}

func (s *WrapTestSuite) TestQuote() {
	// With non empty value
	result := s.wrap.Quote("value")
	s.Equal("'value'", result)

	// With empty value
	result = s.wrap.Quote("")
	s.Equal("", result)
}

func (s *WrapTestSuite) TestQuotes() {
	result := s.wrap.Quotes([]string{"value1", "value2"})
	s.Equal([]string{"'value1'", "'value2'"}, result)
}

func (s *WrapTestSuite) TestSegmentsWithMultipleSegments() {
	result := s.wrap.Segments([]string{"table", "column"})
	s.Equal(`"prefix_table"."column"`, result)
}

func (s *WrapTestSuite) TestTable() {
	// With alias
	result := s.wrap.Table("table as alias")
	s.Equal(`"prefix_table" as "prefix_alias"`, result)

	// With schema
	result = s.wrap.Table("goravel.table")
	s.Equal(`"goravel"."prefix_table"`, result)

	// Without alias
	result = s.wrap.Table("table")
	s.Equal(`"prefix_table"`, result)
}

func (s *WrapTestSuite) TestValue() {
	// With asterisk
	result := s.wrap.Value("*")
	s.Equal("*", result)

	// Without asterisk
	result = s.wrap.Value("value")
	s.Equal(`"value"`, result)
}

func (s *WrapTestSuite) TestAliasedTable() {
	result := s.wrap.aliasedTable("users as u")
	s.Equal(`"prefix_users" as "prefix_u"`, result)
}

func (s *WrapTestSuite) TestAliasedValue() {
	result := s.wrap.aliasedValue("users.name as user_name")
	s.Equal(`"prefix_users"."name" as "prefix_user_name"`, result)
}