Feature/MontFerret#250 wait style (MontFerret#257)

* Added WAIT_ATTR functions
3timeslazy · Apr 10, 2019 · f7ead39 · f7ead39
1 parent 75fe1bb
commit f7ead39
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/pkg/drivers/http/document.go b/pkg/drivers/http/document.go
@@ -237,6 +237,14 @@ func (doc *HTMLDocument) WaitForClassBySelectorAll(_ context.Context, _, _ value
 	return core.ErrNotSupported
 }
 
+func (doc *HTMLDocument) WaitForAttributeBySelector(_ context.Context, _, _ values.String, _ core.Value, _ drivers.WaitEvent) error {
+	return core.ErrNotSupported
+}
+
+func (doc *HTMLDocument) WaitForAttributeBySelectorAll(_ context.Context, _, _ values.String, _ core.Value, _ drivers.WaitEvent) error {
+	return core.ErrNotSupported
+}
+
 func (doc *HTMLDocument) Close() error {
 	return nil
 }
diff --git a/pkg/drivers/http/element.go b/pkg/drivers/http/element.go
@@ -406,6 +406,10 @@ func (el *HTMLElement) WaitForClass(_ context.Context, _ values.String, _ driver
 	return core.ErrNotSupported
 }
 
+func (el *HTMLElement) WaitForAttribute(_ context.Context, _ values.String, _ core.Value, _ drivers.WaitEvent) error {
+	return core.ErrNotSupported
+}
+
 func (el *HTMLElement) ensureStyles(ctx context.Context) error {
 	if el.styles == nil {
 		styles, err := el.parseStyles(ctx)

diff --git a/pkg/drivers/value.go b/pkg/drivers/value.go
@@ -96,6 +96,8 @@ type (
 		WaitForStyle(ctx context.Context, name values.String, value core.Value, when WaitEvent) error
 
 		WaitForClass(ctx context.Context, class values.String, when WaitEvent) error
+
+		WaitForAttribute(ctx context.Context, name values.String, value core.Value, when WaitEvent) error
 	}
 
 	// The Document interface represents any web page loaded in the browser
@@ -160,6 +162,10 @@ type (
 		WaitForClassBySelector(ctx context.Context, selector, class values.String, when WaitEvent) error
 
 		WaitForClassBySelectorAll(ctx context.Context, selector, class values.String, when WaitEvent) error
+
+		WaitForAttributeBySelector(ctx context.Context, selector, name values.String, value core.Value, when WaitEvent) error
+
+		WaitForAttributeBySelectorAll(ctx context.Context, selector, name values.String, value core.Value, when WaitEvent) error
 	}
 )