// SPDX-FileCopyrightText: 2022-2023 The go-mail Authors
//
// SPDX-License-Identifier: MIT

package mail

import "testing"

func TestFile(t *testing.T) {
	t.Run("setHeader", func(t *testing.T) {
		f := File{
			Name:   "testfile.txt",
			Header: make(map[string][]string),
		}
		f.setHeader(HeaderContentType, "text/plain")
		contentType, ok := f.Header[HeaderContentType.String()]
		if !ok {
			t.Fatalf("setHeader failed. Expected header %s to be set", HeaderContentType)
		}
		if len(contentType) != 1 {
			t.Fatalf("setHeader failed. Expected header %s to have one value, got: %d", HeaderContentType,
				len(contentType))
		}
		if contentType[0] != "text/plain" {
			t.Fatalf("setHeader failed. Expected header %s to have value %s, got: %s",
				HeaderContentType.String(), "text/plain", contentType[0])
		}
	})
	t.Run("getHeader", func(t *testing.T) {
		f := File{
			Name:   "testfile.txt",
			Header: make(map[string][]string),
		}
		f.setHeader(HeaderContentType, "text/plain")
		contentType, ok := f.getHeader(HeaderContentType)
		if !ok {
			t.Fatalf("setHeader failed. Expected header %s to be set", HeaderContentType)
		}
		if contentType != "text/plain" {
			t.Fatalf("setHeader failed. Expected header %s to have value %s, got: %s",
				HeaderContentType.String(), "text/plain", contentType)
		}
	})
	t.Run("WithFileDescription", func(t *testing.T) {
		tests := []struct {
			name string
			desc string
		}{
			{"File description: test", "test"},
			{"File description: with newline", "test\n"},
			{"File description: empty", ""},
		}
		for _, tt := range tests {
			t.Run(tt.name, func(t *testing.T) {
				message := NewMsg()
				message.AttachFile("file.go", WithFileDescription(tt.desc))
				attachments := message.GetAttachments()
				if len(attachments) <= 0 {
					t.Fatalf("failed to retrieve attachments list")
				}
				firstAttachment := attachments[0]
				if firstAttachment == nil {
					t.Fatalf("failed to retrieve first attachment, got nil")
				}
				if firstAttachment.Desc != tt.desc {
					t.Errorf("WithFileDescription() failed. Expected: %s, got: %s", tt.desc,
						firstAttachment.Desc)
				}
			})
		}
	})
	t.Run("WithFileContentID", func(t *testing.T) {
		tests := []struct {
			name string
			id   string
		}{
			{"Content-ID: test", "test"},
			{"Content-ID: with newline", "test\n"},
			{"Content-ID: empty", ""},
		}
		for _, tt := range tests {
			t.Run(tt.name, func(t *testing.T) {
				message := NewMsg()
				message.AttachFile("file.go", WithFileContentID(tt.id))
				attachments := message.GetAttachments()
				if len(attachments) <= 0 {
					t.Fatalf("failed to retrieve attachments list")
				}
				firstAttachment := attachments[0]
				if firstAttachment == nil {
					t.Fatalf("failed to retrieve first attachment, got nil")
				}
				contentID := firstAttachment.Header.Get(HeaderContentID.String())
				if contentID != tt.id {
					t.Errorf("WithFileContentID() failed. Expected: %s, got: %s", tt.id,
						contentID)
				}
			})
		}
	})
	t.Run("WithFileEncoding", func(t *testing.T) {
		tests := []struct {
			name     string
			encoding Encoding
			want     Encoding
		}{
			{"File encoding: US-ASCII", EncodingUSASCII, EncodingUSASCII},
			{"File encoding: 8bit raw", NoEncoding, NoEncoding},
			{"File encoding: Base64", EncodingB64, EncodingB64},
			{"File encoding: quoted-printable (not allowed)", EncodingQP, ""},
		}
		for _, tt := range tests {
			t.Run(tt.name, func(t *testing.T) {
				message := NewMsg()
				message.AttachFile("file.go", WithFileEncoding(tt.encoding))
				attachments := message.GetAttachments()
				if len(attachments) <= 0 {
					t.Fatalf("failed to retrieve attachments list")
				}
				firstAttachment := attachments[0]
				if firstAttachment == nil {
					t.Fatalf("failed to retrieve first attachment, got nil")
				}
				if firstAttachment.Enc != tt.want {
					t.Errorf("WithFileEncoding() failed. Expected: %s, got: %s", tt.want, firstAttachment.Enc)
				}
			})
		}
	})
	t.Run("WithFileName", func(t *testing.T) {
		tests := []struct {
			name     string
			fileName string
		}{
			{"File name: test", "test"},
			{"File name: with newline", "test\n"},
			{"File name: empty", ""},
		}
		for _, tt := range tests {
			t.Run(tt.name, func(t *testing.T) {
				message := NewMsg()
				message.AttachFile("file.go", WithFileName(tt.fileName))
				attachments := message.GetAttachments()
				if len(attachments) <= 0 {
					t.Fatalf("failed to retrieve attachments list")
				}
				firstAttachment := attachments[0]
				if firstAttachment == nil {
					t.Fatalf("failed to retrieve first attachment, got nil")
				}
				if firstAttachment.Name != tt.fileName {
					t.Errorf("WithFileName() failed. Expected: %s, got: %s", tt.fileName,
						firstAttachment.Name)
				}
			})
		}
	})
}

/*


// TestFile_WithFileContentType tests the WithFileContentType option
func TestFile_WithFileContentType(t *testing.T) {
	tests := []struct {
		name string
		ct   ContentType
		want string
	}{
		{"File content-type: text/plain", TypeTextPlain, "text/plain"},
		{"File content-type: html/html", TypeTextHTML, "text/html"},
		{"File content-type: application/octet-stream", TypeAppOctetStream, "application/octet-stream"},
		{"File content-type: application/pgp-encrypted", TypePGPEncrypted, "application/pgp-encrypted"},
		{"File content-type: application/pgp-signature", TypePGPSignature, "application/pgp-signature"},
	}
	for _, tt := range tests {
		m := NewMsg()
		t.Run(tt.name, func(t *testing.T) {
			m.AttachFile("file.go", WithFileContentType(tt.ct))
			al := m.GetAttachments()
			if len(al) <= 0 {
				t.Errorf("AttachFile() failed. Attachment list is empty")
			}
			a := al[0]
			if a.ContentType != ContentType(tt.want) {
				t.Errorf("WithFileContentType() failed. Expected: %s, got: %s", tt.want, a.ContentType)
			}
		})
	}
}


*/