def test_should_call_tram_handler_unknown_param(self):
configuration = Configuration()
configuration.home_address = "home"
configuration.work_address = {"sender": { "target": "work", "waypoint": "point"}}
handler = TramHandler()
handler.process = MagicMock(return_value="")
command = TramCommand("fake")
self.assertEqual("Unknown tram params: fake", command.process("sender"))
def test_should_call_tram_handler(self):
configuration = Configuration()
configuration.home_address = "src"
configuration.work_address = {"sender": { "target": "dest", "waypoint": "point"}}
handler = TramHandler()
handler.process = MagicMock(return_value="from src to dest")
command = TramCommand(None)
command._get_tram_configuration = MagicMock(return_value=configuration)
command._get_tram_handler = MagicMock(return_value=handler)
self.assertEqual("from src to dest", command.process("sender"))
def test_should_call_tram_handler_default(self):
configuration = Configuration()
configuration.home_address = "home"
configuration.work_address = {"sender": { "target": "work", "waypoint": "point"}}
handler = TramHandler()
handler.process = MagicMock(return_value="")
command = TramCommand(None)
command._get_tram_configuration = MagicMock(return_value=configuration)
command._get_tram_handler = MagicMock(return_value=handler)
command.process("sender")
handler.process.assert_called_once_with("home", "work", "point")
