305 lines
8.9 KiB
Go
305 lines
8.9 KiB
Go
|
package tastytrade
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestQueryFutures(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/futures"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"data": {"items": [{"symbol": "AAPL"}]}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.QueryFutures(nil)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.Items) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.Items))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Items[0].Symbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestQueryFuturesError(t *testing.T) {
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
rw.WriteHeader(http.StatusInternalServerError)
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
_, err := api.QueryFutures(nil)
|
||
|
|
|
||
|
|
if err == nil {
|
||
|
|
t.Errorf("expected an error, got nil")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestQueryFuturesWithParams(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/futures?product-code%5B%5D=123&symbol%5B%5D=AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"data": {"items": [{"symbol": "AAPL", "product-code": "123"}]}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
params := &FuturesQueryParams{
|
||
|
|
Symbol: []string{"AAPL"},
|
||
|
|
ProductCode: []string{"123"},
|
||
|
|
}
|
||
|
|
resp, err := api.QueryFutures(params)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.Items) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.Items))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Items[0].Symbol)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].ProductCode != "123" {
|
||
|
|
t.Errorf("expected %s, got %s", "123", resp.Data.Items[0].ProductCode)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetFuture(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/futures/AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"data": {"symbol": "AAPL"}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.GetFuture("AAPL")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Symbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetFutureOptionProduct(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/future-option-products/exchange/AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"context": "test", "data": {"root-symbol": "AAPL"}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.GetFutureOptionProduct("exchange", "AAPL")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Context != "test" {
|
||
|
|
t.Errorf("expected %s, got %s", "test", resp.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.RootSymbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.RootSymbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestListFutureOptions(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/future-options?symbol%5B%5D=AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"context": "test", "data": {"items": [{"symbol": "AAPL", "instrument-type": "future-option"}]}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.ListFutureOptions(&FutureOptionsQueryParams{Symbol: []string{"AAPL"}})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Context != "test" {
|
||
|
|
t.Errorf("expected %s, got %s", "test", resp.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.Items) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.Items))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Items[0].Symbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestListFutureOptionChainsNested(t *testing.T) {
|
||
|
|
expectedURL := "/futures-option-chains/AAPL/nested"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"context": "test", "data": {"futures": [{"symbol": "AAPL"}], "option-chains": [{"underlying-symbol": "AAPL"}]}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.ListFutureOptionChainsNested("AAPL")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Context != "test" {
|
||
|
|
t.Errorf("expected %s, got %s", "test", resp.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.Futures) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.Futures))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Futures[0].Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Futures[0].Symbol)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.OptionChains) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.OptionChains))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.OptionChains[0].UnderlyingSymbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.OptionChains[0].UnderlyingSymbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestListFutureOptionChainsDetailed(t *testing.T) {
|
||
|
|
expectedURL := "/futures-option-chains/AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"context": "test", "data": {"items": [{"symbol": "AAPL", "underlying-symbol": "AAPL"}]}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.ListFutureOptionChainsDetailed("AAPL")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Context != "test" {
|
||
|
|
t.Errorf("expected %s, got %s", "test", resp.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.Items) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.Items))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Items[0].Symbol)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].UnderlyingSymbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Items[0].UnderlyingSymbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetFutureOption(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/future-options/AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"context": "test", "data": {"symbol": "AAPL", "instrument-type": "future-option"}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.GetFutureOption("AAPL")
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Context != "test" {
|
||
|
|
t.Errorf("expected %s, got %s", "test", resp.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Symbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Symbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestListFutureOptionProducts(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/future-option-products"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.Write([]byte(`{"context": "test", "data": {"items": [{"root-symbol": "AAPL", "instrument-type": "future-option"}]}}`))
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
resp, err := api.ListFutureOptionProducts()
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expected nil, got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Context != "test" {
|
||
|
|
t.Errorf("expected %s, got %s", "test", resp.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(resp.Data.Items) != 1 {
|
||
|
|
t.Errorf("expected %d, got %d", 1, len(resp.Data.Items))
|
||
|
|
}
|
||
|
|
|
||
|
|
if resp.Data.Items[0].RootSymbol != "AAPL" {
|
||
|
|
t.Errorf("expected %s, got %s", "AAPL", resp.Data.Items[0].RootSymbol)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetFutureOptionProductError(t *testing.T) {
|
||
|
|
expectedURL := "/instruments/future-option-products/exchange/AAPL"
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.URL.String() != expectedURL {
|
||
|
|
t.Errorf("expected URL to be %s, got %s", expectedURL, req.URL.String())
|
||
|
|
}
|
||
|
|
rw.WriteHeader(http.StatusInternalServerError)
|
||
|
|
}))
|
||
|
|
defer server.Close()
|
||
|
|
|
||
|
|
api := NewTastytradeAPI(server.URL)
|
||
|
|
_, err := api.GetFutureOptionProduct("exchange", "AAPL")
|
||
|
|
|
||
|
|
if err == nil {
|
||
|
|
t.Errorf("expected an error, got nil")
|
||
|
|
}
|
||
|
|
}
|