From b5c16e70e14efc867c089704c3f73fd4cc8eb9d6 Mon Sep 17 00:00:00 2001
From: ordinary-jamie <jamie@ordinarylab.dev>
Date: Thu, 9 Mar 2023 13:07:58 +1100
Subject: Add TestResponse.clickbutton onclick kwarg

This change allows the TestResponse to follow customised onclick buttons
---
 webtest/response.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'webtest')

diff --git a/webtest/response.py b/webtest/response.py
index d2eee6c..fea35aa 100644
--- a/webtest/response.py
+++ b/webtest/response.py
@@ -159,16 +159,18 @@ class TestResponse(webob.Response):
         return self.goto(str(found_attrs['uri']), extra_environ=extra_environ)
 
     def clickbutton(self, description=None, buttonid=None, href=None,
-                    index=None, verbose=False):
+                    onclick=None, index=None, verbose=False):
         """
         Like :meth:`~webtest.response.TestResponse.click`, except looks
         for link-like buttons.
         This kind of button should look like
         ``<button onclick="...location.href='url'...">``.
         """
+        href_extract = re.compile(onclick) if onclick else re.compile(r"location\.href='(.*?)'")
+
         found_html, found_desc, found_attrs = self._find_element(
             tag='button', href_attr='onclick',
-            href_extract=re.compile(r"location\.href='(.*?)'"),
+            href_extract=href_extract,
             content=description,
             id=buttonid,
             href_pattern=href,
-- 
cgit v1.2.1