def test_render_error_to_thrift(self): self.assertEqual( types.RenderError( types.I18nMessage("foo", {}), [ types.QuickFix( types.I18nMessage("click"), types.QuickFixAction.PrependStep("filter", {"x": "y"}), ) ], ).to_thrift(), ttypes.RenderError( ttypes.I18nMessage("foo", {}, ttypes.I18nMessageSource()), [ ttypes.QuickFix( ttypes.I18nMessage("click", {}, ttypes.I18nMessageSource()), ttypes.QuickFixAction( prepend_step=ttypes.PrependStepQuickFixAction( "filter", ttypes.RawParams('{"x":"y"}') ) ), ) ], ), )
def test_prepend_step_quick_fix_action_to_thrift(self): self.assertEqual( types.arrow_quick_fix_action_to_thrift( types.QuickFixAction.PrependStep("filter", {"x": "y"})), ttypes.QuickFixAction( prepend_step=ttypes.PrependStepQuickFixAction( "filter", ttypes.RawParams('{"x":"y"}'))), )
def test_raw_params_to_thrift(self): self.assertEqual( types.RawParams({ "A": "x", "B": [1, 2] }).to_thrift(), ttypes.RawParams('{"A":"x","B":[1,2]}'), )
def test_raw_params_from_thrift(self): self.assertEqual( types.thrift_raw_params_to_arrow( ttypes.RawParams('{"A":"x","B":[1,2]}')), types.RawParams({ "A": "x", "B": [1, 2] }), )
def test_quick_fix_to_thrift(self): self.assertEqual( types.QuickFix( types.I18nMessage("click"), types.QuickFixAction.PrependStep("filter", {"x": "y"}), ).to_thrift(), ttypes.QuickFix( ttypes.I18nMessage("click", {}), ttypes.QuickFixAction( prepend_step=ttypes.PrependStepQuickFixAction( "filter", ttypes.RawParams('{"x":"y"}'))), ), )
def migrate_params(self, compiled_module: CompiledModule, params: Dict[str, Any]) -> None: request = RawParams(params).to_thrift() with _chroot_dir_context() as chroot: response = self._run_in_child( chroot=chroot, chroot_paths=DATA_PATHS, compiled_module=compiled_module, timeout=self.migrate_params_timeout, result=ttypes.RawParams(), function="migrate_params_thrift", args=[request], ) return RawParams.from_thrift(response).params
def migrate_params(self, compiled_module: CompiledModule, params: Dict[str, Any]) -> None: """Call a module's migrate_params().""" request = arrow_raw_params_to_thrift(RawParams(params)) response = self._run_in_child( chroot_dir=READONLY_CHROOT_DIR, network_config=None, compiled_module=compiled_module, timeout=self.migrate_params_timeout, result=ttypes.RawParams(), function="migrate_params_thrift", args=[request], ) return thrift_raw_params_to_arrow(response).params