4
with mock.patch('charmhelpers.core.hookenv.config'):
5
import hooks as hacluster_hooks
8
class SwiftContextTestCase(unittest.TestCase):
10
@mock.patch('hooks.config')
11
def test_get_transport(self, mock_config):
12
mock_config.return_value = 'udp'
13
self.assertEqual('udp', hacluster_hooks.get_transport())
15
mock_config.return_value = 'udpu'
16
self.assertEqual('udpu', hacluster_hooks.get_transport())
18
mock_config.return_value = 'hafu'
19
self.assertRaises(ValueError, hacluster_hooks.get_transport)