cmd/tailscale/main: use localapi for login and add tests.
parent
f9310e7a1f
commit
49d35978b6
@ -0,0 +1,73 @@
|
|||||||
|
package localapiclient
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ctx = context.Background()
|
||||||
|
|
||||||
|
type BadStatusHandler struct{}
|
||||||
|
|
||||||
|
func (b *BadStatusHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.WriteHeader(http.StatusBadRequest)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBadStatus(t *testing.T) {
|
||||||
|
ctx, cancel := context.WithDeadline(ctx, time.Now().Add(2*time.Second))
|
||||||
|
client := New(&BadStatusHandler{})
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
_, err := client.Call(ctx, "POST", "test", nil)
|
||||||
|
|
||||||
|
if err.Error() != "request failed with status code 400" {
|
||||||
|
t.Error("Expected bad status error, but got", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type TimeoutHandler struct{}
|
||||||
|
|
||||||
|
var successfulResponse = "successful response!"
|
||||||
|
|
||||||
|
func (b *TimeoutHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
time.Sleep(6 * time.Second)
|
||||||
|
w.Write([]byte(successfulResponse))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTimeout(t *testing.T) {
|
||||||
|
ctx, cancel := context.WithDeadline(ctx, time.Now().Add(2*time.Second))
|
||||||
|
client := New(&TimeoutHandler{})
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
_, err := client.Call(ctx, "GET", "test", nil)
|
||||||
|
|
||||||
|
if err.Error() != "timeout for test" {
|
||||||
|
t.Error("Expected timeout error, but got", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type SuccessfulHandler struct{}
|
||||||
|
|
||||||
|
func (b *SuccessfulHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.Write([]byte(successfulResponse))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSuccess(t *testing.T) {
|
||||||
|
ctx, cancel := context.WithDeadline(ctx, time.Now().Add(2*time.Second))
|
||||||
|
client := New(&SuccessfulHandler{})
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
w, err := client.Call(ctx, "GET", "test", nil)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
t.Error("Expected no error, but got", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
report, err := io.ReadAll(w.Body())
|
||||||
|
if string(report) != successfulResponse {
|
||||||
|
t.Error("Expected successful report, but got", report)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue