def test_with_uasts_rpc_error(self): called = False def func(imposter, ptr: ReferencePointer, config: dict, data_service: DataService, **data): nonlocal called called = True def fail(f): def wrapped(): f() self.assertIsNotNone( self.data_service._data_request_local.channel) raise grpc.RpcError() return wrapped self.data_service._get_channel = fail(self.data_service._get_channel) func = with_uasts(unicode=False)(func) self.assertRaises(grpc.RpcError, func, self, ReferencePointer(self.url, self.ref, self.COMMIT_TO), None, self.data_service) self.assertFalse(called) self.assertIsNone(self.data_service._data_request_local.channel)
def test_with_uasts(self): def func(imposter, ptr: ReferencePointer, config: dict, data_service: DataService, **data): self.assertIsInstance(data_service, DataService) files = list(data["files"]) self.assertEqual(len(files), 18) for file in files: self.assertEqual(file.content, b"") self.assertEqual( type(file.uast).__module__, bblfsh.Node.__module__) self.assertTrue(file.path) self.assertIn(file.language, ("Python", "YAML", "Dockerfile", "Markdown", "Jupyter Notebook", "Shell", "Text", "")) func = with_uasts(unicode=False)(func) func(self, ReferencePointer(self.url, self.ref, self.COMMIT_TO), None, self.data_service)
def test_with_uasts(self): def func(imposter, ptr: ReferencePointer, config: dict, data_request_stub: DataStub, **data): files = list(data["files"]) self.assertEqual(len(files), 61) for file in files: self.assertEqual(file.content, b"") self.assertEqual( type(file.uast).__module__, bblfsh.Node.__module__) self.assertTrue(file.path) self.assertIn(file.language, ("Python", "YAML", "Dockerfile", "Markdown", "Jupyter Notebook", "Shell", "Text", "")) func = with_uasts(func) func( self, ReferencePointer(self.url, self.ref, "5833b4ba94154cf1ed07f37c32928c7b4411b36b"), None, self.data_service.get())