d4d2ad80dc
Changed filename from persistent.state to state.json.gz. Now the file extension matches the content.
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
import mock
|
|
|
|
import pykka
|
|
|
|
from mopidy.core import Core
|
|
from mopidy.internal import storage, versioning
|
|
|
|
|
|
class CoreActorTest(unittest.TestCase):
|
|
|
|
def setUp(self): # noqa: N802
|
|
self.backend1 = mock.Mock()
|
|
self.backend1.uri_schemes.get.return_value = ['dummy1']
|
|
self.backend1.actor_ref.actor_class.__name__ = b'B1'
|
|
|
|
self.backend2 = mock.Mock()
|
|
self.backend2.uri_schemes.get.return_value = ['dummy2']
|
|
self.backend2.actor_ref.actor_class.__name__ = b'B2'
|
|
|
|
self.core = Core(mixer=None, backends=[self.backend1, self.backend2])
|
|
|
|
def tearDown(self): # noqa: N802
|
|
pykka.ActorRegistry.stop_all()
|
|
|
|
def test_uri_schemes_has_uris_from_all_backends(self):
|
|
result = self.core.uri_schemes
|
|
|
|
self.assertIn('dummy1', result)
|
|
self.assertIn('dummy2', result)
|
|
|
|
def test_backends_with_colliding_uri_schemes_fails(self):
|
|
self.backend2.uri_schemes.get.return_value = ['dummy1', 'dummy2']
|
|
|
|
self.assertRaisesRegexp(
|
|
AssertionError,
|
|
'Cannot add URI scheme "dummy1" for B2, '
|
|
'it is already handled by B1',
|
|
Core, mixer=None, backends=[self.backend1, self.backend2])
|
|
|
|
def test_version(self):
|
|
self.assertEqual(self.core.version, versioning.get_version())
|
|
|
|
|
|
class CoreActorExportRestoreTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.temp_dir = tempfile.mkdtemp()
|
|
config = {
|
|
'core': {
|
|
'max_tracklist_length': 10000,
|
|
'restore_state': True,
|
|
'data_dir': self.temp_dir,
|
|
}
|
|
}
|
|
|
|
os.mkdir(os.path.join(self.temp_dir, 'core'))
|
|
|
|
self.core = Core.start(
|
|
config=config, mixer=None, backends=[]).proxy()
|
|
|
|
def tearDown(self): # noqa: N802
|
|
pykka.ActorRegistry.stop_all()
|
|
shutil.rmtree(self.temp_dir)
|
|
|
|
def test_export_state(self):
|
|
self.core.teardown().get()
|
|
# TODO: implement meaningful test
|
|
|
|
def test_restore_state(self):
|
|
self.core.setup().get()
|
|
# TODO: implement meaningful test
|
|
|
|
def test_delete_state_file_on_restore(self):
|
|
file_path = os.path.join(self.temp_dir, 'core', 'state.json.gz')
|
|
|
|
data = {}
|
|
storage.save(file_path, data)
|
|
self.assertTrue(os.path.isfile(file_path), 'missing persistent file')
|
|
|
|
self.core.setup().get()
|
|
|
|
self.assertFalse(os.path.isfile(file_path),
|
|
'persistent file has to be deleted')
|