taigrr/pastebin
1
0
Fork 0
mirror of https://github.com/taigrr/pastebin synced 2026-04-14 02:28:04 -07:00
Code Issues Projects Releases Wiki Activity

feat: add graceful shutdown, health endpoint, and improved client API

Browse Source
This commit is contained in:
Tai Groot
2026-03-09 06:00:16 +00:00
parent ed52f9102a
commit 21bc47697b
4 changed files with 127 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
client/client_test.go
View File

@@ -1,6 +1,7 @@
package client
import (
"bytes"
"net/http"
"net/http/httptest"
"strings"
@@ -18,12 +19,29 @@ func TestPasteSuccess(t *testing.T) {
blob := r.FormValue("blob")
assert.Equal(t, "test content", blob)
w.WriteHeader(http.StatusOK)
_, _ = w.Write([]byte(r.Host + "/p/abc123"))
}))
defer server.Close()
cli := NewClient(server.URL, false)
var output bytes.Buffer
cli := NewClient(server.URL, false).WithOutput(&output)
err := cli.Paste(strings.NewReader("test content"))
assert.NoError(t, err)
assert.Contains(t, output.String(), "/p/abc123")
}
func TestPasteRedirect(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Location", "/p/redirect123")
w.WriteHeader(http.StatusFound)
}))
defer server.Close()
var output bytes.Buffer
cli := NewClient(server.URL, false).WithOutput(&output)
err := cli.Paste(strings.NewReader("redirect content"))
assert.NoError(t, err)
assert.Equal(t, "/p/redirect123", output.String())
}
func TestPasteServerError(t *testing.T) {
@@ -32,7 +50,8 @@ func TestPasteServerError(t *testing.T) {
}))
defer server.Close()
cli := NewClient(server.URL, false)
var output bytes.Buffer
cli := NewClient(server.URL, false).WithOutput(&output)
err := cli.Paste(strings.NewReader("test content"))
assert.Error(t, err)
assert.Contains(t, err.Error(), "unexpected response")
@@ -46,6 +65,12 @@ func TestPasteInvalidURL(t *testing.T) {
func TestNewClient(t *testing.T) {
cli := NewClient("http://example.com", true)
assert.Equal(t, "http://example.com", cli.url)
assert.Equal(t, "http://example.com", cli.serviceURL)
assert.True(t, cli.insecure)
}
func TestWithOutput(t *testing.T) {
var buf bytes.Buffer
cli := NewClient("http://example.com", false).WithOutput(&buf)
assert.Equal(t, &buf, cli.output)
}