GitHub #8: Seed indexes from index.yml

service/datastore/index_test.go

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // adapted from github.com/golang/appengine/datastore
 
 package datastore
 
 import (
+        "fmt"
         "strings"
         "testing"
 
         . "github.com/smartystreets/goconvey/convey"
+        "gopkg.in/yaml.v2"
 )
 
 var indexDefinitionTests = []struct {
         id       *IndexDefinition
         builtin  bool
         compound bool
         str      string
         yaml     []string
 }{
         {
@@ skipping 49 matching lines @@
                 for _, tc := range indexDefinitionTests {
                         Convey(tc.str, func() {
                                 So(tc.id.String(), ShouldEqual, tc.str)
                                 So(tc.id.Builtin(), ShouldEqual, tc.builtin)
                                 So(tc.id.Compound(), ShouldEqual, tc.compound)
                                 yaml, _ := tc.id.YAMLString()
                                 So(yaml, ShouldEqual, strings.Join(tc.yaml, "\n"))
                         })
                 }
         })
+
+        Convey("Test MarshalYAML/UnmarshalYAML", t, func() {
+                for _, tc := range indexDefinitionTests {
+                        Convey(fmt.Sprintf("marshallable index definition `%s` is marshalled and unmarshalled correctly", tc.str), func() {
+                                if tc.yaml != nil {
+                                        // marshal
+                                        _, err := yaml.Marshal(tc.id)
+                                        So(err, ShouldBeNil)
+
+                                        // unmarshal
+                                        var ids []*IndexDefinition
+                                        yaml.Unmarshal([]byte(strings.Join(tc.yaml, "\n")), &ids)
+                                        So(ids[0], ShouldResemble, tc.id)
+                                }
+                        })
+                }
+        })
 }