fd069dff82
This commit was originally from @fredfishgames, but it needed a big rebase due to use letting it sit too long. Also, we decided not to have shebangs at the top of test files.
122 lines
2.9 KiB
Python
122 lines
2.9 KiB
Python
from zulip_bots.test_lib import StubBotTestCase
|
|
|
|
class TestChessBot(StubBotTestCase):
|
|
bot_name = "chess"
|
|
|
|
START_RESPONSE = '''New game! The board looks like this:
|
|
|
|
```
|
|
a b c d e f g h
|
|
8 ♜ ♞ ♝ ♛ ♚ ♝ ♞ ♜ 8
|
|
7 ♟ ♟ ♟ ♟ ♟ ♟ ♟ ♟ 7
|
|
6 · · · · · · · · 6
|
|
5 · · · · · · · · 5
|
|
4 · · · · · · · · 4
|
|
3 · · · · · · · · 3
|
|
2 ♙ ♙ ♙ ♙ ♙ ♙ ♙ ♙ 2
|
|
1 ♖ ♘ ♗ ♕ ♔ ♗ ♘ ♖ 1
|
|
a b c d e f g h
|
|
```
|
|
|
|
|
|
Now it's **white**'s turn.
|
|
|
|
|
|
To make your next move, respond to Chess Bot with
|
|
|
|
```do <your move>```
|
|
|
|
*Remember to @-mention Chess Bot at the beginning of your response.*'''
|
|
|
|
DO_E4_RESPONSE = '''The board was like this:
|
|
|
|
```
|
|
h g f e d c b a
|
|
1 ♖ ♘ ♗ ♔ ♕ ♗ ♘ ♖ 1
|
|
2 ♙ ♙ ♙ ♙ ♙ ♙ ♙ ♙ 2
|
|
3 · · · · · · · · 3
|
|
4 · · · · · · · · 4
|
|
5 · · · · · · · · 5
|
|
6 · · · · · · · · 6
|
|
7 ♟ ♟ ♟ ♟ ♟ ♟ ♟ ♟ 7
|
|
8 ♜ ♞ ♝ ♚ ♛ ♝ ♞ ♜ 8
|
|
h g f e d c b a
|
|
```
|
|
|
|
|
|
Then *white* moved *e4*:
|
|
|
|
```
|
|
h g f e d c b a
|
|
1 ♖ ♘ ♗ ♔ ♕ ♗ ♘ ♖ 1
|
|
2 ♙ ♙ ♙ · ♙ ♙ ♙ ♙ 2
|
|
3 · · · · · · · · 3
|
|
4 · · · ♙ · · · · 4
|
|
5 · · · · · · · · 5
|
|
6 · · · · · · · · 6
|
|
7 ♟ ♟ ♟ ♟ ♟ ♟ ♟ ♟ 7
|
|
8 ♜ ♞ ♝ ♚ ♛ ♝ ♞ ♜ 8
|
|
h g f e d c b a
|
|
```
|
|
|
|
|
|
Now it's **black**'s turn.
|
|
|
|
|
|
To make your next move, respond to Chess Bot with
|
|
|
|
```do <your move>```
|
|
|
|
*Remember to @-mention Chess Bot at the beginning of your response.*'''
|
|
|
|
DO_KE4_RESPONSE = '''Sorry, the move *Ke4* isn't legal.
|
|
|
|
```
|
|
h g f e d c b a
|
|
1 ♖ ♘ ♗ ♔ ♕ ♗ ♘ ♖ 1
|
|
2 ♙ ♙ ♙ · ♙ ♙ ♙ ♙ 2
|
|
3 · · · · · · · · 3
|
|
4 · · · ♙ · · · · 4
|
|
5 · · · · · · · · 5
|
|
6 · · · · · · · · 6
|
|
7 ♟ ♟ ♟ ♟ ♟ ♟ ♟ ♟ 7
|
|
8 ♜ ♞ ♝ ♚ ♛ ♝ ♞ ♜ 8
|
|
h g f e d c b a
|
|
```
|
|
|
|
|
|
To make your next move, respond to Chess Bot with
|
|
|
|
```do <your move>```
|
|
|
|
*Remember to @-mention Chess Bot at the beginning of your response.*'''
|
|
|
|
RESIGN_RESPONSE = '''*Black* resigned. **White** wins!
|
|
|
|
```
|
|
h g f e d c b a
|
|
1 ♖ ♘ ♗ ♔ ♕ ♗ ♘ ♖ 1
|
|
2 ♙ ♙ ♙ · ♙ ♙ ♙ ♙ 2
|
|
3 · · · · · · · · 3
|
|
4 · · · ♙ · · · · 4
|
|
5 · · · · · · · · 5
|
|
6 · · · · · · · · 6
|
|
7 ♟ ♟ ♟ ♟ ♟ ♟ ♟ ♟ 7
|
|
8 ♜ ♞ ♝ ♚ ♛ ♝ ♞ ♜ 8
|
|
h g f e d c b a
|
|
```'''
|
|
|
|
def test_bot_responds_to_empty_message(self) -> None:
|
|
with self.mock_config_info({'stockfish_location': '/foo/bar'}):
|
|
response = self.get_response(dict(content=''))
|
|
self.assertIn('play chess', response['content'])
|
|
|
|
def test_main(self) -> None:
|
|
with self.mock_config_info({'stockfish_location': '/foo/bar'}):
|
|
self.verify_dialog([
|
|
('start with other user', self.START_RESPONSE),
|
|
('do e4', self.DO_E4_RESPONSE),
|
|
('do Ke4', self.DO_KE4_RESPONSE),
|
|
('resign', self.RESIGN_RESPONSE),
|
|
])
|