package setup import ( "testing" "github.com/mholt/caddy/middleware/inner" ) func TestInternal(t *testing.T) { c := newTestController(`internal /internal`) mid, err := Internal(c) if err != nil { t.Errorf("Expected no errors, got: %v", err) } if mid == nil { t.Fatal("Expected middleware, was nil instead") } handler := mid(emptyNext) myHandler, ok := handler.(inner.Internal) if !ok { t.Fatalf("Expected handler to be type Internal, got: %#v", handler) } if myHandler.Paths[0] != "/internal" { t.Errorf("Expected internal in the list of internal Paths") } if !sameNext(myHandler.Next, emptyNext) { t.Error("'Next' field of handler was not set properly") } } func TestInternalParse(t *testing.T) { tests := []struct { inputInternalPaths string shouldErr bool expectedInternalPaths []string }{ {`internal /internal`, false, []string{"/internal"}}, {`internal /internal1 internal /internal2`, false, []string{"/internal1", "/internal2"}}, } for i, test := range tests { c := newTestController(test.inputInternalPaths) actualInternalPaths, err := internalParse(c) if err == nil && test.shouldErr { t.Errorf("Test %d didn't error, but it should have", i) } else if err != nil && !test.shouldErr { t.Errorf("Test %d errored, but it shouldn't have; got '%v'", i, err) } if len(actualInternalPaths) != len(test.expectedInternalPaths) { t.Fatalf("Test %d expected %d InternalPaths, but got %d", i, len(test.expectedInternalPaths), len(actualInternalPaths)) } for j, actualInternalPath := range actualInternalPaths { if actualInternalPath != test.expectedInternalPaths[j] { t.Fatalf("Test %d expected %dth Internal Path to be %s , but got %s", i, j, test.expectedInternalPaths[j], actualInternalPath) } } } }