def test_valid_ports_returns_expected_error_for_invalid_protocol(self): under_test = ClcPublicIp(self.module) under_test._validate_ports([{'protocol': 'ICP', 'port': 'Faygo'}]) self.module.fail_json.assert_called_once_with( msg="Valid protocols for this module are: [TCP, UDP]: You specified 'ICP'" )
def test_valid_ports_skips_and_removes_blank_entry(self): expected = [{'protocol': 'TCP', 'port': 456}] under_test = ClcPublicIp(self.module) actual = under_test._validate_ports([{'protocol': 'TCP', 'port': 456}, {}]) self.assertEqual(expected, actual) self.assertFalse(self.module.fail_json.called)
def test_valid_ports_returns_expected_output(self): expected = [ {'port': 80, 'protocol': 'TCP'}, {'port': 34590, 'protocol': 'UDP'}, {'port': 1234, 'port_to': 4567, 'protocol': 'TCP'} ] input_data = [ {'port': 80}, {'port': 34590, 'protocol': 'UDP'}, {'port': 1234, 'port_to': 4567} ] under_test = ClcPublicIp(self.module) actual = under_test._validate_ports(input_data) self.assertEqual(expected, actual) self.assertFalse(self.module.fail_json.called)
def test_valid_ports_returns_expected_error_for_missing_port(self): under_test = ClcPublicIp(self.module) under_test._validate_ports([{'protocol': 'UDP'}]) self.module.fail_json.assert_called_once_with(msg="You must provide a port")