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")
	}
}