Refactor: Make StateHandler() function independent.
This refactor makes the nested class 'StateHandler' in the file /contrib_bots/bot_lib.py independent class. It previously was nested in 'run_message_handler_for_bot' function. This is done to write a cleaner test file for contrib_bots using mock library.
This commit is contained in:
		
							parent
							
								
									a8b825388d
								
							
						
					
					
						commit
						63efed5c73
					
				
					 1 changed files with 10 additions and 10 deletions
				
			
		| 
						 | 
				
			
			@ -71,6 +71,16 @@ class BotHandlerApi(object):
 | 
			
		|||
                content=response,
 | 
			
		||||
            ))
 | 
			
		||||
 | 
			
		||||
class StateHandler(object):
 | 
			
		||||
    def __init__(self):
 | 
			
		||||
        self.state = None
 | 
			
		||||
 | 
			
		||||
    def set_state(self, state):
 | 
			
		||||
        self.state = state
 | 
			
		||||
 | 
			
		||||
    def get_state(self):
 | 
			
		||||
        return self.state
 | 
			
		||||
 | 
			
		||||
def run_message_handler_for_bot(lib_module, quiet, config_file):
 | 
			
		||||
    # Make sure you set up your ~/.zuliprc
 | 
			
		||||
    client = Client(config_file=config_file)
 | 
			
		||||
| 
						 | 
				
			
			@ -78,16 +88,6 @@ def run_message_handler_for_bot(lib_module, quiet, config_file):
 | 
			
		|||
 | 
			
		||||
    message_handler = lib_module.handler_class()
 | 
			
		||||
 | 
			
		||||
    class StateHandler(object):
 | 
			
		||||
        def __init__(self):
 | 
			
		||||
            self.state = None
 | 
			
		||||
 | 
			
		||||
        def set_state(self, state):
 | 
			
		||||
            self.state = state
 | 
			
		||||
 | 
			
		||||
        def get_state(self):
 | 
			
		||||
            return self.state
 | 
			
		||||
 | 
			
		||||
    state_handler = StateHandler()
 | 
			
		||||
 | 
			
		||||
    if not quiet:
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue