from unittest.mock import patch, call, mock_open from spoc import autostart from spoc import config @patch('builtins.open', new_callable=mock_open, read_data='someapp\nanotherapp\n') def test_get_apps(file_open): apps = autostart.get_apps() file_open.assert_called_once_with(config.AUTOSTART_FILE) assert apps == {'someapp', 'anotherapp'} @patch('builtins.open', side_effect=FileNotFoundError('someapp.env')) def test_get_apps_filenotfounderror(file_open): apps = autostart.get_apps() file_open.assert_called_once_with(config.AUTOSTART_FILE) assert apps == set() @patch('os.makedirs') @patch('spoc.autostart.get_apps', return_value={'someapp'}) @patch('builtins.open', new_callable=mock_open) def test_set_app_enable(file_open, get_apps, makedirs): autostart.set_app('anotherapp', True) get_apps.assert_called_once() makedirs.assert_called_once_with(config.DATA_DIR, exist_ok=True) file_open.assert_called_once_with(config.AUTOSTART_FILE, 'w') file_open().write.assert_has_calls([ call('someapp\n'), call('anotherapp\n'), ], any_order=True) @patch('os.makedirs') @patch('spoc.autostart.get_apps', return_value={'someapp', 'anotherapp'}) @patch('builtins.open', new_callable=mock_open) def test_set_app_disable(file_open, get_apps, makedirs): autostart.set_app('anotherapp', False) get_apps.assert_called_once() makedirs.assert_called_once_with(config.DATA_DIR, exist_ok=True) file_open.assert_called_once_with(config.AUTOSTART_FILE, 'w') file_open().write.assert_has_calls([ call('someapp\n'), ]) @patch('os.makedirs') @patch('spoc.autostart.get_apps', return_value={'someapp'}) @patch('builtins.open', new_callable=mock_open) def test_set_app_nonexistent(file_open, get_apps, makedirs): autostart.set_app('anotherapp', False) get_apps.assert_called_once() makedirs.assert_called_once_with(config.DATA_DIR, exist_ok=True) file_open.assert_called_once_with(config.AUTOSTART_FILE, 'w') file_open().write.assert_has_calls([ call('someapp\n'), ])