auth
auth/session_test.go
Add support for registering Clients. Add an API endpoint to register Clients, which was the last step necessary before the OAuth2 integration could be tried out.
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 }