beeref/tests/fileio/test_init.py
2021-03-27 21:55:47 +01:00

21 lines
625 B
Python

import os.path
import tempfile
from unittest.mock import patch
from beeref import fileio
@patch('beeref.fileio.sql.SQLiteIO.write')
def test_save_create_new_false(write_mock):
with tempfile.TemporaryDirectory() as dirname:
fname = os.path.join(dirname, 'test.bee')
fileio.save(fname, 'myscene', create_new=False)
write_mock.assert_called_once()
@patch('beeref.fileio.sql.SQLiteIO.read')
def test_write(read_mock):
with tempfile.TemporaryDirectory() as dirname:
fname = os.path.join(dirname, 'test.bee')
fileio.load(fname, 'myscene')
read_mock.assert_called_once()