Add AddCookie functionality tests

Author: heraclmene

remote_driver_cookie.go

@@ -1,6 +1,7 @@
 package goselenium
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
 )
@@ -27,6 +28,11 @@ type Cookie struct {
 	HTTPOnly   bool   `json:"httpOnly"`
 }
 
+// AddCookieResponse is the result returned from calling the AddCookie method.
+type AddCookieResponse struct {
+	State string
+}
+
 func (s *seleniumWebDriver) AllCookies() (*AllCookiesResponse, error) {
 	if len(s.sessionID) == 0 {
 		return nil, newSessionIDError("AllCookies")
@@ -72,3 +78,33 @@ func (s *seleniumWebDriver) Cookie(name string) (*CookieResponse, error) {
 
 	return &response, nil
 }
+
+func (s *seleniumWebDriver) AddCookie(c *Cookie) (*AddCookieResponse, error) {
+	if len(s.sessionID) == 0 {
+		return nil, newSessionIDError("AddCookie")
+	}
+
+	var err error
+
+	url := fmt.Sprintf("%s/session/%s/cookie", s.seleniumURL, s.sessionID)
+
+	j := map[string]Cookie{
+		"cookie": *c,
+	}
+	b, err := json.Marshal(j)
+	if err != nil {
+		return nil, newMarshallingError(err, "AddCookie", c)
+	}
+	body := bytes.NewReader(b)
+	resp, err := s.stateRequest(&request{
+		url:           url,
+		body:          body,
+		method:        "POST",
+		callingMethod: "AddCookie",
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &AddCookieResponse{State: resp.State}, nil
+}

remote_driver_cookie_test.go

@@ -156,3 +156,75 @@ func Test_CookieCookie_CorrectResponseIsReturned(t *testing.T) {
 		t.Errorf(correctResponseErrorText)
 	}
 }
+
+/*
+	AddCookie tests
+*/
+func Test_CookieAddCookie_InvalidSessionIdResultsInError(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "",
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	_, err := d.AddCookie(nil)
+	if err == nil || !IsSessionIDError(err) {
+		t.Errorf(sessionIDErrorText)
+	}
+}
+
+func Test_CookieAddCookie_CommunicationErrorIsReturnedCorrectly(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "",
+		errorToReturn: errors.New("An error :<"),
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	_, err := d.AddCookie(&Cookie{Name: "cookie", Path: "/"})
+	if err == nil || !IsCommunicationError(err) {
+		t.Errorf(apiCommunicationErrorText)
+	}
+}
+
+func Test_CookieAddCookie_UnmarshallingErrorIsReturnedCorrectly(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "Invalid JSON!",
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	_, err := d.AddCookie(&Cookie{Name: "cookie", Path: "/"})
+	if err == nil || !IsUnmarshallingError(err) {
+		t.Errorf(unmarshallingErrorText)
+	}
+}
+
+func Test_CookieAddCookie_CorrectResponseIsReturned(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn: `{
+			"state": "success",
+			"value": { 
+					"name": "Test Cookie",
+					"value": "Test Value",
+					"path": "/",
+					"domain": "www.google.com",
+					"secure": true,
+					"httpOnly": true,
+					"expiry": "2016-12-25T00:00:00Z"
+			}
+		}`,
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	resp, err := d.AddCookie(&Cookie{Name: "cookie", Path: "/"})
+	if err != nil || resp.State != "success" {
+		t.Errorf(correctResponseErrorText)
+	}
+}

test/integration_tests/cookie_addcookie_test.go

@@ -0,0 +1,38 @@
+package integrationtests
+
+import (
+	"testing"
+
+	"github.com/bunsenapp/go-selenium"
+)
+
+func Test_CookieAddCookie_CanAddCookieWithCorrectFields(t *testing.T) {
+	setUp()
+	defer tearDown()
+
+	driver := createDriver(t)
+	_, err := driver.CreateSession()
+	if err != nil {
+		errorAndWrap(t, "Error thrown whilst trying to create session.", err)
+	}
+
+	_, err = driver.Go("https://news.ycombinator.com")
+	if err != nil {
+		errorAndWrap(t, "Error was thrown whilst navigating or result was not a success.", err)
+	}
+
+	resp, err := driver.AddCookie(&goselenium.Cookie{
+		Name:       "cookie",
+		Value:      "cookieValue",
+		Path:       "/",
+		Domain:     ".ycombinator.com",
+		SecureOnly: false,
+		HTTPOnly:   true,
+	})
+	if err != nil || resp.State != "success" {
+		errorAndWrap(t, "Error was thrown whilst retrieving cookies", err)
+	}
+
+	printObjectResult(resp)
+
+}

web_driver.go

@@ -198,6 +198,9 @@ type WebDriver interface {
 	// Cookie gets a single named cookie associated with the active URL of the
 	// current browsing context.
 	Cookie(name string) (*CookieResponse, error)
+
+	// AddCookie adds a cookie to the current browsing context.
+	AddCookie(c *Cookie) (*AddCookieResponse, error)
 }
 
 // Element is an interface which specifies what all WebDriver elements