def setup(self):
self.configuration = DeviceConfiguration.get(sample_configuration, is_file_path=False)
self.configuration['commands'][3]['delay'] = 0
self.terminal = MockTerminal()
self.context_manager = ContextManager()
base_ctx = DeviceContext('base')
base_ctx.prompt = 'device#'
self.context_manager.add(base_ctx)
self.command_manager = CommandManager(self.configuration)
self.run = partial(self.command_manager.run, self.terminal, self.context_manager)
class TestDeviceContext(object):
def setup(self):
self.ctx01 = DeviceContext('ctx01')
def test_initialization(self):
assert_equals(self.ctx01.name, 'ctx01')
assert_equals(self.ctx01.prompt, None)
assert_raises(TypeError, DeviceContext, 20)
def test_equality(self):
ctx01 = DeviceContext('ctx01')
ctx02 = DeviceContext('ctx02')
ctx03 = DeviceContext('ctx03')
assert_equals(self.ctx01, ctx01)
assert_not_equals(self.ctx01, ctx03)
assert_not_equals(ctx02, ctx03)
def test_repr(self):
expected_repr = '<{} {}>'.format(DeviceContext.__name__, self.ctx01.name)
assert_equals(self.ctx01.__repr__(), expected_repr)
def setup(self):
self.ctx01 = DeviceContext('ctx01')
