def test_dont_disable_apt_run_if_already_disabled(self): apt_run = AptRun() apt_run.set_folder(AptRunTests.TEST_FOLDER) result = apt_run.disable() self.assertFalse(result)
def run(args): if args['update']: init_result = run_initialize() if init_result != SUCCESS: return init_result train_result = run_train() if train_result != SUCCESS: return train_result return SUCCESS elif args['init']: return run_initialize() elif args['train']: return run_train() elif args['contribute']: collect_user_data.main() elif args['show_classifications']: show_classifications.main() elif args['enable_apt']: try: apt_run = AptRun() if apt_run.enable(): print 'AppRecommender now makes recommendations when you ' \ ' install new packages with apt' else: print 'This is already enabled' return SUCCESS except OSError: return PERMISSION_DENIED elif args['disable_apt']: try: apt_run = AptRun() if apt_run.disable(): print 'AppRecommender now dont makes recommendations when' \ 'you install new packages with apt' else: print 'This is already disabled' return SUCCESS except OSError: return PERMISSION_DENIED else: config = Config() parse_options(args, config) reference_pkgs = args['packages'] return run_apprecommender(reference_pkgs)
def test_enable_apt_run(self): apt_run = AptRun() apt_run.set_folder(AptRunTests.TEST_FOLDER) apt_run.enable() result = apt_run.is_enable() self.assertTrue(result)
def test_disable_apt_run(self): apt_run = AptRun() apt_run.set_folder(AptRunTests.TEST_FOLDER) os.makedirs(AptRunTests.TEST_FOLDER) apt_run.disable() result = apt_run.is_enable() self.assertFalse(result)
def test_pre_install_pkgs(self): apt_run = AptRun() apt_run.set_folder(AptRunTests.TEST_FOLDER) apt_run.enable() apt_run.pre_install_pkgs() user = LocalSystem() assert_pkgs = user.pkg_profile pkgs = [] with open(apt_run.installed_pkgs_file, 'r') as text: pkgs = [line.strip() for line in text] self.assertListEqual(assert_pkgs, pkgs)
def test_post_invoke(self, mock_user_pkgs): mock_user_pkgs.return_value = ['vim', 'gedit'] apt_run = AptRun() apt_run.set_folder(AptRunTests.TEST_FOLDER) apt_run.enable() with open(apt_run.installed_pkgs_file, 'w') as text: text.write('gedit') installed_pkgs = apt_run.post_invoke() self.assertEqual(['vim'], installed_pkgs)