def test_open_file_passes_file_to_callback(self):
cfg = Config(self.__build_args())
with patch("builtins.open", mock_open(read_data="A Test file")):
cfg.open_file(
"mockfile", "r", lambda f: self.assertEqual("A Test file", f.readline())
)
def test_open_file_calls_callback_with_expected_stream(self):
cfg = Config(self.__build_args())
cb = Mock()
with patch("builtins.open", mock_open(read_data="A Test file")):
cfg.open_file("mockfile", "r", cb)
cb.assert_called_once()
def test_open_file_can_write(self):
cfg = Config(self.__build_args())
m = mock_open()
with patch("builtins.open", m):
cfg.open_file("mockfile", "w", lambda f: f.write("some data"))
m.assert_called_once_with("mockfile", "w")
handle = m()
handle.write.assert_called_once_with("some data")
def __init__(self, config: Config):
self.package_map: Dict[str, str] = {}
config.open_file(config.package, "r", self.__parse_packages)
