Refactor to enable implementing #7545 without changing the API

2 files changed, 2 insertions, 2 deletions

diff --git a/tests/testbench/com/vaadin/tests/tickets/Ticket1589.java b/tests/testbench/com/vaadin/tests/tickets/Ticket1589.java
index b1b652381f..37f9197285 100644
--- a/tests/testbench/com/vaadin/tests/tickets/Ticket1589.java
+++ b/tests/testbench/com/vaadin/tests/tickets/Ticket1589.java
@@ -91,7 +91,7 @@ class MyDynamicResource implements RequestHandler {
                 downloadStream.setParameter("Content-Disposition",
                         "attachment; filename=\"downloadedPNG.png\"");
             }
-            downloadStream.writeTo(response);
+            downloadStream.writeResponse(request, response);
             return true;
         } catch (IOException e) {
             return false;
diff --git a/tests/testbench/com/vaadin/tests/tickets/Ticket2292.java b/tests/testbench/com/vaadin/tests/tickets/Ticket2292.java
index c4c600edaa..1e7fe5d78a 100644
--- a/tests/testbench/com/vaadin/tests/tickets/Ticket2292.java
+++ b/tests/testbench/com/vaadin/tests/tickets/Ticket2292.java
@@ -87,7 +87,7 @@ public class Ticket2292 extends com.vaadin.Application.LegacyApplication
             // Return a stream from the buffer.
             ByteArrayInputStream istream = new ByteArrayInputStream(
                     imagebuffer.toByteArray());
-            new DownloadStream(istream, null, null).writeTo(response);
+            new DownloadStream(istream, null, null).writeResponse(request, response);
             return true;
         } catch (IOException e) {
             return false;