def test_subscription_good_query(self):
# Arrange
app = Mock(ui_methods={}, ui_modules={}, async_db={"tests": None})
query_request = Mock()
query_request.arguments = dict(
query=['{ "name": "test", "ts": { "gte": 145 } }'])
handler = SubscriptionHandler(app, query_request)
manager_mock = Mock()
handler._manager = Mock()
handler.write_message = Mock(side_effects=ValueError)
handler._manager = manager_mock
# Act
handler.open(resource_type="test")
# Assert
self.assertTrue(manager_mock.createChannel.called)
self.assertEqual(manager_mock.createChannel.call_args[0][0], {
"name": "test",
"ts": {
"gte": 145
}
})
def test_subscription_bad_query(self):
# Arrange
app = Mock(ui_methods = {}, ui_modules = {}, async_db = { "tests": None})
query_request = Mock()
query_request.arguments = dict(query = ["{ name: 'test', ts: { gte: 145 }"])
manager_mock = Mock()
handler = SubscriptionHandler(app, query_request)
handler._manager = manager_mock
handler.write_message = Mock(side_effects = ValueError)
# Act
handler.open(resource_type = "test")
# Assert
self.assertFalse(handler.client)
self.assertFalse(manager_mock.createChannel.called)
def test_subscription_good_query(self):
# Arrange
app = Mock(ui_methods = {}, ui_modules = {}, async_db = {"tests": None})
query_request = Mock()
query_request.arguments = dict(query = ['{ "name": "test", "ts": { "gte": 145 } }'])
handler = SubscriptionHandler(app, query_request)
manager_mock = Mock()
handler._manager = Mock()
handler.write_message = Mock(side_effects = ValueError)
handler._manager = manager_mock
# Act
handler.open(resource_type = "test")
# Assert
self.assertTrue(manager_mock.createChannel.called)
self.assertEqual(manager_mock.createChannel.call_args[0][0], { "name": "test", "ts": { "gte": 145 } })
def test_subscription_bad_query(self):
# Arrange
app = Mock(ui_methods={}, ui_modules={}, async_db={"tests": None})
query_request = Mock()
query_request.arguments = dict(
query=["{ name: 'test', ts: { gte: 145 }"])
manager_mock = Mock()
handler = SubscriptionHandler(app, query_request)
handler._manager = manager_mock
handler.write_message = Mock(side_effects=ValueError)
# Act
handler.open(resource_type="test")
# Assert
self.assertFalse(handler.client)
self.assertFalse(manager_mock.createChannel.called)
