def test_manage_no_params(self): self.init_ptah() sys.argv[:] = ['ptah-manage', 'ptah.ini'] stdout = sys.stdout out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('[-h] [--list-modules] [--list-models] config', val)
def test_manage_no_params(self): self.init_ptah() sys.argv[:] = ['ptah-manage', 'ptah.ini'] stdout = sys.stdout out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn( '[-h] [--list-modules] [--list-models] config', val)
def test_list_models(self): @ptah.tinfo( 'custom', title='Custom model', description = 'Custom model description') class CustomModel(object): """ Custom module description """ title = 'Custom Module' self.init_ptah() sys.argv[1:] = ['--list-models', 'ptah.ini'] stdout = sys.stdout out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* type:custom: Custom model (disabled: False)', val) self.assertIn('Custom model description', val) self.assertIn('class: CustomModel', val) self.assertIn('module: test_manage', val) # disable cfg = ptah.get_settings(ptah.CFG_ID_PTAH) cfg['disable_models'] = ('type:custom',) out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* type:custom: Custom model (disabled: True)', val)
def test_list_models(self): @ptah.tinfo('custom', title='Custom model', description='Custom model description') class CustomModel(object): """ Custom module description """ title = 'Custom Module' self.init_ptah() sys.argv[1:] = ['--list-models', 'ptah.ini'] stdout = sys.stdout out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* type:custom: Custom model (disabled: False)', val) self.assertIn('Custom model description', val) self.assertIn('class: CustomModel', val) self.assertIn('module: test_manage', val) # disable cfg = ptah.get_settings(ptah.CFG_ID_PTAH) cfg['disable_models'] = ('type:custom', ) out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* type:custom: Custom model (disabled: True)', val)
def test_list_modules(self): @ptah.manage.module('custom') class CustomModule(ptah.manage.PtahModule): """ Custom module description """ title = 'Custom Module' self.init_ptah() sys.argv[1:] = ['--list-modules', 'ptah.ini'] stdout = sys.stdout out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* custom: Custom Module (disabled: False)', val) self.assertIn('Custom module description', val) # disable cfg = ptah.get_settings(ptah.CFG_ID_PTAH) cfg['disable_modules'] = ('custom',) out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* custom: Custom Module (disabled: True)', val)
def test_list_modules(self): @ptah.manage.module('custom') class CustomModule(ptah.manage.PtahModule): """ Custom module description """ title = 'Custom Module' self.init_ptah() sys.argv[1:] = ['--list-modules', 'ptah.ini'] stdout = sys.stdout out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* custom: Custom Module (disabled: False)', val) self.assertIn('Custom module description', val) # disable cfg = ptah.get_settings(ptah.CFG_ID_PTAH) cfg['disable_modules'] = ('custom', ) out = NativeIO() sys.stdout = out manage.main(False) sys.stdout = stdout val = out.getvalue() self.assertIn('* custom: Custom Module (disabled: True)', val)