auth

Paddy 2015-01-10 Parent:c03b5eb3179e Child:e000b1c24fc0

112:da2a0954e8d3 Go to Latest

auth/session_test.go

Flesh out auth code grant (in)validation. Flesh out our tests for functions that do the validation and invalidation of the authorization code grant type's authorization codes. Basically, make sure that the auth code's are being checked right and that marking them as used after they're used works.

Download raw file
View source Diff to previous Annotate
History
paddy@77 1 package auth
paddy@77 2
paddy@77 3 import (
paddy@77 4 "testing"
paddy@77 5 "time"
paddy@77 6
paddy@107 7 "code.secondbit.org/uuid.hg"
paddy@77 8 )
paddy@77 9
paddy@77 10 var sessionStores = []sessionStore{NewMemstore()}
paddy@77 11
paddy@77 12 func compareSessions(session1, session2 Session) (success bool, field string, val1, val2 interface{}) {
paddy@77 13 if session1.ID != session2.ID {
paddy@77 14 return false, "ID", session1.ID, session2.ID
paddy@77 15 }
paddy@77 16 if session1.IP != session2.IP {
paddy@77 17 return false, "IP", session1.IP, session2.IP
paddy@77 18 }
paddy@77 19 if session1.UserAgent != session2.UserAgent {
paddy@77 20 return false, "UserAgent", session1.UserAgent, session2.UserAgent
paddy@77 21 }
paddy@77 22 if !session1.ProfileID.Equal(session2.ProfileID) {
paddy@77 23 return false, "ProfileID", session1.ProfileID, session2.ProfileID
paddy@77 24 }
paddy@77 25 if !session1.Created.Equal(session2.Created) {
paddy@77 26 return false, "Created", session1.Created, session2.Created
paddy@77 27 }
paddy@77 28 if session1.Login != session2.Login {
paddy@77 29 return false, "Login", session1.Login, session2.Login
paddy@77 30 }
paddy@77 31 if session1.Active != session2.Active {
paddy@77 32 return false, "Active", session1.Active, session2.Active
paddy@77 33 }
paddy@77 34 return true, "", nil, nil
paddy@77 35 }
paddy@77 36
paddy@77 37 func TestSessionStoreSuccess(t *testing.T) {
paddy@77 38 t.Parallel()
paddy@77 39 session := Session{
paddy@77 40 ID: uuid.NewID().String() + uuid.NewID().String(),
paddy@77 41 IP: "127.0.0.1",
paddy@77 42 UserAgent: "TestRunner",
paddy@77 43 ProfileID: uuid.NewID(),
paddy@77 44 Created: time.Now(),
paddy@77 45 Login: "test@example.com",
paddy@77 46 Active: true,
paddy@77 47 }
paddy@77 48 for _, store := range sessionStores {
paddy@77 49 err := store.createSession(session)
paddy@77 50 if err != nil {
paddy@77 51 t.Errorf("Error saving session to %T: %s", store, err)
paddy@77 52 }
paddy@77 53 err = store.createSession(session)
paddy@77 54 if err != ErrSessionAlreadyExists {
paddy@77 55 t.Errorf("Expected ErrSessionAlreadyExists from %T, got %s", store, err)
paddy@77 56 }
paddy@77 57 retrieved, err := store.getSession(session.ID)
paddy@77 58 if err != nil {
paddy@77 59 t.Errorf("Error retrieving session from %T: %s", store, err)
paddy@77 60 }
paddy@77 61 success, field, expectation, result := compareSessions(session, retrieved)
paddy@77 62 if !success {
paddy@77 63 t.Errorf("Expected field %s to be %v, but got %v from %T", field, expectation, result, store)
paddy@77 64 }
paddy@77 65 retrievedList, err := store.listSessions(session.ProfileID, time.Time{}, 10)
paddy@77 66 if err != nil {
paddy@77 67 t.Errorf("Error retrieving sessions by profile from %T: %s", store, err)
paddy@77 68 }
paddy@77 69 if len(retrievedList) != 1 {
paddy@77 70 t.Errorf("Expected 1 session retrieved by profile from %T, got %d", store, len(retrievedList))
paddy@77 71 }
paddy@77 72 success, field, expectation, result = compareSessions(session, retrievedList[0])
paddy@77 73 if !success {
paddy@77 74 t.Errorf("Expected field %s to be %v, but got %v from %T", field, expectation, result, store)
paddy@77 75 }
paddy@77 76 err = store.removeSession(session.ID)
paddy@77 77 if err != nil {
paddy@77 78 t.Errorf("Error removing session from %T: %s", store, err)
paddy@77 79 }
paddy@77 80 retrieved, err = store.getSession(session.ID)
paddy@77 81 if err != ErrSessionNotFound {
paddy@77 82 t.Errorf("Expected ErrSessionNotFound from %T, got %s", store, err)
paddy@77 83 }
paddy@77 84 retrievedList, err = store.listSessions(session.ProfileID, time.Time{}, 10)
paddy@77 85 if err != nil {
paddy@77 86 t.Errorf("Error retrieving sessions by profile from %T: %s", store, err)
paddy@77 87 }
paddy@77 88 if len(retrievedList) != 0 {
paddy@77 89 t.Errorf("Expected 0 sessions retrieved by profile from %T, got %d", store, len(retrievedList))
paddy@77 90 }
paddy@77 91 err = store.removeSession(session.ID)
paddy@77 92 if err != ErrSessionNotFound {
paddy@77 93 t.Errorf("Expected ErrSessionNotFound from %T, got %s", store, err)
paddy@77 94 }
paddy@77 95 }
paddy@77 96 }