mirror of
https://github.com/taigrr/pastebin
synced 2026-04-07 17:41:30 -07:00
77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPasteSuccess(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, http.MethodPost, r.Method)
|
|
err := r.ParseForm()
|
|
require.NoError(t, err)
|
|
blob := r.FormValue("blob")
|
|
assert.Equal(t, "test content", blob)
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(r.Host + "/p/abc123"))
|
|
}))
|
|
defer server.Close()
|
|
|
|
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) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer server.Close()
|
|
|
|
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")
|
|
}
|
|
|
|
func TestPasteInvalidURL(t *testing.T) {
|
|
cli := NewClient("http://invalid.invalid.invalid:99999", false)
|
|
err := cli.Paste(strings.NewReader("test content"))
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestNewClient(t *testing.T) {
|
|
cli := NewClient("http://example.com", true)
|
|
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)
|
|
}
|