straf_test.go

package straf

import (
	"reflect"
	"testing"

	"github.com/graphql-go/graphql"
	"github.com/stretchr/testify/assert"
)

type extra struct {
	Age     int
	FavNums []float32
}

type colors struct {
	Name     string
	HexValue string
}

type user struct {
	UserID    int `description:"UserID"`
	ExtraData extra
	FavColors []colors
	Age       int    `deprecationReason:"Age is now in ExtraData"`
	ID        string `unique:"true"`
}

//Tests generation of graphQL Object from struct
func TestGraphQLObjectGen(t *testing.T) {
	assert := assert.New(t)

	graphQLObject, err := GetGraphQLObject(user{})
	assert.NoError(err, "GetGraphQLObject Returned Error")
	extraType, err := GetGraphQLObject(extra{})
	assert.NoError(err, "GetGraphQLObject Returned Error")
	colorType, err := GetGraphQLObject(colors{})
	assert.NoError(err, "GetGraphQLObject Returned Error")

	testField(
		"UserID",
		*graphQLObject.Fields()["UserID"],
		graphql.Int,
		*assert,
	)

	testField(
		"ID",
		*graphQLObject.Fields()["ID"],
		graphql.ID,
		*assert,
	)

	testField(
		"ExtraData",
		*graphQLObject.Fields()["ExtraData"],
		extraType,
		*assert,
	)

	testField(
		"FavColors",
		*graphQLObject.Fields()["FavColors"],
		graphql.NewList(colorType),
		*assert,
	)

	assert.Equal(graphQLObject.Fields()["Age"].DeprecationReason, "Age is now in ExtraData")
	assert.Equal(graphQLObject.Fields()["UserID"].Description, "UserID")

}

func testField(name string,
	definition graphql.FieldDefinition,
	graphqlType graphql.Output,
	assert assert.Assertions) {

	assert.Equal(definition.Name, name)
	assert.Equal(definition.Type, graphqlType)
}

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

	userType := reflect.TypeOf(user{})
	scalar, err := convertSimpleType(userType)
	assert.Error(err)
	assert.Equal("Invalid Type", err.Error())
	assert.Equal(&graphql.Scalar{}, scalar)
}

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

	userType := reflect.TypeOf(user{})
	field, _ := userType.FieldByName("UserID")
	assert.Equal("", getTagValue(field, "random"))
	assert.Equal("UserID", getTagValue(field, "description"))
}