Skip to content

Commit

Permalink
Merge pull request #38 from mysteriumnetwork/remove-session-cleaner
Browse files Browse the repository at this point in the history
removed the unused session cleaner from auth middleware
  • Loading branch information
zolia committed Jan 30, 2019
2 parents 73afdde + 0115401 commit 1230974
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 19 deletions.
7 changes: 1 addition & 6 deletions openvpn/middlewares/server/auth/middleware.go
Original file line number Diff line number Diff line change
Expand Up @@ -27,22 +27,17 @@ import (
type middleware struct {
// TODO: consider implementing event channel to communicate required callbacks
credentialsValidator CredentialsValidator
sessionCleaner SessionCleaner
commandWriter management.CommandWriter
currentEvent clientEvent
}

// CredentialsValidator callback checks given auth primitives (i.e. customer identity signature / node's sessionId)
type CredentialsValidator func(clientID int, username, password string) (bool, error)

// SessionCleaner callback cleans up session after client disconnects
type SessionCleaner func(username string) error

// NewMiddleware creates server user_auth challenge authentication middleware
func NewMiddleware(credentialsValidator CredentialsValidator, cleaner SessionCleaner) *middleware {
func NewMiddleware(credentialsValidator CredentialsValidator) *middleware {
return &middleware{
credentialsValidator: credentialsValidator,
sessionCleaner: cleaner,
commandWriter: nil,
currentEvent: undefinedEvent,
}
Expand Down
20 changes: 7 additions & 13 deletions openvpn/middlewares/server/auth/middleware_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -38,15 +38,9 @@ func (f *fakeAuthenticatorStub) fakeAuthenticator(clientID int, username, passwo
return f.authenticated, nil
}

func (f *fakeAuthenticatorStub) fakeSessionCleanup(username string) error {
f.called = true
f.username = username
return nil
}

func Test_Factory(t *testing.T) {
fas := fakeAuthenticatorStub{}
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)
assert.NotNil(t, middleware)
}

Expand All @@ -60,7 +54,7 @@ func Test_ConsumeLineSkips(t *testing.T) {
{">USERNAME"},
}
fas := fakeAuthenticatorStub{}
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)

for _, test := range tests {
consumed, err := middleware.ConsumeLine(test.line)
Expand All @@ -80,7 +74,7 @@ func Test_ConsumeLineTakes(t *testing.T) {
}

fas := fakeAuthenticatorStub{}
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)
mockConnection := &management.MockConnection{}
middleware.Start(mockConnection)

Expand All @@ -101,7 +95,7 @@ func Test_ConsumeLineAuthState(t *testing.T) {

for _, test := range tests {
fas := fakeAuthenticatorStub{}
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)
mockConnection := &management.MockConnection{}
middleware.Start(mockConnection)

Expand All @@ -121,7 +115,7 @@ func Test_ConsumeLineNotAuthState(t *testing.T) {

for _, test := range tests {
fas := fakeAuthenticatorStub{}
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)
mockConnection := &management.MockConnection{}
middleware.Start(mockConnection)

Expand All @@ -143,7 +137,7 @@ func Test_ConsumeLineAuthTrueChecker(t *testing.T) {
}
fas := fakeAuthenticatorStub{}
fas.authenticated = true
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)
mockConnection := &management.MockConnection{}
middleware.Start(mockConnection)

Expand All @@ -169,7 +163,7 @@ func Test_ConsumeLineAuthFalseChecker(t *testing.T) {
}
fas := fakeAuthenticatorStub{}
fas.authenticated = false
middleware := NewMiddleware(fas.fakeAuthenticator, fas.fakeSessionCleanup)
middleware := NewMiddleware(fas.fakeAuthenticator)
mockConnection := &management.MockConnection{}
middleware.Start(mockConnection)

Expand Down

0 comments on commit 1230974

Please sign in to comment.