Merge pull request #6 from kilisima/fix-argument

fix Get() function. argument type string to interface{}

Author: smallfish

simpleyaml.go

@@ -52,7 +52,7 @@ func NewYaml(body []byte) (*Yaml, error) {
 //
 // Example:
 //      y.Get("xx").Get("yy").Int()
-func (y *Yaml) Get(key string) *Yaml {
+func (y *Yaml) Get(key interface{}) *Yaml {
 	m, err := y.Map()
 	if err == nil {
 		if val, ok := m[key]; ok {
@@ -66,7 +66,7 @@ func (y *Yaml) Get(key string) *Yaml {
 //
 // Example:
 //      y.GetPath("bb", "cc").Int()
-func (y *Yaml) GetPath(branch ...string) *Yaml {
+func (y *Yaml) GetPath(branch ...interface{}) *Yaml {
 	yin := y
 	for _, p := range branch {
 		yin = yin.Get(p)

simpleyaml_test.go

@@ -9,6 +9,7 @@ name: smallfish
 age: 99
 float: 3.14159
 bool: true
+0: IntKey
 emails:
    - xxx@xx.com
    - yyy@yy.com
@@ -51,6 +52,22 @@ func TestString(t *testing.T) {
 	}
 }
 
+func TestStringFromIntKey(t *testing.T) {
+	y, err := NewYaml(data)
+	if err != nil {
+		t.Fatal("init yaml failed")
+	}
+	v, err := y.Get(0).String()
+	if err != nil {
+		t.Fatal("get yaml failed")
+	}
+
+	t.Log(v)
+	if v != "IntKey" {
+		t.Fatal("match IntKey failed")
+	}
+}
+
 func TestFloat(t *testing.T) {
 	y, err := NewYaml(data)
 	if err != nil {
@@ -156,3 +173,4 @@ func TestMap(t *testing.T) {
 		t.Fatal("fail to check number of keys")
 	}
 }
+