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_blank_subscription(self, listen_mock):
# Arrange
app = Mock(ui_methods = {}, ui_modules = {}, async_db = {"tests": None})
handler = SubscriptionHandler(app, Mock())
handler._manager = Mock()
# Act
handler.open()
# Assert
self.assertFalse(listen_mock.called)
self.assertFalse(handler.listening)
def test_blank_subscription(self, listen_mock):
# Arrange
app = Mock(ui_methods={}, ui_modules={}, async_db={"tests": None})
handler = SubscriptionHandler(app, Mock())
handler._manager = Mock()
# Act
handler.open()
# Assert
self.assertFalse(listen_mock.called)
self.assertFalse(handler.listening)
def test_resource_id_subscription(self):
# Arrange
app = Mock(ui_methods = {}, ui_modules = {}, async_db = {"tests": None})
resource_mock = Mock()
resource_mock.arguments = dict()
handler = SubscriptionHandler(app, resource_mock)
handler._manager = Mock()
# Act
handler.open(resource_type = "test", resource_id="1")
# Assert
self.assertEqual(handler._manager.createChannel.call_args[0][0], { "id": "1" })
def test_resource_id_subscription(self):
# Arrange
app = Mock(ui_methods={}, ui_modules={}, async_db={"tests": None})
resource_mock = Mock()
resource_mock.arguments = dict()
handler = SubscriptionHandler(app, resource_mock)
handler._manager = Mock()
# Act
handler.open(resource_type="test", resource_id="1")
# Assert
self.assertEqual(handler._manager.createChannel.call_args[0][0],
{"id": "1"})
def test_resource_subscription(self, listen_mock):
# Arrange
app = Mock(ui_methods = {}, ui_modules = {}, async_db = {"tests": None})
resource_mock = Mock()
resource_mock.arguments = dict()
handler = SubscriptionHandler(app, resource_mock)
handler._manager = Mock()
# Act
handler.open(resource_type = "test")
# Assert
self.assertTrue(listen_mock.called)
self.assertEqual(handler._manager.createChannel.call_count, 1)
self.assertEqual(handler._manager.createChannel.call_args[0][1], "test")
def test_subscription_good_fields(self):
# Arrange
app = Mock(ui_methods = {}, ui_modules = {}, async_db = {"tests": None})
fields_request = Mock()
fields_request.arguments = dict(fields = ["name,ts"])
manager_mock = Mock()
handler = SubscriptionHandler(app, fields_request)
handler._manager = manager_mock
# Act
handler.open(resource_type = "test")
# Assert
self.assertEqual(manager_mock.createChannel.call_args[0][2], ["name", "ts"])
def test_resource_subscription(self, listen_mock):
# Arrange
app = Mock(ui_methods={}, ui_modules={}, async_db={"tests": None})
resource_mock = Mock()
resource_mock.arguments = dict()
handler = SubscriptionHandler(app, resource_mock)
handler._manager = Mock()
# Act
handler.open(resource_type="test")
# Assert
self.assertTrue(listen_mock.called)
self.assertEqual(handler._manager.createChannel.call_count, 1)
self.assertEqual(handler._manager.createChannel.call_args[0][1],
"test")
def test_subscription_good_fields(self):
# Arrange
app = Mock(ui_methods={}, ui_modules={}, async_db={"tests": None})
fields_request = Mock()
fields_request.arguments = dict(fields=["name,ts"])
manager_mock = Mock()
handler = SubscriptionHandler(app, fields_request)
handler._manager = manager_mock
# Act
handler.open(resource_type="test")
# Assert
self.assertEqual(manager_mock.createChannel.call_args[0][2],
["name", "ts"])
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)