mirror of
https://github.com/tiennm99/tcbs-api.git
synced 2026-06-10 16:11:28 +00:00
tiennm99
ee236d1fe0
- Split monolithic models.go (738 lines) into 6 domain files - Fix schema drift: BasicInfo, PersonalInfo, TotalCashDerivativeResponse, derivative order types, money transfer types aligned to spec - Add missing REST endpoints: bsa-ext, bsa-month (supply/demand) - Add WebSocket support for 5 streaming endpoints (nhooyr.io/websocket) - Add 45 httptest-based tests (74.9% coverage) - Rewrite README with full API coverage table BREAKING CHANGE: struct fields and types changed to match OpenAPI spec. BasicInfo reduced to 5 fields, TokenResponse uses 'token' field, PlaceOrderRequest uses int types, derivative order types renamed.
86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package tcbs
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
// newTestServer creates a test HTTP server and a Client pointing at it.
|
|
func newTestServer(t *testing.T, handler http.HandlerFunc) (*Client, *httptest.Server) {
|
|
t.Helper()
|
|
srv := httptest.NewServer(handler)
|
|
t.Cleanup(srv.Close)
|
|
client := NewClient(WithBaseURL(srv.URL), WithToken("test-token"))
|
|
return client, srv
|
|
}
|
|
|
|
// writeJSON is a test helper to write JSON responses.
|
|
func writeJSON(t *testing.T, w http.ResponseWriter, v any) {
|
|
t.Helper()
|
|
w.Header().Set("Content-Type", "application/json")
|
|
if err := json.NewEncoder(w).Encode(v); err != nil {
|
|
t.Fatalf("failed to encode response: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestNewClient_Defaults(t *testing.T) {
|
|
c := NewClient()
|
|
if c.baseURL != ProductionBaseURL {
|
|
t.Errorf("expected base URL %s, got %s", ProductionBaseURL, c.baseURL)
|
|
}
|
|
if c.httpClient == nil {
|
|
t.Error("expected non-nil http client")
|
|
}
|
|
}
|
|
|
|
func TestNewClient_Options(t *testing.T) {
|
|
c := NewClient(WithBaseURL(SITBaseURL), WithToken("tok"))
|
|
if c.baseURL != SITBaseURL {
|
|
t.Errorf("expected base URL %s, got %s", SITBaseURL, c.baseURL)
|
|
}
|
|
if c.currentToken() != "tok" {
|
|
t.Errorf("expected token 'tok', got %q", c.currentToken())
|
|
}
|
|
}
|
|
|
|
func TestSetToken(t *testing.T) {
|
|
c := NewClient()
|
|
c.SetToken("abc")
|
|
if c.currentToken() != "abc" {
|
|
t.Errorf("expected token 'abc', got %q", c.currentToken())
|
|
}
|
|
}
|
|
|
|
func TestAPIError(t *testing.T) {
|
|
client, _ := newTestServer(t, func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
_, _ = w.Write([]byte(`{"error":"bad"}`))
|
|
})
|
|
err := client.get(context.Background(), "/fail", nil, nil)
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
apiErr, ok := err.(*APIError)
|
|
if !ok {
|
|
t.Fatalf("expected *APIError, got %T", err)
|
|
}
|
|
if apiErr.StatusCode != 400 {
|
|
t.Errorf("expected status 400, got %d", apiErr.StatusCode)
|
|
}
|
|
}
|
|
|
|
func TestAuthorizationHeader(t *testing.T) {
|
|
var gotAuth string
|
|
client, _ := newTestServer(t, func(w http.ResponseWriter, r *http.Request) {
|
|
gotAuth = r.Header.Get("Authorization")
|
|
writeJSON(t, w, []MarketStockInfo{})
|
|
})
|
|
_, _ = client.GetStockPrices(context.Background(), []string{"FPT"})
|
|
if gotAuth != "Bearer test-token" {
|
|
t.Errorf("expected 'Bearer test-token', got %q", gotAuth)
|
|
}
|
|
}
|