fix: fix tests for DAQJobStoreRaw

ENRG-tr · Nov 22, 2024 · 82228a4 · 82228a4
1 parent ad57907
commit 82228a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tests/test_raw.py b/src/tests/test_raw.py
@@ -37,7 +37,7 @@ def test_handle_message_new_file(
         self.store.handle_message(message)
 
         mock_add_date.assert_called_once_with("test.raw", True, None)
-        mock_open.assert_called_once_with("out/test.raw", "ab")
+        mock_open.assert_called_once_with("out/test.raw", "wb")
         self.assertIn("out/test.raw", self.store._open_raw_files)
         file = self.store._open_raw_files["out/test.raw"]
         self.assertEqual(len(file.write_queue), 1)  # 1 raw data
@@ -56,7 +56,7 @@ def test_handle_message_existing_file(self, mock_exists, mock_open, mock_add_dat
         self.store.handle_message(message)
 
         mock_add_date.assert_called_once_with("test.raw", True, None)
-        mock_open.assert_called_once_with("out/test.raw", "ab")
+        mock_open.assert_called_once_with("out/test.raw", "wb")
         self.assertIn("out/test.raw", self.store._open_raw_files)
         file = self.store._open_raw_files["out/test.raw"]
         self.assertEqual(len(file.write_queue), 1)  # 1 raw data