def test_update_called_when_traffic_type_exists(self): set_module_args(base_module_args) module = setup_module_object() actt = AnsibleCloudStackTrafficType(module) actt.get_traffic_type = MagicMock( return_value=EXISTING_TRAFFIC_TYPES_RESPONSE['traffictype'][0]) actt.update_traffic_type = MagicMock() actt.present_traffic_type() self.assertTrue(actt.update_traffic_type.called)
def test_update_not_called_when_traffic_type_doesnt_exist(self): set_module_args(base_module_args) module = setup_module_object() actt = AnsibleCloudStackTrafficType(module) actt.get_traffic_type = MagicMock(return_value=None) actt.update_traffic_type = MagicMock() actt.add_traffic_type = MagicMock() actt.present_traffic_type() self.assertFalse(actt.update_traffic_type.called) self.assertTrue(actt.add_traffic_type.called)
def test_traffic_type_returned_if_exists(self): set_module_args(base_module_args) module = setup_module_object() actt = AnsibleCloudStackTrafficType(module) actt.get_physical_network = MagicMock( return_value=VALID_LIST_NETWORKS_RESPONSE['physicalnetwork'][0]) actt.get_traffic_types = MagicMock( return_value=EXISTING_TRAFFIC_TYPES_RESPONSE) tt = actt.present_traffic_type() self.assertTrue( tt.get('kvmnetworklabel') == base_module_args['kvm_networklabel']) self.assertTrue( tt.get('traffictype') == base_module_args['traffic_type'])
def test_module_is_created_sensibly(self): set_module_args(base_module_args) module = setup_module_object() assert module.params['traffic_type'] == 'Guest'