mirror of
https://github.com/Luzifer/nginx-sso.git
synced 2024-10-18 15:44:21 +00:00
115182e445
* Define configuration for LDAP module * Implement LDAP auth * Vendor new dependencies * Add documentation for LDAP provider
97 lines
3.6 KiB
Go
97 lines
3.6 KiB
Go
package ldap
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"gopkg.in/asn1-ber.v1"
|
|
)
|
|
|
|
func TestControlPaging(t *testing.T) {
|
|
runControlTest(t, NewControlPaging(0))
|
|
runControlTest(t, NewControlPaging(100))
|
|
}
|
|
|
|
func TestControlManageDsaIT(t *testing.T) {
|
|
runControlTest(t, NewControlManageDsaIT(true))
|
|
runControlTest(t, NewControlManageDsaIT(false))
|
|
}
|
|
|
|
func TestControlString(t *testing.T) {
|
|
runControlTest(t, NewControlString("x", true, "y"))
|
|
runControlTest(t, NewControlString("x", true, ""))
|
|
runControlTest(t, NewControlString("x", false, "y"))
|
|
runControlTest(t, NewControlString("x", false, ""))
|
|
}
|
|
|
|
func runControlTest(t *testing.T, originalControl Control) {
|
|
header := ""
|
|
if callerpc, _, line, ok := runtime.Caller(1); ok {
|
|
if caller := runtime.FuncForPC(callerpc); caller != nil {
|
|
header = fmt.Sprintf("%s:%d: ", caller.Name(), line)
|
|
}
|
|
}
|
|
|
|
encodedPacket := originalControl.Encode()
|
|
encodedBytes := encodedPacket.Bytes()
|
|
|
|
// Decode directly from the encoded packet (ensures Value is correct)
|
|
fromPacket := DecodeControl(encodedPacket)
|
|
if !bytes.Equal(encodedBytes, fromPacket.Encode().Bytes()) {
|
|
t.Errorf("%sround-trip from encoded packet failed", header)
|
|
}
|
|
if reflect.TypeOf(originalControl) != reflect.TypeOf(fromPacket) {
|
|
t.Errorf("%sgot different type decoding from encoded packet: %T vs %T", header, fromPacket, originalControl)
|
|
}
|
|
|
|
// Decode from the wire bytes (ensures ber-encoding is correct)
|
|
fromBytes := DecodeControl(ber.DecodePacket(encodedBytes))
|
|
if !bytes.Equal(encodedBytes, fromBytes.Encode().Bytes()) {
|
|
t.Errorf("%sround-trip from encoded bytes failed", header)
|
|
}
|
|
if reflect.TypeOf(originalControl) != reflect.TypeOf(fromPacket) {
|
|
t.Errorf("%sgot different type decoding from encoded bytes: %T vs %T", header, fromBytes, originalControl)
|
|
}
|
|
}
|
|
|
|
func TestDescribeControlManageDsaIT(t *testing.T) {
|
|
runAddControlDescriptions(t, NewControlManageDsaIT(false), "Control Type (Manage DSA IT)")
|
|
runAddControlDescriptions(t, NewControlManageDsaIT(true), "Control Type (Manage DSA IT)", "Criticality")
|
|
}
|
|
|
|
func TestDescribeControlPaging(t *testing.T) {
|
|
runAddControlDescriptions(t, NewControlPaging(100), "Control Type (Paging)", "Control Value (Paging)")
|
|
runAddControlDescriptions(t, NewControlPaging(0), "Control Type (Paging)", "Control Value (Paging)")
|
|
}
|
|
|
|
func TestDescribeControlString(t *testing.T) {
|
|
runAddControlDescriptions(t, NewControlString("x", true, "y"), "Control Type ()", "Criticality", "Control Value")
|
|
runAddControlDescriptions(t, NewControlString("x", true, ""), "Control Type ()", "Criticality", "Control Value")
|
|
runAddControlDescriptions(t, NewControlString("x", false, "y"), "Control Type ()", "Control Value")
|
|
runAddControlDescriptions(t, NewControlString("x", false, ""), "Control Type ()", "Control Value")
|
|
}
|
|
|
|
func runAddControlDescriptions(t *testing.T, originalControl Control, childDescriptions ...string) {
|
|
header := ""
|
|
if callerpc, _, line, ok := runtime.Caller(1); ok {
|
|
if caller := runtime.FuncForPC(callerpc); caller != nil {
|
|
header = fmt.Sprintf("%s:%d: ", caller.Name(), line)
|
|
}
|
|
}
|
|
|
|
encodedControls := encodeControls([]Control{originalControl})
|
|
addControlDescriptions(encodedControls)
|
|
encodedPacket := encodedControls.Children[0]
|
|
if len(encodedPacket.Children) != len(childDescriptions) {
|
|
t.Errorf("%sinvalid number of children: %d != %d", header, len(encodedPacket.Children), len(childDescriptions))
|
|
}
|
|
for i, desc := range childDescriptions {
|
|
if encodedPacket.Children[i].Description != desc {
|
|
t.Errorf("%sdescription not as expected: %s != %s", header, encodedPacket.Children[i].Description, desc)
|
|
}
|
|
}
|
|
|
|
}
|