def test_save_not_successful(test_image, tmp_folder):
# ARRANGE
s = Steganography(test_image)
mock_png_path = os.path.join(tmp_folder, "save_not_sucessful.png")
s._path_as_png = MagicMock()
# ACT
s.save(mock_png_path)
# ASSERT
assert not os.path.exists(mock_png_path)
s._path_as_png.assert_not_called()
def test_save_successful(test_image, tmp_folder):
# ARRANGE
s = Steganography(test_image)
s.encode("doesnt matter")
mock_png_path = os.path.join(tmp_folder, "save_sucess.png")
s._path_as_png = MagicMock(return_value=mock_png_path)
# ACT
s.save(mock_png_path)
# ASSERT
assert os.path.exists(mock_png_path)