22 lines
486 B
Go
22 lines
486 B
Go
|
package main
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestParseModeAcceptsServerAndDeliver(t *testing.T) {
|
||
|
|
for _, raw := range []string{"server", "deliver"} {
|
||
|
|
mode, err := parseMode(raw)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("parseMode(%q) returned error: %v", raw, err)
|
||
|
|
}
|
||
|
|
if string(mode) != raw {
|
||
|
|
t.Fatalf("parseMode(%q) = %q", raw, mode)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseModeRejectsUnknownMode(t *testing.T) {
|
||
|
|
if _, err := parseMode("worker"); err == nil {
|
||
|
|
t.Fatal("parseMode accepted unknown mode")
|
||
|
|
}
|
||
|
|
}
|