Function TestAttachmentTelemetry

internal/attachments/flags_test.go:100–193

(t *testing.T)

}

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

	t.Run("flag not passed ignores operation updates", func(t *testing.T) {
		t.Parallel()

		// Given a command with no attachment flag
		_, attachFlag := attachCmd(t, "")
		recorder := &telemetry.InvocationRecorderSpy{}

		// When an operation update is attempted without an attachment event
		event := BeginTelemetry(recorder, "gh issue comment", attachFlag.Count())
		event.RecordOperations(UploadResult{AppendOperations: 1, ReplaceOperations: 1})

		// Then no event or sampling promotion occurs
		assert.Nil(t, event)
		assert.Empty(t, recorder.Events())
		assert.Zero(t, recorder.LastSampleRate)
	})

	tests := []struct {
		name      string
		input     string
		wantCount int64
	}{
		{
			name:      "one attachment",
			input:     "--attach ./first.png",
			wantCount: 1,
		},
		{
			name:      "several attachments before validation",
			input:     "--attach ./first.png --attach ./second.png --attach ./third.png",
			wantCount: 3,
		},
		{
			name:      "empty path counts before validation",
			input:     `--attach ""`,
			wantCount: 1,
		},
		{
			name:      "over limit counts before validation",
			input:     strings.Repeat("--attach ./missing.png ", maxAttachments+1),
			wantCount: maxAttachments + 1,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			// Given raw attachment inputs that have not been validated
			_, attachFlag := attachCmd(t, tt.input)
			recorder := &telemetry.InvocationRecorderSpy{}

			// When an attachment event is recorded without any upload operations
			BeginTelemetry(recorder, "gh issue comment", attachFlag.Count())