1 package mailmux
2 
3 import (
4 	"encoding/json"
5 	"io"
6 	"net/http"
7 	"net/http/httptest"
8 	"net/url"
9 	"strconv"
10 	"strings"
11 	"testing"
12 	"time"
13 )
14 
15 func TestBadRegister(t *testing.T) {
16 	srv := newTestServer()
17 	httpsrv := httptest.NewServer(NewWebServer(srv.Aliases, srv.Users))
18 	client := httpsrv.Client()
19 
20 	registrations := []url.Values{
21 		// this user already exists; created in newTestServer.
22 		url.Values{
23 			"username": []string{testUsername},
24 			"password": []string{testPassword},
25 		},
26 		// username isn't an email address
27 		url.Values{
28 			"username": []string{"bla bla hello world!"},
29 			"password": []string{testPassword},
30 		},
31 		// empty password
32 		url.Values{
33 			"username": []string{testUsername},
34 			"password": []string{},
35 		},
36 	}
37 	for i, form := range registrations {
38 		resp, err := client.PostForm(httpsrv.URL+"/v1/register", form)
39 		if err != nil {
40 			t.Error(err)
41 		}
42 		if resp.StatusCode != http.StatusBadRequest {
43 			t.Errorf("bad registration case %d got HTTP status %s", i, resp.Status)
44 			b, err := io.ReadAll(resp.Body)
45 			if err != nil {
46 				t.Fatal(err)
47 			}
48 			t.Log(string(b))
49 		}
50 	}
51 }
52 
53 // TestAliasLifecycle tests creating, reading, updating then deleting an alias.
54 func TestAliasLifecycle(t *testing.T) {
55 	srv := newTestServer()
56 	httpsrv := httptest.NewServer(NewWebServer(srv.Aliases, srv.Users))
57 	client := httpsrv.Client()
58 
59 	req, err := http.NewRequest(http.MethodPost, httpsrv.URL+"/v1/aliases", nil)
60 	req.SetBasicAuth(testUsername, testPassword)
61 	resp, err := client.Do(req)
62 	if err != nil {
63 		t.Fatalf("create alias: %v", err)
64 	}
65 	defer resp.Body.Close()
66 	var alias Alias
67 	if err := json.NewDecoder(resp.Body).Decode(&alias); err != nil {
68 		t.Fatalf("decode new alias response: %v", err)
69 	}
70 
71 	req, err = http.NewRequest(http.MethodGet, httpsrv.URL+"/v1/aliases", nil)
72 	req.SetBasicAuth(testUsername, testPassword)
73 	resp, err = client.Do(req)
74 	if err != nil {
75 		t.Fatalf("list aliases: %v", err)
76 	}
77 	defer resp.Body.Close()
78 	if resp.StatusCode != http.StatusOK {
79 		b, _ := io.ReadAll(resp.Body)
80 		t.Log(string(b))
81 		t.Fatalf("list aliases: HTTP status %s", resp.Status)
82 	}
83 	var aliases []Alias
84 	if err := json.NewDecoder(resp.Body).Decode(&aliases); err != nil {
85 		t.Fatalf("decode list aliases response: %v", err)
86 	}
87 
88 	// update alias by adding a note and expiry time
89 	v := make(url.Values)
90 	v.Set("expiry", strconv.Itoa(int(time.Now().Add(time.Hour).Unix())))
91 	v.Set("note", "a test note to describe something")
92 	req, err = http.NewRequest(http.MethodPost, httpsrv.URL+"/v1/aliases/"+alias.Recipient, strings.NewReader(v.Encode()))
93 	if err != nil {
94 		t.Fatalf("create update request: %v", err)
95 	}
96 	req.SetBasicAuth(testUsername, testPassword)
97 	resp, err = client.Do(req)
98 	if err != nil {
99 		t.Fatalf("update alias: %v", err)
100 	}
101 	if resp.StatusCode != http.StatusOK {
102 		b, _ := io.ReadAll(resp.Body)
103 		t.Log(string(b))
104 		t.Fatalf("update alias: HTTP status %s", resp.Status)
105 	}
106 	defer resp.Body.Close()
107 	if err := json.NewDecoder(resp.Body).Decode(&alias); err != nil {
108 		t.Fatalf("decode updated alias response: %v", err)
109 	}
110 
111 	req, err = http.NewRequest(http.MethodDelete, httpsrv.URL+"/v1/aliases/"+alias.Recipient, nil)
112 	if err != nil {
113 		t.Fatalf("create delete request: %v", err)
114 	}
115 	req.SetBasicAuth(testUsername, testPassword)
116 	resp, err = client.Do(req)
117 	if err != nil {
118 		t.Fatalf("delete alias: %v", err)
119 	}
120 	if resp.StatusCode != http.StatusNoContent {
121 		b, _ := io.ReadAll(resp.Body)
122 		t.Log(string(b))
123 		t.Fatalf("delete alias: HTTP status %s", resp.Status)
124 	}
125 }