Function TestParseMessage

notification/sender/notification_message_test.go:13–73

(t *testing.T)

)

func TestParseMessage(t *testing.T) {
	testCases := []struct {
		name     string
		input    string
		expected *sender.Message
	}{
		{
			name: "ValidMessage",
			input: `Subject: Test Subject
Header1: Value1
InvalidHeaderLine will be dropped
Header2: Value2

This is the body of the message.`,
			expected: &sender.Message{
				Subject: "Test Subject",
				Headers: map[string]string{
					"Header1": "Value1",
					"Header2": "Value2",
				},
				Body: "This is the body of the message.",
			},
		},
		{
			name: "ValidMessage",
			input: `Subject: Test Subject
Header1: Value1
InvalidHeaderLine will be dropped
Header2: Value2

This is the body of the message.`,
			expected: &sender.Message{
				Subject: "Test Subject",
				Headers: map[string]string{
					"Header1": "Value1",
					"Header2": "Value2",
				},
				Body: "This is the body of the message.",
			},
		}, // Add more test cases here...
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			reader := strings.NewReader(tc.input)
			ctx := testlogging.Context(t)
			actual, err := sender.ParseMessage(ctx, reader)

			require.NoError(t, err)
			require.Equal(t, tc.expected.Subject, actual.Subject, "ParseMessage() Subject mismatch")
			require.Equal(t, tc.expected.Body, actual.Body, "ParseMessage() Body mismatch")
			require.Equal(t, tc.expected.Headers, actual.Headers, "ParseMessage() Headers mismatch")

			actualString := actual.ToString()
			roundTrip, err := sender.ParseMessage(ctx, strings.NewReader(actualString))
			require.NoError(t, err)

			require.Equal(t, tc.expected, roundTrip, "ToString() did not roundtrip")