Function Test_TaskLogs_Golden

cli/task_logs_test.go:22–279

(t *testing.T)

)

func Test_TaskLogs_Golden(t *testing.T) {
	t.Parallel()

	testMessages := []agentapisdk.Message{
		{
			Id:      0,
			Role:    agentapisdk.RoleUser,
			Content: "What is 1 + 1?",
			Time:    time.Now().Add(-2 * time.Minute),
		},
		{
			Id:      1,
			Role:    agentapisdk.RoleAgent,
			Content: "2",
			Time:    time.Now().Add(-1 * time.Minute),
		},
	}

	t.Run("ByTaskName_JSON", func(t *testing.T) {
		t.Parallel()

		setupCtx := testutil.Context(t, testutil.WaitLong)
		setup := setupCLITaskTest(setupCtx, t, fakeAgentAPITaskLogsOK(testMessages))

		inv, root := clitest.New(t, "task", "logs", setup.task.Name, "--output", "json")
		output := clitest.Capture(inv)
		clitest.SetupConfig(t, setup.userClient, root)

		ctx := testutil.Context(t, testutil.WaitLong)
		err := inv.WithContext(ctx).Run()
		require.NoError(t, err)

		// Verify JSON is valid.
		var logs []codersdk.TaskLogEntry
		err = json.NewDecoder(strings.NewReader(output.Stdout())).Decode(&logs)
		require.NoError(t, err)

		// Verify output format with golden file.
		clitest.TestGoldenFile(t, t.Name(), output.Golden(), nil)
	})

	t.Run("ByTaskID_JSON", func(t *testing.T) {
		t.Parallel()

		setupCtx := testutil.Context(t, testutil.WaitLong)
		setup := setupCLITaskTest(setupCtx, t, fakeAgentAPITaskLogsOK(testMessages))

		inv, root := clitest.New(t, "task", "logs", setup.task.ID.String(), "--output", "json")
		output := clitest.Capture(inv)
		clitest.SetupConfig(t, setup.userClient, root)

		ctx := testutil.Context(t, testutil.WaitLong)
		err := inv.WithContext(ctx).Run()
		require.NoError(t, err)

		// Verify JSON is valid.
		var logs []codersdk.TaskLogEntry
		err = json.NewDecoder(strings.NewReader(output.Stdout())).Decode(&logs)