Files
mongo/buildscripts/tests/util/test_fileops.py
Steve McClure b4b23946cd SERVER-90570: Enable formatting checks for buildscripts directory, excluding idl (#22254)
GitOrigin-RevId: 9d997a9f44cd43a8dec7c2a17fa2dbcd875e92f6
2024-05-16 22:07:36 +00:00

28 lines
885 B
Python

"""Unit tests for fileops.py."""
import unittest
from unittest.mock import patch
import buildscripts.util.fileops as under_test
class TestWriteFileToDir(unittest.TestCase):
@patch("os.path.exists")
@patch("buildscripts.util.fileops.write_file")
def test_existing_file_can_be_overriden(self, mock_write_file, mock_exists):
mock_exists.return_value = True
under_test.write_file_to_dir("dir", "file", "contents", overwrite=True)
mock_write_file.assert_called()
@patch("os.path.exists")
@patch("buildscripts.util.fileops.write_file")
def test_existing_file_can_cause_exception(self, mock_write_file, mock_exists):
mock_exists.return_value = True
with self.assertRaises(FileExistsError):
under_test.write_file_to_dir("dir", "file", "contents", overwrite=False)
mock_write_file.assert_not_called()