diff --git a/src/tests/test_csv.py b/src/tests/test_csv.py index 8e59072..8f7951f 100644 --- a/src/tests/test_csv.py +++ b/src/tests/test_csv.py @@ -17,7 +17,7 @@ def setUp(self): self.config = MagicMock() self.store = DAQJobStoreCSV(self.config) - @patch("daq.store.csv.add_date_to_file_name", return_value="test.csv") + @patch("daq.store.csv.modify_file_path", return_value="test.csv") @patch("builtins.open", new_callable=mock_open) @patch("os.path.exists", return_value=False) @patch("pathlib.Path.touch") @@ -39,7 +39,7 @@ def test_handle_message_new_file( file = self.store._open_csv_files["test.csv"] self.assertEqual(file.write_queue.qsize(), 3) # 1 header + 2 rows - @patch("daq.store.csv.add_date_to_file_name", return_value="test.csv") + @patch("daq.store.csv.modify_file_path", return_value="test.csv") @patch("builtins.open", new_callable=mock_open) @patch("os.path.exists", return_value=True) def test_handle_message_existing_file(self, mock_exists, mock_open, mock_add_date):