Fix ListEmailMessages (#2)

Author: lcd1232

.codecov.yaml

@@ -0,0 +1,2 @@
+ignore:
+  - "mock_*.go"       # Ignore all generated mocks

.mockery.yaml

@@ -0,0 +1,10 @@
+with-expecter: true
+packages:
+  github.com/temp-mail-io/temp-mail-go:
+    interfaces:
+      doer:
+        config:
+          filename: mock_doer.go
+          dir: .
+          mockname: mockDoer
+          inpackage: true

.pre-commit-config.yaml

@@ -14,3 +14,14 @@ repos:
       - id: golangci-lint
         entry: golangci-lint run --fix
         pass_filenames: false
+
+  - repo: local
+    hooks:
+      - id: go-generate
+        name: go generate
+        entry: go generate ./...
+        language: golang
+        language_version: 1.23.4
+        pass_filenames: false
+        additional_dependencies:
+          - github.com/vektra/mockery/[email protected]

README.md

@@ -38,7 +38,6 @@ package main
 
 import (
 	"context"
-	"fmt"
 	"log"
 
 	"github.com/temp-mail-io/temp-mail-go"
@@ -48,12 +47,23 @@ func main() {
 	// Replace with your real API key
 	client := tempmail.NewClient("YOUR_API_KEY", nil)
 
-	domains, _, err := client.ListDomains(context.Background())
+	email, _, err := client.CreateEmail(context.Background(), tempmail.CreateEmailOptions{})
 	if err != nil {
-		log.Fatalf("Error listing domains: %v", err)
+		log.Fatalf("Failed to create temporary email: %v", err)
 	}
 
-	fmt.Println("Available domains:", domains)
+    // Use the created temporary email on the website, service, etc...
+	...
+
+	// Fetch messages for the created email
+    data, _, err := client.ListEmailMessages(context.Background(), email.Email)
+	if err != nil {
+        log.Fatalf("Failed to fetch messages: %v", err)
+    }
+
+	for _, m := range data.Messages {
+		// Iterate over messages
+    }
 }
 ```
 

client_test.go

@@ -1,9 +1,11 @@
 package tempmail
 
 import (
+	"bytes"
 	"context"
 	"io"
 	"net/http"
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -55,3 +57,18 @@ func TestNewRequest(t *testing.T) {
 		assert.JSONEq(t, `{"domain":"example.com"}`, string(b))
 	})
 }
+
+// readFile reads the file from the given path.
+func readFile(t *testing.T, path string) []byte {
+	b, err := os.ReadFile(path)
+	require.NoError(t, err)
+	return b
+}
+
+// newTestResponse creates a new test response with the given status code and body.
+func newTestResponse(statusCode int, body []byte) *http.Response {
+	return &http.Response{
+		StatusCode: statusCode,
+		Body:       io.NopCloser(bytes.NewReader(body)),
+	}
+}

go.mod

@@ -7,5 +7,6 @@ require github.com/stretchr/testify v1.10.0
 require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/stretchr/objx v0.5.2 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

go.sum

@@ -2,6 +2,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=
+github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
 github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
 github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=

list_email_messages.go

@@ -46,16 +46,16 @@ type ListEmailMessagesAttachmentResponse struct {
 }
 
 // ListEmailMessages returns all messages for the email address.
-func (c *Client) ListEmailMessages(ctx context.Context, email string) ([]ListEmailMessagesResponse, *Response, error) {
+func (c *Client) ListEmailMessages(ctx context.Context, email string) (ListEmailMessagesResponse, *Response, error) {
 	req, err := c.newRequest(ctx, http.MethodGet, fmt.Sprintf("/v1/emails/%s/messages", email), nil)
 	if err != nil {
-		return nil, nil, err
+		return ListEmailMessagesResponse{}, nil, err
 	}
 
-	var resp []ListEmailMessagesResponse
+	var resp ListEmailMessagesResponse
 	r, err := c.do(req, &resp)
 	if err != nil {
-		return nil, nil, err
+		return ListEmailMessagesResponse{}, nil, err
 	}
 
 	return resp, r, nil

list_email_messages_test.go

@@ -0,0 +1,40 @@
+package tempmail
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	"github.com/stretchr/testify/require"
+)
+
+func TestClient_ListEmailMessages(t *testing.T) {
+	mDoer := newMockDoer(t)
+	mDoer.EXPECT().Do(mock.Anything).Return(newTestResponse(200, readFile(t, "testdata/list_email_messages.json")), nil)
+
+	c := newClient()
+	c.doer = mDoer
+	r, resp, err := c.ListEmailMessages(context.Background(), "[email protected]")
+	require.NoError(t, err)
+	assert.Equal(t, 200, resp.StatusCode)
+	require.Len(t, r.Messages, 1)
+	assert.Equal(t, ListEmailMessagesMessageResponse{
+		ID:        "01JE97FT950QRPDYGDXJ4R43QR",
+		From:      "[email protected]",
+		To:        "[email protected]",
+		CC:        []string{"[email protected]"},
+		Subject:   "Your account has been created",
+		BodyText:  "Welcome to our service! Your account has been created successfully.",
+		BodyHTML:  "<p>Welcome to our service! Your account has been created successfully.</p>",
+		CreatedAt: time.Date(2022, 1, 31, 22, 0, 0, 0, time.UTC),
+		Attachments: []ListEmailMessagesAttachmentResponse{
+			{
+				ID:   "01JE97K1PBYVGKY0PVE3KXSBF9",
+				Name: "invoice.pdf",
+				Size: 5120,
+			},
+		},
+	}, r.Messages[0])
+}