Files
matchbox/api/spec_test.go
Dalton Hubble ccd3c7a839 api: Associate machines' attribute labels to specs by group matchers
* Use YAML group config to match machine labels to groups of zero or
more machines sharing a Specification
* Discontinue using machine.json resources representations which are
repetitive, don't support groups, and don't allow label matching
* Previous "default" special case now handled by a Group with an empty
matcher
2016-01-07 23:24:05 -08:00

55 lines
1.4 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",
}
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_MissingConfig(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)
}