Skip to content

Commit a405d66

Browse files
committed
Add TicketStore.Delete function
1 parent c10a898 commit a405d66

File tree

3 files changed

+16
-0
lines changed

3 files changed

+16
-0
lines changed

memory_store.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,13 @@ func (s *MemoryStore) Write(id string, ticket *AuthenticationResponse) error {
4040
return nil
4141
}
4242

43+
func (s *MemoryStore) Delete(id string) error {
44+
s.mu.Lock()
45+
delete(s.store, id)
46+
s.mu.Unlock()
47+
return nil
48+
}
49+
4350
func (s *MemoryStore) Clear() error {
4451
s.mu.Lock()
4552
s.store = nil

memory_store_test.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,14 @@ func TestMemoryStore(t *testing.T) {
2626
t.Errorf("Expected retrieved AuthenticationResponse to be %v, got %v", user2, ar)
2727
}
2828

29+
if err := store.Delete("user2"); err != nil {
30+
t.Errorf("Error while deleting user2, got %v", err)
31+
}
32+
33+
if _, err := store.Read("user2"); err != ErrInvalidTicket {
34+
t.Errorf("Expected store.Read(user2) to fail")
35+
}
36+
2937
if err := store.Clear(); err != nil {
3038
t.Errorf("Expected store.Clear() to succeed, got error: %v", err)
3139
}

ticket_store.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,5 +11,6 @@ var (
1111
type TicketStore interface {
1212
Read(id string) (*AuthenticationResponse, error)
1313
Write(id string, ticket *AuthenticationResponse) error
14+
Delete(id string) error
1415
Clear() error
1516
}

0 commit comments

Comments
 (0)