147 lines
2.9 KiB
Go
147 lines
2.9 KiB
Go
package ingress
|
|
|
|
import (
|
|
"io"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_rule_matches(t *testing.T) {
|
|
type fields struct {
|
|
Hostname string
|
|
Path *regexp.Regexp
|
|
Service OriginService
|
|
}
|
|
type args struct {
|
|
requestURL *url.URL
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
want bool
|
|
}{
|
|
{
|
|
name: "Just hostname, pass",
|
|
fields: fields{
|
|
Hostname: "example.com",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://example.com"),
|
|
},
|
|
want: true,
|
|
},
|
|
{
|
|
name: "Entire hostname is wildcard, should match everything",
|
|
fields: fields{
|
|
Hostname: "*",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://example.com"),
|
|
},
|
|
want: true,
|
|
},
|
|
{
|
|
name: "Just hostname, fail",
|
|
fields: fields{
|
|
Hostname: "example.com",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://foo.bar"),
|
|
},
|
|
want: false,
|
|
},
|
|
{
|
|
name: "Just wildcard hostname, pass",
|
|
fields: fields{
|
|
Hostname: "*.example.com",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://adam.example.com"),
|
|
},
|
|
want: true,
|
|
},
|
|
{
|
|
name: "Just wildcard hostname, fail",
|
|
fields: fields{
|
|
Hostname: "*.example.com",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://tunnel.com"),
|
|
},
|
|
want: false,
|
|
},
|
|
{
|
|
name: "Just wildcard outside of subdomain in hostname, fail",
|
|
fields: fields{
|
|
Hostname: "*example.com",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://www.example.com"),
|
|
},
|
|
want: false,
|
|
},
|
|
{
|
|
name: "Wildcard over multiple subdomains",
|
|
fields: fields{
|
|
Hostname: "*.example.com",
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://adam.chalmers.example.com"),
|
|
},
|
|
want: true,
|
|
},
|
|
{
|
|
name: "Hostname and path",
|
|
fields: fields{
|
|
Hostname: "*.example.com",
|
|
Path: regexp.MustCompile("/static/.*\\.html"),
|
|
},
|
|
args: args{
|
|
requestURL: MustParseURL(t, "https://www.example.com/static/index.html"),
|
|
},
|
|
want: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
r := Rule{
|
|
Hostname: tt.fields.Hostname,
|
|
Path: tt.fields.Path,
|
|
Service: tt.fields.Service,
|
|
}
|
|
u := tt.args.requestURL
|
|
if got := r.Matches(u.Hostname(), u.Path); got != tt.want {
|
|
t.Errorf("rule.matches() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestStaticHTTPStatus(t *testing.T) {
|
|
o := newStatusCode(404)
|
|
buf := make([]byte, 100)
|
|
|
|
sendReq := func() {
|
|
resp, err := o.RoundTrip(nil)
|
|
require.NoError(t, err)
|
|
_, err = resp.Body.Read(buf)
|
|
require.Equal(t, io.EOF, err)
|
|
require.NoError(t, resp.Body.Close())
|
|
require.Equal(t, 404, resp.StatusCode)
|
|
|
|
resp, err = o.RoundTrip(nil)
|
|
require.NoError(t, err)
|
|
w := httptest.NewRecorder()
|
|
n, err := io.Copy(w, resp.Body)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int64(0), n)
|
|
}
|
|
sendReq()
|
|
sendReq()
|
|
}
|