1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
|
from unittest.mock import patch
from kiwi.utils.temporary import Temporary
class TestTemporary:
def setup(self):
self.temporary = Temporary()
def setup_method(self, cls):
self.setup()
@patch('kiwi.utils.temporary.NamedTemporaryFile')
def test_new_file(self, mock_NamedTemporaryFile):
self.temporary.new_file()
mock_NamedTemporaryFile.assert_called_once_with(
dir='/var/tmp', prefix='kiwi_'
)
@patch('kiwi.utils.temporary.TemporaryDirectory')
def test_new_dir(self, mock_TemporaryDirectory):
self.temporary.new_dir()
mock_TemporaryDirectory.assert_called_once_with(
dir='/var/tmp', prefix='kiwi_'
)
|