Files
matchbox/api/spec_test.go
Dalton Hubble 68437b9e2d api: Allow Ignition config templates in YAML
* Match machines to their Ignition template file, render the
template, parse as JSON or YAML, and serve as Ignition JSON
* Existing JSON ignition configs should still work
* Remove the Ignition YAML -> JSON transform script which used fuze
2016-02-04 13:45:11 -08:00

59 lines
1.5 KiB
Go

package api
import (
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
)
var (
// testSpec specifies a named group of configs for testing purposes.
testSpec = &Spec{
ID: "g1h2i3j4",
BootConfig: &BootConfig{
Kernel: "/image/kernel",
Initrd: []string{"/image/initrd_a", "/image/initrd_b"},
Cmdline: map[string]interface{}{
"a": "b",
"c": "",
},
},
CloudConfig: "cloud-config.yml",
IgnitionConfig: "ignition.json",
}
testSpecWithIgnitionYAML = &Spec{
ID: "g1h2i3j4",
IgnitionConfig: "ignition.yaml",
}
emptySpec = &Spec{
ID: "empty",
}
expectedSpecJSON = `{"id":"g1h2i3j4","boot":{"kernel":"/image/kernel","initrd":["/image/initrd_a","/image/initrd_b"],"cmdline":{"a":"b","c":""}},"cloud_id":"cloud-config.yml","ignition_id":"ignition.json"}`
)
func TestSpecHandler(t *testing.T) {
store := &fixedStore{
Specs: map[string]*Spec{"g1h2i3j4": testSpec},
}
h := specResource{store: store}
req, _ := http.NewRequest("GET", "/g1h2i3j4", nil)
w := httptest.NewRecorder()
h.ServeHTTP(w, req)
// assert that:
// - spec is rendered as JSON
assert.Equal(t, http.StatusOK, w.Code)
assert.Equal(t, jsonContentType, w.HeaderMap.Get(contentType))
assert.Equal(t, expectedSpecJSON, w.Body.String())
}
func TestSpecHandler_MissingSpec(t *testing.T) {
store := &emptyStore{}
h := specResource{store}
req, _ := http.NewRequest("GET", "/", nil)
w := httptest.NewRecorder()
h.ServeHTTP(w, req)
assert.Equal(t, http.StatusNotFound, w.Code)
}