forked from test/mau_mau_bot
111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Telegram bot to play UNO in group chats
|
|
# Copyright (c) 2016 Jannes Höke <uno@jhoeke.de>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
import unittest
|
|
|
|
from telegram import User, Chat
|
|
|
|
from game_manager import GameManager
|
|
from errors import AlreadyJoinedError, LobbyClosedError, NoGameInChatError, \
|
|
NotEnoughPlayersError
|
|
|
|
|
|
class Test(unittest.TestCase):
|
|
|
|
game = None
|
|
|
|
def setUp(self):
|
|
self.gm = GameManager()
|
|
|
|
self.chat0 = Chat(0, 'group')
|
|
self.chat1 = Chat(1, 'group')
|
|
self.chat2 = Chat(2, 'group')
|
|
|
|
self.user0 = User(0, 'user0')
|
|
self.user1 = User(1, 'user1')
|
|
self.user2 = User(2, 'user2')
|
|
|
|
def test_new_game(self):
|
|
g0 = self.gm.new_game(self.chat0)
|
|
g1 = self.gm.new_game(self.chat1)
|
|
|
|
self.assertListEqual(self.gm.chatid_games[0], [g0])
|
|
self.assertListEqual(self.gm.chatid_games[1], [g1])
|
|
|
|
def test_join_game(self):
|
|
|
|
self.assertRaises(NoGameInChatError,
|
|
self.gm.join_game,
|
|
*(self.user0, self.chat0))
|
|
|
|
g0 = self.gm.new_game(self.chat0)
|
|
|
|
self.gm.join_game(self.user0, self.chat0)
|
|
self.assertEqual(len(g0.players), 1)
|
|
|
|
self.gm.join_game(self.user1, self.chat0)
|
|
self.assertEqual(len(g0.players), 2)
|
|
|
|
g0.open = False
|
|
self.assertRaises(LobbyClosedError,
|
|
self.gm.join_game,
|
|
*(self.user2, self.chat0))
|
|
|
|
g0.open = True
|
|
self.assertRaises(AlreadyJoinedError,
|
|
self.gm.join_game,
|
|
*(self.user1, self.chat0))
|
|
|
|
def test_leave_game(self):
|
|
g0 = self.gm.new_game(self.chat0)
|
|
|
|
self.gm.join_game(self.user0, self.chat0)
|
|
self.gm.join_game(self.user1, self.chat0)
|
|
|
|
self.assertRaises(NotEnoughPlayersError,
|
|
self.gm.leave_game,
|
|
*(self.user1, self.chat0))
|
|
|
|
self.gm.join_game(self.user2, self.chat0)
|
|
self.gm.leave_game(self.user0, self.chat0)
|
|
|
|
self.assertRaises(NoGameInChatError,
|
|
self.gm.leave_game,
|
|
*(self.user0, self.chat0))
|
|
|
|
def test_end_game(self):
|
|
g0 = self.gm.new_game(self.chat0)
|
|
|
|
self.gm.join_game(self.user0, self.chat0)
|
|
self.gm.join_game(self.user1, self.chat0)
|
|
|
|
self.assertEqual(len(self.gm.userid_players[0]), 1)
|
|
|
|
g1 = self.gm.new_game(self.chat0)
|
|
self.gm.join_game(self.user2, self.chat0)
|
|
|
|
self.gm.end_game(self.chat0, self.user0)
|
|
self.assertEqual(len(self.gm.chatid_games[0]), 1)
|
|
|
|
self.gm.end_game(self.chat0, self.user2)
|
|
self.assertFalse(0 in self.gm.chatid_games)
|
|
self.assertFalse(0 in self.gm.userid_players)
|
|
self.assertFalse(1 in self.gm.userid_players)
|
|
self.assertFalse(2 in self.gm.userid_players)
|