From 52ceb81a73c310f023ae72d4c36d497a8828d529 Mon Sep 17 00:00:00 2001
From: Florian Bruhin <me@the-compiler.org>
Date: Tue, 16 Oct 2018 11:00:05 +0200
Subject: [PATCH] Add TabbedBrowserStub.current_url()

---
 tests/helpers/stubs.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/helpers/stubs.py b/tests/helpers/stubs.py
index 7b81b8c64..9bf5b837d 100644
--- a/tests/helpers/stubs.py
+++ b/tests/helpers/stubs.py
@@ -502,6 +502,7 @@ class TabbedBrowserStub(QObject):
         self.widget = TabWidgetStub()
         self.shutting_down = False
         self.opened_url = None
+        self.cur_url = None
 
     def on_tab_close_requested(self, idx):
         del self.widget.tabs[idx]
@@ -515,6 +516,11 @@ class TabbedBrowserStub(QObject):
     def openurl(self, url, *, newtab):
         self.opened_url = url
 
+    def current_url(self):
+        if self.current_url is None:
+            raise ValueError("current_url got called with cur_url None!")
+        return self.cur_url
+
 
 class TabWidgetStub(QObject):