test_frontendbase.py
114 lines
| 3.2 KiB
| text/x-python
|
PythonLexer
Barry Wark
|
r1263 | # encoding: utf-8 | ||
"""This file contains unittests for the frontendbase module.""" | ||||
__docformat__ = "restructuredtext en" | ||||
#------------------------------------------------------------------------------- | ||||
# Copyright (C) 2008 The IPython Development Team | ||||
# | ||||
# Distributed under the terms of the BSD License. The full license is in | ||||
# the file COPYING, distributed as part of this software. | ||||
#------------------------------------------------------------------------------- | ||||
#------------------------------------------------------------------------------- | ||||
# Imports | ||||
#------------------------------------------------------------------------------- | ||||
import unittest | ||||
from IPython.frontend import frontendbase | ||||
from IPython.kernel.engineservice import EngineService | ||||
class FrontEndCallbackChecker(frontendbase.FrontEndBase): | ||||
"""FrontEndBase subclass for checking callbacks""" | ||||
def __init__(self, engine=None, history=None): | ||||
super(FrontEndCallbackChecker, self).__init__(engine=engine, history=history) | ||||
self.updateCalled = False | ||||
self.renderResultCalled = False | ||||
self.renderErrorCalled = False | ||||
def update_cell_prompt(self, result): | ||||
self.updateCalled = True | ||||
return result | ||||
def render_result(self, result): | ||||
self.renderResultCalled = True | ||||
return result | ||||
def render_error(self, failure): | ||||
self.renderErrorCalled = True | ||||
return failure | ||||
class TestFrontendBase(unittest.TestCase): | ||||
def setUp(self): | ||||
"""Setup the EngineService and FrontEndBase""" | ||||
self.fb = FrontEndCallbackChecker(engine=EngineService()) | ||||
def test_implementsIFrontEnd(self): | ||||
assert(frontendbase.IFrontEnd.implementedBy(frontendbase.FrontEndBase)) | ||||
def test_is_completeReturnsFalseForIncompleteBlock(self): | ||||
"""""" | ||||
block = """def test(a):""" | ||||
assert(self.fb.is_complete(block) == False) | ||||
def test_is_completeReturnsTrueForCompleteBlock(self): | ||||
"""""" | ||||
block = """def test(a): pass""" | ||||
assert(self.fb.is_complete(block)) | ||||
block = """a=3""" | ||||
assert(self.fb.is_complete(block)) | ||||
def test_blockIDAddedToResult(self): | ||||
block = """3+3""" | ||||
d = self.fb.execute(block, blockID='TEST_ID') | ||||
d.addCallback(self.checkBlockID, expected='TEST_ID') | ||||
def checkBlockID(self, result, expected=""): | ||||
assert(result['blockID'] == expected) | ||||
def test_callbacksAddedToExecuteRequest(self): | ||||
"""test that | ||||
update_cell_prompt | ||||
render_result | ||||
are added to execute request | ||||
""" | ||||
d = self.fb.execute("10+10") | ||||
d.addCallback(self.checkCallbacks) | ||||
def checkCallbacks(self, result): | ||||
assert(self.fb.updateCalled) | ||||
assert(self.fb.renderResultCalled) | ||||
def test_errorCallbackAddedToExecuteRequest(self): | ||||
"""test that render_error called on execution error""" | ||||
d = self.fb.execute("raise Exception()") | ||||
d.addCallback(self.checkRenderError) | ||||
def checkRenderError(self, result): | ||||
assert(self.fb.renderErrorCalled) | ||||
# TODO: add tests for history | ||||