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)