class TicTacToeUserHandler(UserHandler):
def __init__(self, send_message: Callable[[str], None]) -> None:
super(TicTacToeUserHandler, self).__init__(send_message)
self.game: Optional[TicTacToe] = None
def handle_message(self, message: str) -> None:
command = message.rstrip('\n')
if command == 'start':
self.start_game()
return
if not self.game:
self.send_message('Game is not started')
return
name, col, row = command.rstrip('\n').split(maxsplit=2)
if name == 'O':
self.make_turn(player=Player.O, row=int(row), col=int(col))
elif name == 'X':
self.make_turn(player=Player.X, row=int(row), col=int(col))
def start_game(self) -> None:
self.game = None
self.game = TicTacToe()
self.send_field()
def make_turn(self, player: Player, *, row: int, col: int) -> None:
assert self.game
if not self.game.can_make_turn(player, row=row, col=col):
self.send_message('Invalid turn')
return
else:
self.game.make_turn(player, row=row, col=col)
self.send_field()
if self.game.is_finished():
if self.game.current_player() == Player.X:
self.send_message('Game is finished, O wins')
elif self.game.current_player() == Player.O:
self.send_message('Game is finished, X wins')
else:
self.send_message('Game is finished, draw')
self.game = None
def send_field(self) -> None:
assert self.game
field_print = ''
for column in self.game.field:
for row in column:
if row:
field_print += row.name
else:
field_print += '.'
field_print += '\n'
self.send_message(field_print.rstrip('\n'))
def test_current_player(self):
global player1
global player2
global click
self.assertTrue(True, TicTacToe.current_player(player1))
def test_game_draw() -> None:
game = TicTacToe()
assert not game.is_finished()
assert game.winner() is None
assert game.current_player() == Player.X
assert game.can_make_turn(Player.X, col=0, row=0)
assert not game.can_make_turn(Player.O, col=0, row=0)
game.make_turn(Player.X, col=0, row=0)
assert not game.is_finished()
assert game.winner() is None
assert not game.can_make_turn(Player.X, col=0, row=0)
assert not game.can_make_turn(Player.O, col=0, row=0)
assert game.current_player() == Player.O
assert not game.can_make_turn(Player.X, col=1, row=0)
assert game.can_make_turn(Player.O, col=1, row=0)
game.make_turn(Player.O, col=1, row=0)
assert not game.is_finished()
assert game.winner() is None
assert not game.can_make_turn(Player.X, col=0, row=0)
assert not game.can_make_turn(Player.O, col=0, row=0)
assert not game.can_make_turn(Player.X, col=1, row=0)
assert not game.can_make_turn(Player.O, col=1, row=0)
assert game.current_player() == Player.X
assert game.can_make_turn(Player.X, col=2, row=0)
assert not game.can_make_turn(Player.O, col=2, row=0)
game.make_turn(Player.X, col=2, row=0)
assert not game.is_finished()
assert game.winner() is None
# XOX
# ...
# ...
assert game.current_player() == Player.O
assert game.can_make_turn(Player.O, col=0, row=1)
game.make_turn(Player.O, col=0, row=1)
assert not game.is_finished()
assert game.winner() is None
# XOX
# O..
# ...
assert game.current_player() == Player.X
assert game.can_make_turn(Player.X, col=2, row=1)
game.make_turn(Player.X, col=2, row=1)
assert not game.is_finished()
assert game.winner() is None
# XOX
# O.X
# ...
assert game.current_player() == Player.O
assert game.can_make_turn(Player.O, col=1, row=1)
game.make_turn(Player.O, col=1, row=1)
assert not game.is_finished()
assert game.winner() is None
# XOX
# OOX
# ...
assert game.current_player() == Player.X
assert game.can_make_turn(Player.X, col=1, row=2)
game.make_turn(Player.X, col=1, row=2)
assert not game.is_finished()
assert game.winner() is None
# XOX
# OOX
# .X.
assert game.current_player() == Player.O
assert game.can_make_turn(Player.O, col=2, row=2)
game.make_turn(Player.O, col=2, row=2)
assert not game.is_finished()
assert game.winner() is None
# XOX
# OOX
# .XO
assert game.current_player() == Player.X
assert game.can_make_turn(Player.X, col=0, row=2)
game.make_turn(Player.X, col=0, row=2)
# XOX
# OOX
# XXO
assert game.is_finished()
assert game.winner() is None
assert game.current_player() is None
