(t *testing.T, agentID string)
| 127 | } |
| 128 | |
| 129 | func (e *testEnv) waitWSConnected(t *testing.T, agentID string) { |
| 130 | t.Helper() |
| 131 | deadline := time.After(2 * time.Second) |
| 132 | for !e.wsHub.IsConnected(agentID) { |
| 133 | select { |
| 134 | case <-deadline: |
| 135 | t.Fatalf("timed out waiting for agent %s to register in WS hub", agentID) |
| 136 | default: |
| 137 | time.Sleep(10 * time.Millisecond) |
| 138 | } |
| 139 | } |
| 140 | } |
| 141 | |
| 142 | func (e *testEnv) waitWSDisconnected(t *testing.T, agentID string) { |
| 143 | t.Helper() |
no test coverage detected