Update INFO.yaml with new PTL
[testsuite/python-testing-utils.git] / robotframework-onap / ONAPLibrary / DNSKeywords.py
1 # Copyright 2019 AT&T Intellectual Property. All rights reserved.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #         http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 import dns.message
16 import dns.name
17 import dns.query
18 from robot.api.deco import keyword
19
20
21 class DNSKeywords(object):
22     """ Utilities useful for DNS requests """
23
24     def __init__(self):
25         super(DNSKeywords, self).__init__()
26
27     @keyword
28     def dns_request(self, domain, ns):
29         """ return the ip address of the given domain name from the given nameserver """
30         request = dns.message.make_query(domain, dns.rdatatype.A)
31         request.flags |= dns.flags.AD
32         request.find_rrset(request.additional, dns.name.root, 65535, dns.rdatatype.OPT, create=True, force_unique=True)
33         response = dns.query.udp(request, ns)
34
35         for answer in response.answer:
36             for item in answer.items:
37                 return item