attrs_test.go

package gomol

import (
	"github.com/aphistic/sweet"
	. "github.com/onsi/gomega"
)

type AttrsSuite struct{}

func (s *AttrsSuite) TestNewAttrsFromMap(t sweet.T) {
	attrs := NewAttrsFromMap(map[string]interface{}{
		"attr1": "val1",
		"attr2": 1234,
	})
	Expect(attrs.attrs).To(HaveLen(2))
	Expect(attrs.attrs[getAttrHash("attr1")]).To(Equal("val1"))
	Expect(attrs.attrs[getAttrHash("attr2")]).To(Equal(1234))
}

func (s *AttrsSuite) TestAttrsMergeNilAttrs(t sweet.T) {
	attrs := NewAttrs()
	attrs.MergeAttrs(nil)
}

func (s *AttrsSuite) TestNewAttrsFromAttrsNone(t sweet.T) {
	testAttrs := NewAttrsFromAttrs()
	Expect(testAttrs).ToNot(BeNil())
	Expect(testAttrs.Attrs()).To(HaveLen(0))
}

func (s *AttrsSuite) TestNewAttrsFromAttrsSingle(t sweet.T) {
	attrs1 := NewAttrsFromMap(map[string]interface{}{
		"attr1": "val1",
		"attr3": 1234,
	})

	testAttrs := NewAttrsFromAttrs(attrs1)
	Expect(testAttrs).ToNot(BeNil())
	Expect(testAttrs.GetAttr("attr1")).To(Equal("val1"))
	Expect(testAttrs.GetAttr("attr3")).To(Equal(1234))
}

func (s *AttrsSuite) TestNewAttrsFromAttrsMultiple(t sweet.T) {
	attrs1 := NewAttrsFromMap(map[string]interface{}{
		"attr1": "val1",
		"attr3": 1234,
	})
	attrs2 := NewAttrsFromMap(map[string]interface{}{
		"attr2": true,
		"attr4": 1234.4321,
	})

	testAttrs := NewAttrsFromAttrs(attrs1, attrs2)
	Expect(testAttrs).ToNot(BeNil())
	Expect(testAttrs.GetAttr("attr1")).To(Equal("val1"))
	Expect(testAttrs.GetAttr("attr2")).To(Equal(true))
	Expect(testAttrs.GetAttr("attr3")).To(Equal(1234))
	Expect(testAttrs.GetAttr("attr4")).To(Equal(1234.4321))
}

func (s *AttrsSuite) TestNewAttrsFromAttrsNil(t sweet.T) {
	testAttrs := NewAttrsFromAttrs(nil, nil, nil)
	Expect(testAttrs).ToNot(BeNil())
	Expect(testAttrs.Attrs()).To(HaveLen(0))
}

func (s *AttrsSuite) TestAttrsGetMissing(t sweet.T) {
	attrs := NewAttrs()
	Expect(attrs.GetAttr("not an attr")).To(BeNil())
}

func (s *AttrsSuite) TestAttrsRemoveMissing(t sweet.T) {
	attrs := NewAttrs()
	// Just run it to make sure it doesn't panic
	attrs.RemoveAttr("not an attr")
}

func (s *AttrsSuite) TestAttrsChaining(t sweet.T) {
	attrs := NewAttrs().
		SetAttr("attr1", "val1").
		SetAttr("attr2", "val2").
		SetAttr("attr3", 3).
		SetAttr("attr4", 4)

	Expect(attrs.attrs[getAttrHash("attr1")]).To(Equal("val1"))
	Expect(attrs.attrs[getAttrHash("attr2")]).To(Equal("val2"))
	Expect(attrs.attrs[getAttrHash("attr3")]).To(Equal(3))
	Expect(attrs.attrs[getAttrHash("attr4")]).To(Equal(4))
}

func (s *AttrsSuite) TestGetHashAttrMissing(t sweet.T) {
	res, err := getHashAttr(1234)

	Expect(res).To(Equal(""))
	Expect(err.Error()).To(Equal("Could not find attr for hash 1234"))
}

func (s *AttrsSuite) TestSetWithFuncValue(t sweet.T) {
	attrs := NewAttrs()
	attrs.SetAttr("attr1", func() int { return 1 })

	Expect(attrs.GetAttr("attr1")).To(Equal("func() int"))
}

func (s *AttrsSuite) TestMapWithFuncValue(t sweet.T) {
	attrs := NewAttrsFromMap(map[string]interface{}{
		"attr1": func() int { return 1 },
	})

	Expect(attrs.GetAttr("attr1")).To(Equal("func() int"))
}