1
0
mirror of https://github.com/fumiama/go-nd-portal.git synced 2026-06-30 16:10:26 +08:00

feat: auto select server IP when not specified (#4)

This commit is contained in:
chasey-dev
2025-05-21 16:09:57 +08:00
committed by GitHub
parent 81957d1c29
commit 32fdf3ae90
4 changed files with 79 additions and 42 deletions

View File

@@ -12,12 +12,21 @@ import (
"github.com/fumiama/go-nd-portal/helper"
)
func TestGetUserInfo(t *testing.T) {
u, err := NewPortal("2000010101001", "12345678", net.IPv4(1, 2, 3, 4).To4(),"qsh-edu")
func TestAutoSelectServerIP(t *testing.T) {
u, err := NewPortal("2000010101001", "12345678", "", net.IPv4(1, 2, 3, 4).To4(), LoginTypeQshEdu)
if err != nil {
t.Fatal(err)
}
info, err := GetUserInfo(u.name, u.domain, u.pswd, u.ip, u.acid)
t.Log(u.sip)
assert.Equal(t, PortalServerIPQsh, u.sip)
}
func TestGetUserInfo(t *testing.T) {
u, err := NewPortal("2000010101001", "12345678", "", net.IPv4(1, 2, 3, 4).To4(), LoginTypeQshEdu)
if err != nil {
t.Fatal(err)
}
info, err := GetUserInfo(u.name, u.domain, u.pswd, u.cip, u.acid)
if err != nil {
t.Fatal(err)
}
@@ -53,11 +62,11 @@ func TestDecodeKey(t *testing.T) {
}
func TestEncodeUserInfo(t *testing.T) {
u, err := NewPortal("2001010101001", "1234567890", net.IPv4(113, 54, 148, 243).To4(),"qsh-edu")
u, err := NewPortal("2001010101001", "1234567890", "", net.IPv4(113, 54, 148, 243).To4(), LoginTypeQshEdu)
if err != nil {
t.Fatal(err)
}
info, err := GetUserInfo(u.name, u.domain, u.pswd, u.ip, u.acid)
info, err := GetUserInfo(u.name, u.domain, u.pswd, u.cip, u.acid)
if err != nil {
t.Fatal(err)
}
@@ -67,7 +76,7 @@ func TestEncodeUserInfo(t *testing.T) {
}
func TestHMd5(t *testing.T) {
u, err := NewPortal("2001010101001", "1234567890", net.IPv4(113, 54, 148, 243).To4(),"qsh-edu")
u, err := NewPortal("2001010101001", "1234567890", "", net.IPv4(113, 54, 148, 243).To4(), LoginTypeQshEdu)
if err != nil {
t.Fatal(err)
}
@@ -81,11 +90,11 @@ func TestSha1(t *testing.T) {
}
func TestCheckSum(t *testing.T) {
u, err := NewPortal("2001010101001", "1234567890", net.IPv4(113, 54, 148, 243).To4(),"qsh-edu")
u, err := NewPortal("2001010101001", "1234567890", "", net.IPv4(113, 54, 148, 243).To4(), LoginTypeQshEdu)
if err != nil {
t.Fatal(err)
}
info, err := GetUserInfo(u.name, u.domain, u.pswd, u.ip, u.acid)
info, err := GetUserInfo(u.name, u.domain, u.pswd, u.cip, u.acid)
if err != nil {
t.Fatal(err)
}
@@ -97,7 +106,7 @@ func TestCheckSum(t *testing.T) {
PortalDomainQsh,
u.PasswordHMd5(challenge),
u.acid,
u.ip,
u.cip,
EncodeUserInfo(
info,
challenge,