def test_get_data(self):
open_mock = mock_open(read_data='hallo:\n test: world\n')
with patch('mpf.file_interfaces.yaml_interface.open', open_mock, create=True):
with patch('mpf.core.file_manager.os.path.isfile') as isfile_mock:
manager = DataManager(self.machine, "machine_vars", min_wait_secs=0)
self.assertTrue(isfile_mock.called)
self.assertTrue(open_mock.called)
self.assertEqual({"test": "world"}, manager.get_data("hallo"))
self.assertEqual({}, manager.get_data("invalid"))
def test_save_and_load(self):
open_mock = mock_open(read_data="")
with patch('mpf.file_interfaces.yaml_interface.open', open_mock, create=True):
with patch('mpf.core.file_manager.os.path.isfile') as isfile_mock:
isfile_mock.return_value = True
manager = DataManager(self.machine, "machine_vars", min_wait_secs=0)
self.assertTrue(isfile_mock.called)
self.assertTrue(open_mock.called)
self.assertNotIn("hallo", manager.get_data())
open_mock = mock_open(read_data="")
with patch('mpf.file_interfaces.yaml_interface.open', open_mock, create=True):
with patch('mpf.core.data_manager.os.replace') as move_mock:
manager.save_all({"hallo": "world"})
while not move_mock.called:
time.sleep(.00001)
open_mock().write.assert_called_once_with('hallo: world\n')
self.assertTrue(move_mock.called)
open_mock = mock_open(read_data='hallo: world\n')
with patch('mpf.file_interfaces.yaml_interface.open', open_mock, create=True):
with patch('mpf.core.file_manager.os.path.isfile') as isfile_mock:
isfile_mock.return_value = True
manager2 = DataManager(self.machine, "machine_vars", min_wait_secs=0)
self.assertTrue(isfile_mock.called)
self.assertTrue(open_mock.called)
self.assertEqual("world", manager2.get_data()["hallo"])
self.assertEqual({}, manager.get_data("hallo"))
def create_data_manager(self, config_name: str) -> DataManager: # pragma: no cover
"""Return a new DataManager for a certain config.
Args:
config_name: Name of the config
"""
return DataManager(self, config_name)
def test_paths_and_disable(self):
open_mock = mock_open(read_data='hallo:\n test: world\n')
with patch('mpf.file_interfaces.yaml_interface.open',
open_mock,
create=True):
with patch('mpf.core.file_manager.os.path.isfile') as isfile_mock:
with patch('mpf.core.file_manager.os.mkdir') as mkdir_mock:
manager = DataManager(self.machine, "absolute_test")
isfile_mock.assert_called_with(
'/data/test_dir/test_file.yaml')
mkdir_mock.assert_called_with('/data/test_dir', 511)
open_mock.assert_called_once_with(
'/data/test_dir/test_file.yaml', encoding='utf8')
self.assertEqual({"test": "world"}, manager.get_data("hallo"))
self.assertEqual({}, manager.get_data("invalid"))
open_mock = mock_open(read_data='hallo:\n test: world\n')
with patch('mpf.file_interfaces.yaml_interface.open',
open_mock,
create=True):
with patch('mpf.core.file_manager.os.path.isfile') as isfile_mock:
with patch('mpf.core.file_manager.os.mkdir') as mkdir_mock:
manager = DataManager(self.machine, "relative_test")
path = os.path.join(
os.path.abspath(self.machine.machine_path),
'subdir/subdir2')
file_path = os.path.join(
os.path.abspath(self.machine.machine_path),
'subdir/subdir2/test.yaml')
isfile_mock.assert_called_with(file_path)
mkdir_mock.assert_called_with(path, 511)
open_mock.assert_called_once_with(file_path,
encoding='utf8')
self.assertEqual({"test": "world"}, manager.get_data("hallo"))
self.assertEqual({}, manager.get_data("invalid"))
open_mock = mock_open(read_data='hallo:\n test: world\n')
with patch('mpf.file_interfaces.yaml_interface.open',
open_mock,
create=True):
with patch('mpf.core.file_manager.os.path.isfile') as isfile_mock:
with patch('mpf.core.file_manager.os.mkdir') as mkdir_mock:
manager = DataManager(self.machine, "disabled_test")
self.assertFalse(isfile_mock.called)
self.assertFalse(mkdir_mock.called)
self.assertFalse(open_mock.called)
self.assertEqual({}, manager.get_data("hallo"))
self.assertEqual({}, manager.get_data("invalid"))