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