(t *testing.T)
| 72 | } |
| 73 | |
| 74 | func TestSchedulerBasicEnqueue(t *testing.T) { |
| 75 | scheduler, frontendClient, querierClient := setupScheduler(t, nil, false) |
| 76 | |
| 77 | frontendLoop := initFrontendLoop(t, frontendClient, "frontend-12345") |
| 78 | frontendToScheduler(t, frontendLoop, &schedulerpb.FrontendToScheduler{ |
| 79 | Type: schedulerpb.ENQUEUE, |
| 80 | QueryID: 1, |
| 81 | UserID: "test", |
| 82 | HttpRequest: &httpgrpc.HTTPRequest{Method: "GET", Url: "/hello"}, |
| 83 | }) |
| 84 | |
| 85 | { |
| 86 | querierLoop, err := querierClient.QuerierLoop(context.Background()) |
| 87 | require.NoError(t, err) |
| 88 | require.NoError(t, querierLoop.Send(&schedulerpb.QuerierToScheduler{QuerierID: "querier-1"})) |
| 89 | |
| 90 | msg2, err := querierLoop.Recv() |
| 91 | require.NoError(t, err) |
| 92 | require.Equal(t, uint64(1), msg2.QueryID) |
| 93 | require.Equal(t, "frontend-12345", msg2.FrontendAddress) |
| 94 | require.Equal(t, "GET", msg2.HttpRequest.Method) |
| 95 | require.Equal(t, "/hello", msg2.HttpRequest.Url) |
| 96 | require.NoError(t, querierLoop.Send(&schedulerpb.QuerierToScheduler{})) |
| 97 | } |
| 98 | |
| 99 | verifyNoPendingRequestsLeft(t, scheduler) |
| 100 | } |
| 101 | |
| 102 | func TestSchedulerEnqueueWithCancel(t *testing.T) { |
| 103 | scheduler, frontendClient, querierClient := setupScheduler(t, nil, false) |
nothing calls this directly
no test coverage detected