def test_delete_password(self): self.new_password.save_password() new_password = Locker("Test", "test", "passTest1") new_password.save_password() new_password.delete_password() #test if it has deleted the password self.assertEqual(len(Locker.locker_list), 1)
if __name__ == '__main__': from locker import Locker pw_manager = Locker() data = pw_manager.get_locker_data() pw_manager.print_options() selected_option = pw_manager.ask_option() while selected_option != 0: if selected_option == 1: pw_manager.print_data() elif selected_option == 2: pw_manager.add_password() elif selected_option == 3: pw_manager.edit_password() elif selected_option == 4: pw_manager.delete_password() elif selected_option == 5: pw_manager.print_options() selected_option = pw_manager.ask_option()