Function TestTrialProtoTaskIDs

master/internal/api_trials_intg_test.go:833–915

(t *testing.T)

}

func TestTrialProtoTaskIDs(t *testing.T) {
	api, curUser, ctx := setupAPITest(t, nil)
	trial, task0 := createTestTrial(t, api, curUser)

	_, err := db.Bun().NewUpdate().Table("experiments").
		Set("best_trial_id = ?", trial.ID).
		Where("id = ?", trial.ExperimentID).Exec(ctx)
	require.NoError(t, err)

	task1 := &model.Task{
		TaskType:   model.TaskTypeTrial,
		LogVersion: model.TaskLogVersion1,
		StartTime:  task0.StartTime.Add(time.Second),
		TaskID:     trialTaskID(trial.ExperimentID, model.NewRequestID(rand.Reader)),
	}
	require.NoError(t, db.AddTask(ctx, task1))

	task2 := &model.Task{
		TaskType:   model.TaskTypeTrial,
		LogVersion: model.TaskLogVersion1,
		StartTime:  task1.StartTime.Add(time.Second),
		TaskID:     trialTaskID(trial.ExperimentID, model.NewRequestID(rand.Reader)),
	}
	require.NoError(t, db.AddTask(ctx, task2))

	_, err = db.Bun().NewInsert().Model(&[]model.RunTaskID{
		{RunID: trial.ID, TaskID: task1.TaskID},
		{RunID: trial.ID, TaskID: task2.TaskID},
	}).Exec(ctx)
	require.NoError(t, err)

	taskIDs := []string{string(task0.TaskID), string(task1.TaskID), string(task2.TaskID)}

	cases := []struct {
		name string
		f    func(t *testing.T) *trialv1.Trial
	}{
		{"GetTrial", func(t *testing.T) *trialv1.Trial {
			resp, err := api.GetTrial(ctx, &apiv1.GetTrialRequest{
				TrialId: int32(trial.ID),
			})
			require.NoError(t, err)
			return resp.Trial
		}},
		{"GetExperimentTrials", func(t *testing.T) *trialv1.Trial {
			resp, err := api.GetExperimentTrials(ctx, &apiv1.GetExperimentTrialsRequest{
				ExperimentId: int32(trial.ExperimentID),
			})
			require.NoError(t, err)
			require.Len(t, resp.Trials, 1)
			return resp.Trials[0]
		}},
		/* CompareTrials previously and now sends TaskID="". We also will send TaskIDs=[].
		{"CompareTrials", func(t *testing.T) *trialv1.Trial {
			resp, err := api.CompareTrials(ctx, &apiv1.CompareTrialsRequest{
				TrialIds: []int32{int32(trial.ID)},
			})
			require.NoError(t, err)