class TestPresentation(unittest.TestCase):
def setUp(self):
self.pres_cli = CLIPresentation(CLInterface())
self.pres_gui = GUIPresentation(GUInterface())
def test_display(self):
self.pres_cli.display("Some test message")
self.pres_gui.display("Some test message")
def test_choice(self):
self.pres_cli.get_choice(["Test Action 1", "Test Action 2",
"Test Action 3"])
print self.pres_gui.get_choice(["Test Action 1", "Test Action 2",
"Test Action 3"])
class ElectionGame(Game):
"""
Main Class for the Election game
:author: James Heslin
"""
""" ATTRIBUTES
Holds the current state of the application
states (private)
Holds a reference to the current GamePresentation instance
presentation (private)
"""
def __init__(self):
self.states = [self.create_state('menu'), self.create_state('game')]
self.presentation = CLIPresentation()
#self.presentation = GUIPresentation()
def turn(self):
"""
Method to call each turn in the Election game
@return int: flag to tell the run method what to do next
@author
"""
opt = self.states[0].get_options()
#print self.states[0]
#print opt
inp = self.presentation.get_choice(opt[1:])
result = self.states[0].process_input(inp)
while (result[0] == CHOOSE_FLAG):
inp = self.presentation.get_choice(result[1:])
result = self.states[0].process_input(inp)
if (result[0] == QUIT_FLAG):
self.presentation.display(result[1:])
return -1
elif (result[0] == SWITCH_FLAG):
self.presentation.display(result[1:])
return 1
elif (result[0] == DISPLAY_FLAG):
self.presentation.display(result[1:])
return 0
def run(self):
"""
Main loop for the Election game
@author
"""
turn_flag = 0
while not (turn_flag == -1):
try:
turn_flag = self.turn()
if turn_flag == 1:
# Switch state
tmp = self.states[0]
self.states[0] = self.states[1]
self.states[1] = tmp
elif turn_flag == -1:
# Quit
exit(0)
end = self.states[0].end_turn()
if end is not None and end[0] == -1:
end.extend(["Game over, man!", "GAME OVER!"])
self.presentation.display(end)
exit(0)
self.presentation.display(self.states[0].get_state_desc())
except ValueError:
self.presentation.display(["Invalid input!"])
except IndexError:
self.presentation.display(["Invalid action!"])
def create_state(self, s_type):
"""
Create a state
@return State : an instance of a subclass of State
@author
"""
if s_type == 'game':
return GameState()
elif s_type == 'menu':
return MenuState()
def __init__(self):
self.states = [self.create_state('menu'), self.create_state('game')]
self.presentation = CLIPresentation()
def setUp(self):
self.pres_cli = CLIPresentation(CLInterface())
self.pres_gui = GUIPresentation(GUInterface())
class ElectionGame (Game):
"""
Main Class for the Election game
:author: James Heslin
"""
""" ATTRIBUTES
Holds the current state of the application
states (private)
Holds a reference to the current GamePresentation instance
presentation (private)
"""
def __init__(self):
self.states = [self.create_state('menu'), self.create_state('game')]
self.presentation = CLIPresentation()
#self.presentation = GUIPresentation()
def turn(self):
"""
Method to call each turn in the Election game
@return int: flag to tell the run method what to do next
@author
"""
opt = self.states[0].get_options()
#print self.states[0]
#print opt
inp = self.presentation.get_choice(opt[1:])
result = self.states[0].process_input(inp)
while (result[0] == CHOOSE_FLAG):
inp = self.presentation.get_choice(result[1:])
result = self.states[0].process_input(inp)
if(result[0] == QUIT_FLAG):
self.presentation.display(result[1:])
return -1
elif(result[0] == SWITCH_FLAG):
self.presentation.display(result[1:])
return 1
elif(result[0] == DISPLAY_FLAG):
self.presentation.display(result[1:])
return 0
def run(self):
"""
Main loop for the Election game
@author
"""
turn_flag = 0
while not(turn_flag == -1):
try:
turn_flag = self.turn()
if turn_flag == 1:
# Switch state
tmp = self.states[0]
self.states[0] = self.states[1]
self.states[1] = tmp
elif turn_flag == -1:
# Quit
exit(0)
end = self.states[0].end_turn()
if end is not None and end[0] == -1:
end.extend(["Game over, man!", "GAME OVER!"])
self.presentation.display(end)
exit(0)
self.presentation.display(self.states[0].get_state_desc())
except ValueError:
self.presentation.display(["Invalid input!"])
except IndexError:
self.presentation.display(["Invalid action!"])
def create_state(self, s_type):
"""
Create a state
@return State : an instance of a subclass of State
@author
"""
if s_type == 'game':
return GameState()
elif s_type == 'menu':
return MenuState()
def __init__(self):
self.states = [self.create_state('menu'), self.create_state('game')]
self.presentation = CLIPresentation()
