Updated tests

hooklift · Feb 1, 2019 · 0ef471c · 0ef471c
1 parent 9f5ceb6
commit 0ef471c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/example/gen/gen.go b/example/gen/gen.go
@@ -1,6 +1,7 @@
 package gen
 
 import (
+	"context"
 	"encoding/xml"
 	"time"
 
@@ -39,7 +40,7 @@ func NewStockQuotePortType(client *soap.Client) StockQuotePortType {
 
 func (service *stockQuotePortType) GetLastTradePrice(request *TradePriceRequest) (*TradePrice, error) {
 	response := new(TradePrice)
-	err := service.client.Call("http://example.com/GetLastTradePrice", request, response)
+	err := service.client.Call(context.Background(), "http://example.com/GetLastTradePrice", request, response)
 	if err != nil {
 		return nil, err
 	}

diff --git a/soap/soap_test.go b/soap/soap_test.go
@@ -1,6 +1,7 @@
 package soap
 
 import (
+	"context"
 	"encoding/xml"
 	"net/http"
 	"net/http/httptest"
@@ -52,7 +53,7 @@ func TestClient_Call(t *testing.T) {
 	client := NewClient(ts.URL)
 	req := &Ping{Request: &PingRequest{Message: "Hi"}}
 	reply := &PingResponse{}
-	if err := client.Call("GetData", req, reply); err != nil {
+	if err := client.Call(context.Background(), "GetData", req, reply); err != nil {
 		t.Fatalf("couln't call service: %v", err)
 	}
 
@@ -105,7 +106,7 @@ func TestClient_Send_Correct_Headers(t *testing.T) {
 		client := NewClient(ts.URL, WithHTTPHeaders(test.reqHeaders))
 		req := struct{}{}
 		reply := struct{}{}
-		client.Call(test.action, req, reply)
+		client.Call(context.Background(), test.action, req, reply)
 
 		for k, v := range test.expectedHeaders {
 			h := gotHeaders.Get(k)