Add AlertText functionality and tests.

Author: heraclmene

remote_driver_alert.go

@@ -14,6 +14,13 @@ type AcceptAlertResponse struct {
 	State string
 }
 
+// AlertTextResponse is the response returned from calling the AlertText
+// method.
+type AlertTextResponse struct {
+	State string
+	Text  string
+}
+
 func (s *seleniumWebDriver) DismissAlert() (*DismissAlertResponse, error) {
 	if len(s.sessionID) == 0 {
 		return nil, newSessionIDError("DismissAlert")
@@ -57,3 +64,25 @@ func (s *seleniumWebDriver) AcceptAlert() (*AcceptAlertResponse, error) {
 
 	return &AcceptAlertResponse{State: resp.State}, nil
 }
+
+func (s *seleniumWebDriver) AlertText() (*AlertTextResponse, error) {
+	if len(s.sessionID) == 0 {
+		return nil, newSessionIDError("AlertTextResponse")
+	}
+
+	var err error
+
+	url := fmt.Sprintf("%s/session/%s/alert/text", s.seleniumURL, s.sessionID)
+
+	resp, err := s.valueRequest(&request{
+		url:           url,
+		method:        "GET",
+		body:          nil,
+		callingMethod: "AlertText",
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &AlertTextResponse{State: resp.State, Text: resp.Value}, nil
+}

remote_driver_alert_test.go

@@ -136,3 +136,69 @@ func Test_AlertAcceptAlert_CorrectResponseIsReturned(t *testing.T) {
 		t.Errorf(correctResponseErrorText)
 	}
 }
+
+/*
+	AlertText() Tests
+*/
+
+func Test_AlertAlertText_InvalidSessionIdResultsInError(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "",
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+
+	_, err := d.AlertText()
+	if err == nil || !IsSessionIDError(err) {
+		t.Errorf(sessionIDErrorText)
+	}
+}
+
+func Test_AlertAlertText_CommunicationErrorIsReturnedCorrectly(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "",
+		errorToReturn: errors.New("An error :<"),
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	_, err := d.AlertText()
+	if err == nil || !IsCommunicationError(err) {
+		t.Errorf(apiCommunicationErrorText)
+	}
+}
+
+func Test_AlertAlertText_UnmarshallingErrorIsReturnedCorrectly(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn:  "Invalid JSON!",
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	_, err := d.AlertText()
+	if err == nil || !IsUnmarshallingError(err) {
+		t.Errorf(unmarshallingErrorText)
+	}
+}
+
+func Test_AlertAlertText_CorrectResponseIsReturned(t *testing.T) {
+	api := &testableAPIService{
+		jsonToReturn: `{
+			"state": "success",
+			"value": "this is text"
+		}`,
+		errorToReturn: nil,
+	}
+
+	d := setUpDriver(setUpDefaultCaps(), api)
+	d.sessionID = "12345"
+
+	resp, err := d.AlertText()
+	if err != nil || resp.State != "success" || resp.Text != "this is text" {
+		t.Errorf(correctResponseErrorText)
+	}
+}

test/integration_tests/alert_alerttext_test.go

@@ -0,0 +1,31 @@
+package integrationtests
+
+import (
+	"testing"
+	"time"
+)
+
+func Test_AlertAlertText_CanGetTheAlertText(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/alert.html")
+	if err != nil {
+		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")
+	}
+
+	printObjectResult(resp)
+}

web_driver.go

@@ -252,6 +252,10 @@ type WebDriver interface {
 	// AcceptAlertResponse accepts an alert if there is one present. If not,
 	// it will throw an error.
 	AcceptAlert() (*AcceptAlertResponse, error)
+
+	// AlertText gets the text associated with the current alert. If there is
+	// not an alert, it will throw an error.
+	AlertText() (*AlertTextResponse, error)
 }
 
 // Element is an interface which specifies what all WebDriver elements