def test_emit_with_callback_without_server(self): standalone_pm = pubsub_manager.PubSubManager() self.assertRaises(RuntimeError, standalone_pm.emit, 'foo', 'bar', callback='cb')
def test_write_only_init(self): mock_server = mock.MagicMock() pm = pubsub_manager.PubSubManager(write_only=True) pm.initialize(mock_server) self.assertEqual(pm.channel, 'socketio') self.assertEqual(len(pm.host_id), 32) self.assertEqual(pm.server.start_background_task.call_count, 0)
def test_write_only_with_provided_logger(self): test_logger = logging.getLogger('new_logger') pm = pubsub_manager.PubSubManager(write_only=True, logger=test_logger) pm.initialize() assert pm.channel == 'socketio' assert len(pm.host_id) == 32 assert pm._get_logger() == test_logger
def test_write_only_with_provided_logger(self): test_logger = logging.getLogger('new_logger') pm = pubsub_manager.PubSubManager(write_only=True, logger=test_logger) pm.initialize() self.assertEqual(pm.channel, 'socketio') self.assertEqual(len(pm.host_id), 32) self.assertEqual(pm._get_logger(), test_logger)
def setUp(self): mock_server = mock.MagicMock() self.pm = pubsub_manager.PubSubManager() self.pm._publish = mock.MagicMock() self.pm.set_server(mock_server) self.pm.host_id = '123456' self.pm.initialize()
def test_write_only_init(self): mock_server = mock.MagicMock() pm = pubsub_manager.PubSubManager(write_only=True) pm.set_server(mock_server) pm.initialize() assert pm.channel == 'socketio' assert len(pm.host_id) == 32 assert pm.server.start_background_task.call_count == 0
def setUp(self): id = 0 def generate_id(): nonlocal id id += 1 return str(id) mock_server = mock.MagicMock() mock_server.eio.generate_id = generate_id self.pm = pubsub_manager.PubSubManager() self.pm._publish = mock.MagicMock() self.pm.set_server(mock_server) self.pm.host_id = '123456' self.pm.initialize()
def test_custom_init(self): pubsub = pubsub_manager.PubSubManager(channel='foo') self.assertEqual(pubsub.channel, 'foo') self.assertEqual(len(pubsub.host_id), 32)
def setUp(self): mock_server = mock.MagicMock() self.pm = pubsub_manager.PubSubManager() self.pm._publish = mock.MagicMock() self.pm.initialize(mock_server)
def test_write_only_default_logger(self): pm = pubsub_manager.PubSubManager(write_only=True) pm.initialize() self.assertEqual(pm.channel, 'socketio') self.assertEqual(len(pm.host_id), 32) self.assertEqual(pm._get_logger(), logging.getLogger('socketio'))
def test_write_only_default_logger(self): pm = pubsub_manager.PubSubManager(write_only=True) pm.initialize() assert pm.channel == 'socketio' assert len(pm.host_id) == 32 assert pm._get_logger() == logging.getLogger('socketio')
def test_custom_init(self): pubsub = pubsub_manager.PubSubManager(channel='foo') assert pubsub.channel == 'foo' assert len(pubsub.host_id) == 32
def test_emit_with_callback_without_server(self): standalone_pm = pubsub_manager.PubSubManager() with pytest.raises(RuntimeError): standalone_pm.emit('foo', 'bar', callback='cb')