request_test.go

/*
 * Copyright © 2015-2018 Aeneas Rekkas <aeneas+oss@aeneas.io>
 *
 * 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.
 *
 * @author		Aeneas Rekkas <aeneas+oss@aeneas.io>
 * @copyright 	2015-2018 Aeneas Rekkas <aeneas+oss@aeneas.io>
 * @license 	Apache-2.0
 *
 */

package fosite_test

import (
	"net/url"
	"testing"
	"time"

	. "github.com/ory/fosite"
	"github.com/stretchr/testify/assert"
)

func TestRequest(t *testing.T) {
	r := &Request{
		RequestedAt:   time.Now().UTC(),
		Client:        &DefaultClient{},
		Scopes:        Arguments{},
		GrantedScopes: []string{},
		Form:          url.Values{"foo": []string{"bar"}},
		Session:       new(DefaultSession),
	}

	assert.Equal(t, r.RequestedAt, r.GetRequestedAt())
	assert.Equal(t, r.Client, r.GetClient())
	assert.Equal(t, r.GrantedScopes, r.GetGrantedScopes())
	assert.Equal(t, r.Scopes, r.GetRequestedScopes())
	assert.Equal(t, r.Form, r.GetRequestForm())
	assert.Equal(t, r.Session, r.GetSession())
}

func TestMergeRequest(t *testing.T) {
	a := &Request{
		RequestedAt:   time.Now().UTC(),
		Client:        &DefaultClient{ID: "123"},
		Scopes:        Arguments{"asdff"},
		GrantedScopes: []string{"asdf"},
		Form:          url.Values{"foo": []string{"fasdf"}},
		Session:       new(DefaultSession),
	}
	b := &Request{
		RequestedAt:   time.Now().UTC(),
		Client:        &DefaultClient{},
		Scopes:        Arguments{},
		GrantedScopes: []string{},
		Form:          url.Values{},
		Session:       new(DefaultSession),
	}

	b.Merge(a)
	assert.EqualValues(t, a.RequestedAt, b.RequestedAt)
	assert.EqualValues(t, a.Client, b.Client)
	assert.EqualValues(t, a.Scopes, b.Scopes)
	assert.EqualValues(t, a.GrantedScopes, b.GrantedScopes)
	assert.EqualValues(t, a.Form, b.Form)
	assert.EqualValues(t, a.Session, b.Session)
}

func TestSanitizeRequest(t *testing.T) {
	a := &Request{
		RequestedAt:   time.Now().UTC(),
		Client:        &DefaultClient{ID: "123"},
		Scopes:        Arguments{"asdff"},
		GrantedScopes: []string{"asdf"},
		Form: url.Values{
			"foo": []string{"fasdf"},
			"bar": []string{"fasdf", "fasdf"},
			"baz": []string{"fasdf"},
		},
		Session: new(DefaultSession),
	}

	b := a.Sanitize([]string{"bar", "baz"})
	assert.NotEqual(t, a.Form.Encode(), b.GetRequestForm().Encode())
	assert.Empty(t, b.GetRequestForm().Get("foo"))
	assert.Equal(t, "fasdf", b.GetRequestForm().Get("bar"))
	assert.Equal(t, "fasdf", b.GetRequestForm().Get("baz"))

	assert.Equal(t, "fasdf", a.GetRequestForm().Get("bar"))
	assert.Equal(t, "fasdf", a.GetRequestForm().Get("baz"))
	assert.Equal(t, "fasdf", a.GetRequestForm().Get("foo"))
}

func TestIdentifyRequest(t *testing.T) {
	a := &Request{
		RequestedAt:   time.Now().UTC(),
		Client:        &DefaultClient{},
		Scopes:        Arguments{},
		GrantedScopes: []string{},
		Form:          url.Values{"foo": []string{"bar"}},
		Session:       new(DefaultSession),
	}

	b := a.Sanitize([]string{})
	b.GetID()
	assert.Equal(t, a.ID, b.GetID())
}