From 0935dabb3615afadef3bdef9134db78a795ae80d Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Wed, 20 May 2020 17:56:08 +0200
Subject: [PATCH] Adjust tests to interface change

---
 sf-writer/test/manual/test_sf_writer.cpp  | 2 +-
 sf-writer/test/test_WriterZmqReceiver.cpp | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/sf-writer/test/manual/test_sf_writer.cpp b/sf-writer/test/manual/test_sf_writer.cpp
index 57102b7..c992586 100644
--- a/sf-writer/test/manual/test_sf_writer.cpp
+++ b/sf-writer/test/manual/test_sf_writer.cpp
@@ -44,7 +44,7 @@ void receive_replay(
 
             cout << "Received " << image_metadata->pulse_id << endl;
 
-            receiver.get_next_image(
+            receiver.get_next_batch(
                     current_pulse_id, image_metadata, image_buffer);
 
             if (image_metadata->pulse_id != current_pulse_id) {
diff --git a/sf-writer/test/test_WriterZmqReceiver.cpp b/sf-writer/test/test_WriterZmqReceiver.cpp
index 847352e..90996c2 100644
--- a/sf-writer/test/test_WriterZmqReceiver.cpp
+++ b/sf-writer/test/test_WriterZmqReceiver.cpp
@@ -70,7 +70,7 @@ TEST(WriterZmqReceiver, basic_test)
                  0);
     }
 
-    receiver.get_next_image(pulse_id, &image_metadata, image_buffer.get());
+    receiver.get_next_batch(pulse_id, &image_metadata, image_buffer.get());
     EXPECT_EQ(pulse_id, image_metadata.pulse_id);
     EXPECT_EQ(image_metadata.is_good_frame, 1);
     EXPECT_EQ(image_metadata.daq_rec, 4);