mirror of
https://github.com/jimeh/ozu.io.git
synced 2026-02-19 08:06:39 +00:00
141 lines
2.9 KiB
Go
141 lines
2.9 KiB
Go
package shortner
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/jimeh/ozu.io/shortner/mocks"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
// Mocks
|
|
|
|
//go:generate mockery -name Store -dir .. -recursive
|
|
|
|
// Suite Setup
|
|
|
|
type ShortnerSuite struct {
|
|
suite.Suite
|
|
store *mocks.Store
|
|
shortner *Shortner
|
|
errNotFound error
|
|
}
|
|
|
|
func (s *ShortnerSuite) SetupTest() {
|
|
s.store = new(mocks.Store)
|
|
s.shortner = New(s.store)
|
|
s.errNotFound = errors.New("not found")
|
|
}
|
|
|
|
// Tests
|
|
|
|
func (s *ShortnerSuite) TestShortenExisting() {
|
|
rawURL := []byte("http://google.com/")
|
|
uid := []byte("ig")
|
|
|
|
s.store.On("Get", append([]byte("url:"), rawURL...)).Return(uid, nil)
|
|
|
|
resultUID, resultURL, err := s.shortner.Shorten(rawURL)
|
|
s.NoError(err)
|
|
s.Equal(uid, resultUID)
|
|
s.Equal(rawURL, resultURL)
|
|
s.store.AssertExpectations(s.T())
|
|
}
|
|
|
|
func (s *ShortnerSuite) TestShortenNew() {
|
|
rawURL := []byte("https://google.com")
|
|
url := []byte("https://google.com/")
|
|
uid := []byte("ig")
|
|
|
|
s.store.On("Get", append([]byte("url:"), url...)).Return(nil, s.errNotFound)
|
|
s.store.On("NextSequence").Return(1001, nil)
|
|
s.store.On("Set", append([]byte("url:"), url...), uid).Return(nil)
|
|
s.store.On("Set", append([]byte("uid:"), uid...), url).Return(nil)
|
|
|
|
rUID, rURL, err := s.shortner.Shorten(rawURL)
|
|
|
|
s.NoError(err)
|
|
s.Equal(uid, rUID)
|
|
s.Equal(url, rURL)
|
|
s.store.AssertExpectations(s.T())
|
|
}
|
|
|
|
func (s *ShortnerSuite) TestShortenInvalidURL() {
|
|
examples := []struct {
|
|
url string
|
|
error string
|
|
}{
|
|
{
|
|
url: "*$)]+_<?)",
|
|
error: "invalid URL",
|
|
},
|
|
{
|
|
url: "",
|
|
error: "invalid URL",
|
|
},
|
|
{
|
|
url: "file:///bin/bash",
|
|
error: "schema 'file://' not allowed",
|
|
},
|
|
{
|
|
url: "/users/view.php?uid=138495",
|
|
error: "invalid URL",
|
|
},
|
|
{
|
|
url: "http://long.com/" + strings.Repeat("0", 3000),
|
|
error: "invalid URL",
|
|
},
|
|
}
|
|
|
|
for _, e := range examples {
|
|
rUID, rURL, err := s.shortner.Shorten([]byte(e.url))
|
|
s.Nil(rUID)
|
|
s.Nil(rURL)
|
|
s.EqualError(err, e.error)
|
|
}
|
|
}
|
|
|
|
func (s *ShortnerSuite) TestShortenStoreError() {
|
|
url := []byte("https://google.com/")
|
|
storeErr := errors.New("leveldb: something wrong")
|
|
|
|
s.store.On("Get", append([]byte("url:"), url...)).Return(nil, storeErr)
|
|
|
|
rUID, rURL, err := s.shortner.Shorten(url)
|
|
s.Nil(rUID)
|
|
s.Nil(rURL)
|
|
s.EqualError(err, storeErr.Error())
|
|
}
|
|
|
|
func (s *ShortnerSuite) TestLookupExisting() {
|
|
url := []byte("https://google.com/")
|
|
uid := []byte("ig")
|
|
|
|
s.store.On("Get", append([]byte("uid:"), uid...)).Return(url, nil)
|
|
|
|
rURL, err := s.shortner.Lookup(uid)
|
|
|
|
s.NoError(err)
|
|
s.Equal(url, rURL)
|
|
s.store.AssertExpectations(s.T())
|
|
}
|
|
|
|
func (s *ShortnerSuite) TestLookupNonExistant() {
|
|
uid := []byte("ig")
|
|
|
|
s.store.On("Get", append([]byte("uid:"), uid...)).Return(nil, s.errNotFound)
|
|
|
|
rURL, err := s.shortner.Lookup(uid)
|
|
|
|
s.EqualError(err, "not found")
|
|
s.Nil(rURL)
|
|
s.store.AssertExpectations(s.T())
|
|
}
|
|
|
|
// Run Suite
|
|
|
|
func TestShortnerSuite(t *testing.T) {
|
|
suite.Run(t, new(ShortnerSuite))
|
|
}
|