def test_get_transport(self, mock_config): mock_config.return_value = 'udp' self.assertEqual('udp', utils.get_transport()) mock_config.return_value = 'udpu' self.assertEqual('udpu', utils.get_transport()) mock_config.return_value = 'hafu' self.assertRaises(ValueError, utils.get_transport)