use context for state calls

Author: matryer

suggestionbox/suggestionbox_state.go

@@ -27,6 +27,7 @@ func (c *Client) OpenState(ctx context.Context, modelID string) (io.ReadCloser,
 	if err != nil {
 		return nil, err
 	}
+	req = req.WithContext(ctx)
 	resp, err := c.client.HTTPClient.Do(req)
 	if err != nil {
 		return nil, err
@@ -62,6 +63,7 @@ func (c *Client) PostState(ctx context.Context, r io.Reader) (Model, error) {
 	if err != nil {
 		return model, err
 	}
+	req = req.WithContext(ctx)
 	req.Header.Set("Accept", "application/json; charset=utf-8")
 	req.Header.Set("Content-Type", w.FormDataContentType())
 	_, err = c.client.Do(req, &model)
@@ -93,6 +95,7 @@ func (c *Client) PostStateURL(ctx context.Context, stateURL *url.URL) (Model, er
 	}
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
 	req.Header.Set("Accept", "application/json; charset=utf-8")
+	req = req.WithContext(ctx)
 	_, err = c.client.Do(req, &model)
 	if err != nil {
 		return model, err