Add SendAlertText functionality and tests

remote_driver_alert.go

@@ -1,6 +1,10 @@
 package goselenium
 
-import "fmt"
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+)
 
 // DismissAlertResponse is the response returned from calling the DismissAlert
 // method.
@@ -21,6 +25,12 @@ type AlertTextResponse struct {
 	Text  string
 }
 
+// SendAlertTextResponse is the response returned from calling the
+// SendAlertText method.
+type SendAlertTextResponse struct {
+	State string
+}
+
 func (s *seleniumWebDriver) DismissAlert() (*DismissAlertResponse, error) {
 	if len(s.sessionID) == 0 {
 		return nil, newSessionIDError("DismissAlert")
@@ -86,3 +96,34 @@ func (s *seleniumWebDriver) AlertText() (*AlertTextResponse, error) {
 
 	return &AlertTextResponse{State: resp.State, Text: resp.Value}, nil
 }
+
+func (s *seleniumWebDriver) SendAlertText(text string) (*SendAlertTextResponse, error) {
+	if len(s.sessionID) == 0 {
+		return nil, newSessionIDError("SendAlertText")
+	}
+
+	var err error
+
+	url := fmt.Sprintf("%s/session/%s/alert/text", s.seleniumURL, s.sessionID)
+
+	b := map[string]string{
+		"text": text,
+	}
+	json, err := json.Marshal(b)
+	if err != nil {
+		return nil, newMarshallingError(err, "SendAlertText", b)
+	}
+
+	body := bytes.NewReader(json)
+	resp, err := s.valueRequest(&request{
+		url:           url,
+		method:        "POST",
+		body:          body,
+		callingMethod: "SendAlertText",
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &SendAlertTextResponse{State: resp.State}, nil
+}

remote_driver_alert_test.go

@@ -202,3 +202,67 @@ func Test_AlertAlertText_CorrectResponseIsReturned(t *testing.T) {
 		t.Errorf(correctResponseErrorText)
 	}
 }
+
+/*
+	SendAlertText() Tests
+*/
+func Test_AlertSendAlertText_InvalidSessionIdResultsInError(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "",
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+
+	_, err := d.SendAlertText("test")
+	if err == nil || !IsSessionIDError(err) {
+		t.Errorf(sessionIDErrorText)
+	}
+}
+
+func Test_AlertSendAlertText_CommunicationErrorIsReturnedCorrectly(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "",
+		errorToReturn: errors.New("An error :<"),
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	_, err := d.SendAlertText("test")
+	if err == nil || !IsCommunicationError(err) {
+		t.Errorf(apiCommunicationErrorText)
+	}
+}
+
+func Test_AlertSendAlertText_UnmarshallingErrorIsReturnedCorrectly(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "Invalid JSON!",
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	_, err := d.SendAlertText("test")
+	if err == nil || !IsUnmarshallingError(err) {
+		t.Errorf(unmarshallingErrorText)
+	}
+}
+
+func Test_AlertSendAlertText_CorrectResponseIsReturned(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn: `{
+			"state": "success"
+		}`,
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	resp, err := d.SendAlertText("test")
+	if err != nil || resp.State != "success" {
+		t.Errorf(correctResponseErrorText)
+	}
+}

test/integration_tests/alert_acceptalert_test.go

@@ -1,9 +1,6 @@
 package integrationtests
 
-import (
-	"testing"
-	"time"
-)
+import "testing"
 
 func Test_AlertAcceptAlert_CanAcceptAnAlertCorrectly(t *testing.T) {
 	setUp()
@@ -20,8 +17,6 @@ func Test_AlertAcceptAlert_CanAcceptAnAlertCorrectly(t *testing.T) {
 		t.Errorf("Error visiting URL")
 	}
 
-	time.Sleep(time.Second)
-
 	resp, err := driver.AcceptAlert()
 	if err != nil || resp.State != "success" {
 		t.Errorf("Error returned or accepting an alert was not a success")

test/integration_tests/alert_alerttext_test.go

@@ -1,9 +1,6 @@
 package integrationtests
 
-import (
-	"testing"
-	"time"
-)
+import "testing"
 
 func Test_AlertAlertText_CanGetTheAlertText(t *testing.T) {
 	setUp()
@@ -20,8 +17,6 @@ func Test_AlertAlertText_CanGetTheAlertText(t *testing.T) {
 		t.Errorf("Error visiting URL")
 	}
 
-	time.Sleep(time.Second)
-
 	resp, err := driver.AlertText()
 	if err != nil || resp.State != "success" || resp.Text != "this is an alert" {
 		t.Errorf("Error returned or alert text was not correct")

test/integration_tests/alert_dismissalert_test.go

@@ -2,7 +2,6 @@ package integrationtests
 
 import (
 	"testing"
-	"time"
 
 	goselenium "github.com/bunsenapp/go-selenium"
 )
@@ -22,8 +21,6 @@ func Test_AlertDismissAlert_CanDismissAnAlertCorrectly(t *testing.T) {
 		t.Errorf("Error visiting URL")
 	}
 
-	time.Sleep(time.Second)
-
 	resp, err := driver.DismissAlert()
 	if err != nil || resp.State != "success" {
 		t.Errorf("Error returned or dismissing an alert was not a success")

test/integration_tests/alert_sendalerttext_test.go

@@ -0,0 +1,45 @@
+package integrationtests
+
+import (
+	"testing"
+
+	goselenium "github.com/bunsenapp/go-selenium"
+)
+
+func Test_AlertSendAlertText_CanSendAlertTextCorrectly(t *testing.T) {
+	setUp()
+	defer tearDown()
+
+	driver := createDriver(t)
+	_, err := driver.CreateSession()
+	if err != nil {
+		t.Errorf("Create session error")
+	}
+
+	_, err = driver.Go("https://heraclmene.github.io/helpers/goselenium/prompt.html")
+	if err != nil {
+		t.Errorf("Error visiting URL")
+	}
+
+	resp, err := driver.SendAlertText("test")
+	if err != nil || resp.State != "success" {
+		t.Errorf("Error returned or sending alert text was not a success")
+	}
+
+	_, err = driver.AcceptAlert()
+	if err != nil {
+		t.Errorf("Error was returned when accepting alert.")
+	}
+
+	_, err = driver.AlertText()
+	if err != nil {
+		comErr := err.(goselenium.CommunicationError)
+		if comErr.Response.State != goselenium.NoSuchAlert {
+			t.Errorf("Error returned was not correct.")
+		}
+	} else {
+		t.Errorf("Error returned was not correct.")
+	}
+
+	printObjectResult(resp)
+}

web_driver.go

@@ -256,6 +256,15 @@ type WebDriver interface {
 	// AlertText gets the text associated with the current alert. If there is
 	// not an alert, it will throw an error.
 	AlertText() (*AlertTextResponse, error)
+
+	// SendAlertText enters the text specified into the value box.
+	//
+	// If the prompt is of type 'alert' or 'confirm', a communication error
+	// with code 'element not interactable' will be returned.
+	//
+	// If the prompt is anything other than 'prompt', a communication error with
+	// code 'unsupported operation' will be returned.
+	SendAlertText(text string) (*SendAlertTextResponse, error)
 }
 
 // Element is an interface which specifies what all WebDriver elements