Included gomill framework for SGF and GTP support, and sketched out SGF game-loading code.

gomill/gomill_tests/gtp_engine_tests.py

@@ -0,0 +1,56 @@
+"""Tests for gtp_engine.py"""
+
+from __future__ import with_statement
+
+from gomill import gtp_engine
+
+from gomill_tests import gomill_test_support
+from gomill_tests import gtp_engine_test_support
+from gomill_tests import test_support
+
+def make_tests(suite):
+    suite.addTests(gomill_test_support.make_simple_tests(globals()))
+
+def test_engine(tc):
+    def handle_test(args):
+        if args:
+            return "args: " + " ".join(args)
+        else:
+            return "test response"
+
+    engine = gtp_engine.Gtp_engine_protocol()
+    engine.add_protocol_commands()
+    engine.add_command('test', handle_test)
+
+    check_engine = gtp_engine_test_support.check_engine
+    check_engine(tc, engine, 'test', ['ab', 'cd'], "args: ab cd")
+
+def test_run_gtp_session(tc):
+    engine = gtp_engine.Gtp_engine_protocol()
+    engine.add_protocol_commands()
+
+    stream = "known_command list_commands\nxyzzy\nquit\n"
+    command_pipe = test_support.Mock_reading_pipe(stream)
+    response_pipe = test_support.Mock_writing_pipe()
+    gtp_engine.run_gtp_session(engine, command_pipe, response_pipe)
+    tc.assertMultiLineEqual(response_pipe.getvalue(),
+                            "= true\n\n? unknown command\n\n=\n\n")
+    command_pipe.close()
+    response_pipe.close()
+
+def test_run_gtp_session_broken_pipe(tc):
+    def break_pipe(args):
+        response_pipe.simulate_broken_pipe()
+
+    engine = gtp_engine.Gtp_engine_protocol()
+    engine.add_protocol_commands()
+    engine.add_command("break", break_pipe)
+
+    stream = "known_command list_commands\nbreak\nquit\n"
+    command_pipe = test_support.Mock_reading_pipe(stream)
+    response_pipe = test_support.Mock_writing_pipe()
+    with tc.assertRaises(gtp_engine.ControllerDisconnected) as ar:
+        gtp_engine.run_gtp_session(engine, command_pipe, response_pipe)
+    command_pipe.close()
+    response_pipe.close()
+