engineering/sshpiper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix build

Browse source
This commit is contained in:
Boshi Lian 2019-01-09 04:03:33 -08:00
parent 374f474c10
commit b4113651af
2 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
sshpiperd/upstream/database/handler_test.go
View file

@ -131,7 +131,7 @@ func TestFindUpstream(t *testing.T) {
createEntry(t, db, "finddown1", "findup1", listener.Addr().String(), false)
createEntry(t, db, "finddown2", "findup2", listener.Addr().String(), false)
_, auth, err := h(testconn{"finddown0"})
_, auth, err := h(testconn{"finddown0"}, nil)
if err != nil {
t.Fatal(err)
}
@ -141,13 +141,13 @@ func TestFindUpstream(t *testing.T) {
}
db.Delete(&config{Entry: fallbackUserEntry})
_, _, err = h(testconn{"not_exists"})
_, _, err = h(testconn{"not_exists"}, nil)
if err == nil {
t.Error("should not found any user")
}
db.NewRecord(&config{Entry: fallbackUserEntry, Value: "findup0"})
_, auth, err = h(testconn{"finddown0"})
_, auth, err = h(testconn{"finddown0"}, nil)
if err != nil {
t.Fatal(err)
}
@ -168,7 +168,7 @@ func TestPublicKeyCallback(t *testing.T) {
pub, _ := createEntry(t, db, "pkdown", "pkdown", listener.Addr().String(), false)
_, auth, err := h(testconn{"pkdown"})
_, auth, err := h(testconn{"pkdown"}, nil)
if err != nil {
t.Fatal(err)
}

8
sshpiperd/upstream/workingdir/workingdir_test.go
View file

@ -243,7 +243,7 @@ func TestFindUpstreamFromUserfile(t *testing.T) {
}
t.Logf("testing file too open")
_, _, err = findUpstreamFromUserfile(stubConnMetadata{user})
_, _, err = findUpstreamFromUserfile(stubConnMetadata{user}, nil)
if err == nil {
t.Fatalf("should return err when file too open")
}
@ -254,7 +254,7 @@ func TestFindUpstreamFromUserfile(t *testing.T) {
}
t.Logf("testing conn dial to %v", addr)
conn, _, err := findUpstreamFromUserfile(stubConnMetadata{user})
conn, _, err := findUpstreamFromUserfile(stubConnMetadata{user}, nil)
if err != nil {
t.Fatalf("findUpstreamFromUserfile failed %v", err)
}
@ -276,14 +276,14 @@ func TestFindUpstreamFromUserfile(t *testing.T) {
t.Logf("testing user not found")
config.FallbackUsername = ""
_, _, err = findUpstreamFromUserfile(stubConnMetadata{"nosuchuser"})
_, _, err = findUpstreamFromUserfile(stubConnMetadata{"nosuchuser"}, nil)
if err == nil {
t.Fatalf("should return err when finding nosuchuser")
}
t.Logf("testing user not found fallback")
config.FallbackUsername = user
_, _, err = findUpstreamFromUserfile(stubConnMetadata{"nosuchuser"})
_, _, err = findUpstreamFromUserfile(stubConnMetadata{"nosuchuser"}, nil)
if err != nil {
t.Fatalf("should return fallbackuser")