imageproxy/proxy/data_test.go

// Copyright 2013 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package proxy

import (
	"net/http"
	"reflect"
	"testing"
)

func TestOptions_String(t *testing.T) {
	tests := []struct {
		Options *Options
		String  string
	}{
		{emptyOptions, "0x0"},
		{
			&Options{1, 2, true, 90, true, true},
			"1x2,fit,r90,fv,fh",
		},
	}

	for i, tt := range tests {
		if got, want := tt.Options.String(), tt.String; got != want {
			t.Errorf("%d. Options.String returned %v, want %v", i, got, want)
		}
	}
}

func TestParseOptions(t *testing.T) {
	tests := []struct {
		Input   string
		Options *Options
	}{
		{"", emptyOptions},
		{"x", emptyOptions},
		{"0", emptyOptions},
		{",,,,", emptyOptions},

		// size variations
		{"1x", &Options{Width: 1}},
		{"x1", &Options{Height: 1}},
		{"1x2", &Options{Width: 1, Height: 2}},
		{"0.1x0.2", &Options{Width: 0.1, Height: 0.2}},

		// additional flags
		{"fit", &Options{Fit: true}},
		{"r90", &Options{Rotate: 90}},
		{"fv", &Options{FlipVertical: true}},
		{"fh", &Options{FlipHorizontal: true}},

		// duplicate flags (last one wins)
		{"1x2,3x4", &Options{Width: 3, Height: 4}},
		{"1x2,3", &Options{Width: 3, Height: 3}},
		{"1x2,0x3", &Options{Width: 0, Height: 3}},
		{"1x,x2", &Options{Width: 1, Height: 2}},
		{"r90,r270", &Options{Rotate: 270}},

		// mix of valid and invalid flags
		{"FOO,1,BAR,r90,BAZ", &Options{Width: 1, Height: 1, Rotate: 90}},

		{"1x2,fit,r90,fv,fh", &Options{1, 2, true, 90, true, true}},
		{"r90,fh,1x2,fv,fit", &Options{1, 2, true, 90, true, true}},
	}

	for i, tt := range tests {
		if got, want := ParseOptions(tt.Input), tt.Options; !reflect.DeepEqual(got, want) {
			t.Errorf("%d. ParseOptions returned %#v, want %#v", i, got, want)
		}
	}
}

// Test that request URLs are properly parsed into Options and RemoteURL.  This
// test verifies that invalid remote URLs throw errors, and that valid
// combinations of Options and URL are accept.  This does not exhaustively test
// the various Options that can be specified; see TestParseOptions for that.
func TestNewRequest(t *testing.T) {
	tests := []struct {
		URL         string
		RemoteURL   string
		Options     *Options
		ExpectError bool
	}{
		// invalid URLs
		{
			"http://localhost/", "", nil, true,
		},
		{
			"http://localhost/1/", "", nil, true,
		},
		{
			"http://localhost//example.com/foo", "", nil, true,
		},
		{
			"http://localhost//ftp://example.com/foo", "", nil, true,
		},

		// invalid options.  These won't return errors, but will not fully parse the options
		{
			"http://localhost/s/http://example.com/",
			"http://example.com/", emptyOptions, false,
		},
		{
			"http://localhost/1xs/http://example.com/",
			"http://example.com/", &Options{Width: 1}, false,
		},

		// valid URLs
		{
			"http://localhost/http://example.com/foo",
			"http://example.com/foo", nil, false,
		},
		{
			"http://localhost//http://example.com/foo",
			"http://example.com/foo", emptyOptions, false,
		},
		{
			"http://localhost//https://example.com/foo",
			"https://example.com/foo", emptyOptions, false,
		},
		{
			"http://localhost/1x2/http://example.com/foo",
			"http://example.com/foo", &Options{Width: 1, Height: 2}, false,
		},
		{
			"http://localhost//http://example.com/foo?bar",
			"http://example.com/foo?bar", emptyOptions, false,
		},
	}

	for i, tt := range tests {
		req, err := http.NewRequest("GET", tt.URL, nil)
		if err != nil {
			t.Errorf("%d. Error parsing request: %v", i, err)
			continue
		}

		r, err := NewRequest(req)
		if tt.ExpectError {
			if err == nil {
				t.Errorf("%d. Expected parsing error", i)
			}
			continue
		} else if err != nil {
			t.Errorf("%d. Error parsing request: %v", i, err)
			continue
		}

		if got := r.URL.String(); tt.RemoteURL != got {
			t.Errorf("%d. Request URL = %v, want %v", i, got, tt.RemoteURL)
		}
		if !reflect.DeepEqual(tt.Options, r.Options) {
			t.Errorf("%d. Request Options = %v, want %v", i, r.Options, tt.Options)
		}
	}
}