Return error if unknown hello session was received while waiting for joined events.
This commit is contained in:
parent
9cf796640c
commit
5d87df4b8e
|
@ -535,11 +535,17 @@ func (c *TestClient) RunUntilJoined(ctx context.Context, hello ...*HelloServerMe
|
||||||
if err := c.checkSingleMessageJoined(message); err != nil {
|
if err := c.checkSingleMessageJoined(message); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
found := false
|
||||||
for idx, h := range hello {
|
for idx, h := range hello {
|
||||||
if err := c.checkMessageJoined(message, h); err == nil {
|
if err := c.checkMessageJoined(message, h); err == nil {
|
||||||
hello = append(hello[:idx], hello[idx+1:]...)
|
hello = append(hello[:idx], hello[idx+1:]...)
|
||||||
|
found = true
|
||||||
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if !found {
|
||||||
|
return fmt.Errorf("expected one of the passed hello sessions, got %+v", message.Event.Join[0])
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|
Loading…
Reference in New Issue