Refactor tests
* Get rid of spurious model.Route{} var, by adding content
directly into model.Handler{} literal.
* Adjust spacing to adhere to the prepare-call-test test structure.
This commit is contained in:
@@ -34,146 +34,120 @@ func locateJailLover() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnRetursErrorWhenEntrypointIsBad(t *testing.T) {
|
func TestSpawnRetursErrorWhenEntrypointIsBad(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: "/bin/this_executable_is_not_likely_to_exist",
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: "/bin/this_executable_is_not_likely_to_exist",
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
err := Spawn(h, nil)
|
err := Spawn(h, nil)
|
||||||
|
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Error("Bad executable not reported")
|
t.Error("Bad executable not reported")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnReturnsNilWhenEntrypointIsGood(t *testing.T) {
|
func TestSpawnReturnsNilWhenEntrypointIsGood(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover(),
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover(),
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
err := Spawn(h, nil)
|
err := Spawn(h, nil)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error("Good executable reported")
|
t.Error("Good executable reported")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnWritesToStdout(t *testing.T) {
|
func TestSpawnWritesToStdout(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover(),
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover(),
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if jldata.Cmdline[0] != locateJailLover() {
|
if jldata.Cmdline[0] != locateJailLover() {
|
||||||
t.Error("Ouput does not match jaillover's")
|
t.Error("Ouput does not match jaillover's")
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnSetsKapowURLEnvVar(t *testing.T) {
|
func TestSpawnSetsKapowURLEnvVar(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover(),
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover(),
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if v, ok := jldata.Env["KAPOW_URL"]; !ok || v != "http://localhost:8081" {
|
if v, ok := jldata.Env["KAPOW_URL"]; !ok || v != "http://localhost:8081" {
|
||||||
t.Error("KAPOW_URL is not set properly")
|
t.Error("KAPOW_URL is not set properly")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnSetsKapowHandlerIDEnvVar(t *testing.T) {
|
func TestSpawnSetsKapowHandlerIDEnvVar(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover(),
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
ID: "HANDLER_ID_FOO",
|
ID: "HANDLER_ID_FOO",
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover(),
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if v, ok := jldata.Env["KAPOW_HANDLER_ID"]; !ok || v != "HANDLER_ID_FOO" {
|
if v, ok := jldata.Env["KAPOW_HANDLER_ID"]; !ok || v != "HANDLER_ID_FOO" {
|
||||||
t.Error("KAPOW_HANDLER_ID is not set properly")
|
t.Error("KAPOW_HANDLER_ID is not set properly")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnRunsOKEntrypointsWithAParam(t *testing.T) {
|
func TestSpawnRunsOKEntrypointsWithAParam(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover() + " -foo",
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover() + " -foo",
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "-foo"}) {
|
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "-foo"}) {
|
||||||
t.Error("Args not as expected")
|
t.Error("Args not as expected")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnRunsOKEntrypointWithArgWithSpace(t *testing.T) {
|
func TestSpawnRunsOKEntrypointWithArgWithSpace(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover() + ` "foo bar"`,
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover() + ` "foo bar"`,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "foo bar"}) {
|
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "foo bar"}) {
|
||||||
t.Error("Args not parsed as expected")
|
t.Error("Args not parsed as expected")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnErrorsWhenEntrypointIsInvalidShell(t *testing.T) {
|
func TestSpawnErrorsWhenEntrypointIsInvalidShell(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover() + ` "`,
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover() + ` "`,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
err := Spawn(h, out)
|
err := Spawn(h, out)
|
||||||
@@ -184,50 +158,41 @@ func TestSpawnErrorsWhenEntrypointIsInvalidShell(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnRunsOKEntrypointWithMultipleArgs(t *testing.T) {
|
func TestSpawnRunsOKEntrypointWithMultipleArgs(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover() + " foo bar",
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover() + " foo bar",
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "foo", "bar"}) {
|
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "foo", "bar"}) {
|
||||||
t.Error("Args not parsed as expected")
|
t.Error("Args not parsed as expected")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnRunsOKEntrypointAndCommand(t *testing.T) {
|
func TestSpawnRunsOKEntrypointAndCommand(t *testing.T) {
|
||||||
r := model.Route{
|
|
||||||
Entrypoint: locateJailLover() + " foo bar",
|
|
||||||
Command: "baz qux",
|
|
||||||
}
|
|
||||||
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{
|
||||||
|
Entrypoint: locateJailLover() + " foo bar",
|
||||||
|
Command: "baz qux",
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
out := &bytes.Buffer{}
|
out := &bytes.Buffer{}
|
||||||
|
|
||||||
_ = Spawn(h, out)
|
_ = Spawn(h, out)
|
||||||
|
|
||||||
jldata := decodeJailLover(out.Bytes())
|
jldata := decodeJailLover(out.Bytes())
|
||||||
|
|
||||||
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "foo", "bar", "baz qux"}) {
|
if !reflect.DeepEqual(jldata.Cmdline, []string{locateJailLover(), "foo", "bar", "baz qux"}) {
|
||||||
t.Error("Malformed cmdline")
|
t.Error("Malformed cmdline")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSpawnReturnsErrorIfEntrypointNotSet(t *testing.T) {
|
func TestSpawnReturnsErrorIfEntrypointNotSet(t *testing.T) {
|
||||||
r := model.Route{}
|
|
||||||
h := &model.Handler{
|
h := &model.Handler{
|
||||||
Route: r,
|
Route: model.Route{},
|
||||||
}
|
}
|
||||||
|
|
||||||
err := Spawn(h, nil)
|
err := Spawn(h, nil)
|
||||||
|
|||||||
Reference in New Issue
Block a user