third-party-mirrors/ssh-chat
1
0
Fork 0
mirror of https://github.com/shazow/ssh-chat.git synced 2025-04-19 18:27:41 +03:00
Code Issues Projects Releases Wiki Activity

hopefully make test less racy

Browse Source
This commit is contained in:
mik2k2 2021-12-22 16:11:52 +01:00
parent 20720ae2c4
commit dbc0bdbeac
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
host_test.go
View File

@ -238,13 +238,17 @@ func TestHostAllowlistCommand(t *testing.T) {
scanner.Scan() // Joined
scanner.Scan()
assertLineEq := func(expected string) {
assertLineEq := func(expected ...string) {
if !scanner.Scan() {
t.Error("no line available")
}
if actual := stripPrompt(scanner.Text()); actual != expected {
t.Errorf("expected %q, got %q", expected, actual)
actual := stripPrompt(scanner.Text())
for _, exp := range expected {
if exp == actual {
return
}
}
t.Errorf("expected %#v, got %q", expected, actual)
}
sendCmd := func(cmd string, formatting ...interface{}) {
host.HandleMsg(message.ParseInput(fmt.Sprintf(cmd, formatting...), m.User))
@ -321,9 +325,9 @@ func TestHostAllowlistCommand(t *testing.T) {
assertLineEq("allowlist is disabled, so nobody will be kicked\r")
sendCmd("/allowlist on")
sendCmd("/allowlist reverify")
assertLineEq(" * [bar] were kicked during pubkey reverification.\r", " * bar left. (After 0 seconds)\r")
assertLineEq(" * [bar] were kicked during pubkey reverification.\r", " * bar left. (After 0 seconds)\r")
kickSignal <- struct{}{}
assertLineEq(" * [bar] were kicked during pubkey reverification.\r")
assertLineEq(" * bar left. (After 0 seconds)\r")
sendCmd("/allowlist add " + testKey)
sendCmd("/allowlist status")