def test_get_existing_session(self):
store = SessionStore(self.datadir, self.hub)
# Create a new session.
new_session = store.get('foo')
# Get the same existing session again.
got_session = store.get('foo')
self.assertEqual(new_session, got_session)
def __init__(self, path, managers):
MultiService.__init__(self)
self.path = path
self.managers = managers
self.servers = set()
self.session_store = SessionStore(path, self)
for signal in self.signals:
self.wrapSignal(signal)
for host, port, password in managers:
self.add_server(host, port, password)
def setUp(self):
self.datadir = _unique_dir_name()
self.hub = mock.MagicMock()
self.store = SessionStore(self.datadir, self.hub)
self.datavault = server.DataVault(self.store)
self.set_default_labrad_server_mocks(self.datavault)
self.context = MockContext()
self.datavault.initServer()
def start():
opts = labrad.util.parseServerOptions(name=DataVault.name)
cxn = yield labrad.wrappers.connectAsync(
host=opts['host'], port=int(opts['port']), password=opts['password'])
datadir = yield load_settings(cxn, opts['name'])
yield cxn.disconnect()
session_store = SessionStore(datadir, hub=None)
server = DataVault(session_store)
session_store.hub = server
# Run the server. We do not need to start the reactor, but we will
# stop it after the data_vault shuts down.
labrad.util.runServer(server, run_reactor=False, stop_reactor=True)
def test_get_all_sessions(self):
store = SessionStore(self.datadir, self.hub)
# Create a new session.
foo_session = store.get('foo')
bar_session = store.get('bar')
self.assertEqual([foo_session, bar_session], store.get_all())
def test_get_new_session(self):
store = SessionStore(self.datadir, self.hub)
self.assertFalse(store.exists('foo'))
store.get('foo')
self.assertTrue(store.exists('foo'))
