forked from ortuman/jackal
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathauth_test.go
49 lines (38 loc) · 1.6 KB
/
auth_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
/*
* Copyright (c) 2018 Miguel Ángel Ortuño.
* See the LICENSE file for more information.
*/
package server
import (
"testing"
"github.com/ortuman/jackal/storage"
"github.com/ortuman/jackal/storage/model"
"github.com/ortuman/jackal/stream/c2s"
"github.com/ortuman/jackal/xml"
"github.com/pborman/uuid"
"github.com/stretchr/testify/require"
)
func authTestSetup(user *model.User) *c2s.MockStream {
storage.Initialize(&storage.Config{Type: storage.Mock})
storage.Instance().InsertOrUpdateUser(user)
jid, _ := xml.NewJID("mariana", "localhost", "res", true)
testStrm := c2s.NewMockStream(uuid.New(), jid)
testStrm.SetUsername("mariana")
testStrm.SetDomain("localhost")
testStrm.SetResource("res")
testStrm.SetJID(jid)
return testStrm
}
func authTestTeardown() {
storage.Instance().Shutdown()
}
func TestAuthError(t *testing.T) {
require.Equal(t, "incorrect-encoding", errSASLIncorrectEncoding.(*saslErrorString).Error())
require.Equal(t, "malformed-request", errSASLMalformedRequest.(*saslErrorString).Error())
require.Equal(t, "not-authorized", errSASLNotAuthorized.(*saslErrorString).Error())
require.Equal(t, "temporary-auth-failure", errSASLTemporaryAuthFailure.(*saslErrorString).Error())
require.Equal(t, "incorrect-encoding", errSASLIncorrectEncoding.(*saslErrorString).Element().Name())
require.Equal(t, "malformed-request", errSASLMalformedRequest.(*saslErrorString).Element().Name())
require.Equal(t, "not-authorized", errSASLNotAuthorized.(*saslErrorString).Element().Name())
require.Equal(t, "temporary-auth-failure", errSASLTemporaryAuthFailure.(*saslErrorString).Element().Name())
}