2f4e5edc13
NTLM tests added NTLM licensing info added Avoid logging NTLM messages as it may contain sensitive information Renamed database authentication to NTLM as requested by bolkedebruin (see PR #109)
44 lines
966 B
Go
44 lines
966 B
Go
package database
|
|
|
|
import (
|
|
"github.com/bolkedebruin/rdpgw/cmd/auth/config"
|
|
"testing"
|
|
)
|
|
|
|
func createTestDatabase () (Database) {
|
|
var users = []config.UserConfig{}
|
|
|
|
user1 := config.UserConfig{}
|
|
user1.Username = "my_username"
|
|
user1.Password = "my_password"
|
|
users = append(users, user1)
|
|
|
|
user2 := config.UserConfig{}
|
|
user2.Username = "my_username2"
|
|
user2.Password = "my_password2"
|
|
users = append(users, user2)
|
|
|
|
config := NewConfig(users)
|
|
|
|
return config
|
|
}
|
|
|
|
func TestDatabaseConfigValidUsername(t *testing.T) {
|
|
database := createTestDatabase()
|
|
|
|
if database.GetPassword("my_username") != "my_password" {
|
|
t.Fatalf("Wrong password returned")
|
|
}
|
|
if database.GetPassword("my_username2") != "my_password2" {
|
|
t.Fatalf("Wrong password returned")
|
|
}
|
|
}
|
|
|
|
func TestDatabaseInvalidUsername(t *testing.T) {
|
|
database := createTestDatabase()
|
|
|
|
if database.GetPassword("my_invalid_username") != "" {
|
|
t.Fatalf("Non empty password returned for invalid username")
|
|
}
|
|
}
|