class ClassToJsUnitTest(SimpleRpcLogicBase):
    '''
    #TODO: document
    '''
    def __post_init__(self):
        self.class_to_js = ClassToJs(self.context)
        self.templates = TemplatesCollector(self.context)
        self.file_manager = FileManager(self.context)

    def translateClass(self, class_):
        '''  '''
        ast_tree = self.class_to_js.translateClass(class_)
        templates_set = 'js_unittest_templates'
        templates = self.templates.collectBuiltIn(templates_set)
        js_str = ast_tree.getString(templates)
        return js_str

    def translateToFile(self, tested_class, file_path, overwrite=False):
        js_str = self.__getClassJsUnit(tested_class)
        #create directory just in case
        test_dir = os.path.dirname(file_path)
        self.file_manager.makedirs(test_dir)
        #Say where it is in a pretty way
        path = self.file_manager.formatFilePath(file_path)
        name = tested_class.__name__
        self.log.d('Creating js for %s test at:\n %s' % (name, path))
        #save file
        self.file_manager.saveTextFile(file_path, js_str, overwrite)

    def __getClassJsUnit(self, tested_class):
        js_str = self.translateClass(tested_class)
        translate = dict(EXPOSED_RPC_API_CLASS=self.context.js_rpc_file)
        js_str = Template(js_str).safe_substitute(translate)
        return js_str
class RPCJavascriptGenerator(SimpleRpcLogicBase):
    def __post_init__(self):
        self.file_manager = FileManager(self.context)
        self.package_translator = PackageToJs(self.context)
        self.templates_collector = TemplatesCollector(self.context)
        self.js_util = JsTranslateUtil(self.context)

    def getRpcNode(self, packages):
        packages_node = AutoTemplateAstNode()
        for p in packages:
            n = self.package_translator.translatePackage(p)
            name = p.__name__.split('.')[-1]
            packages_node.translate(name, n)
        exposed_rpc_node = TranslationAstNode('exposed_rpc.CommandQueueApi')
        exposed_rpc_node.translate(EXPOSED_PACKAGES=packages_node)
        return exposed_rpc_node

    def translateToFile(self, packages, js_rpc_file=None, templates=None,
                        overwrite=False):
        js_rpc_file = self.js_util._getJsRpcFile(js_rpc_file)
        if not templates:
            templates = self.templates_collector.collectBuiltIn()
        text = self.getRpcNode(packages).getString(templates)
        self.file_manager.saveTextFile(js_rpc_file, text, overwrite)
 def __post_init__(self):
     self.file_manager = FileManager(self.context)
     self.package_translator = PackageToJs(self.context)
     self.templates_collector = TemplatesCollector(self.context)
     self.js_util = JsTranslateUtil(self.context)
 def __post_init__(self):
     self.class_to_js = ClassToJs(self.context)
     self.templates = TemplatesCollector(self.context)
     self.file_manager = FileManager(self.context)