constant_test.go

package humanize

import (
	"testing"

	. "github.com/smartystreets/goconvey/convey"
)

var cr = `
package test

const (
   i = iota
   j
)

const s float64 = 10

const i1 = 10

const i2 = 10.6

const i3 = "test"

const i4 = 'C'

const i6 = 10i

`

func TestConstant(t *testing.T) {
	Convey("constant test", t, func() {
		f, err := ParseFile(cr, &Package{})
		So(err, ShouldBeNil)
		var p = &Package{}
		p.Files = append(p.Files, f)
		Convey("Normal define", func() {
			i, err := p.FindConstant("i")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "i")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "iota")

			i, err = p.FindConstant("j")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "j")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "iota")

		})
		Convey("by value s", func() {
			i, err := p.FindConstant("s")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "s")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "float64")
		})
		Convey("by value i1", func() {
			i, err := p.FindConstant("i1")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "i1")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "int")
		})

		Convey("by value i2", func() {
			i, err := p.FindConstant("i2")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "i2")
			So(i.Name, ShouldEqual, "i2")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "float64")
		})

		Convey("by value i3", func() {
			i, err := p.FindConstant("i3")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "i3")
			So(i.Name, ShouldEqual, "i3")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "string")
		})

		Convey("by value i4", func() {
			i, err := p.FindConstant("i4")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "i4")
			So(i.Name, ShouldEqual, "i4")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "char")
		})

		Convey("by value i6", func() {
			i, err := p.FindConstant("i6")
			So(err, ShouldBeNil)
			So(i.Name, ShouldEqual, "i6")
			So(i.Name, ShouldEqual, "i6")
			So(i.Type.(*IdentType).Ident, ShouldEqual, "complex64")
		})
	})
}