Merge pull request #2 from ByteInternet/rrpproxy_extra_commands

Add extra RRPproxy commands

Author: Erik-Lamers1

rrpproxy/rrpproxy.py

@@ -64,19 +64,69 @@ def add_contact(self, **contact_data):
         """Allows you to add a new contact. System automatically substitutes existing handles."""
         return self.call('AddContact', **contact_data)
 
+    def check_contact(self, contact):
+        """Check availability of a contact"""
+        return self.call('CheckContact', contact=contact)
+
     def check_domain(self, domain):
         """Checks if the desired domain name is available and may be registered at the registry"""
         return self.call('CheckDomain', domain=domain)
 
+    def check_domain_transfer(self, domain, **transfer_data):
+        """Check transfer requirements of particular gTLDs"""
+        return self.call('CheckDomainTransfer', domain=domain, **transfer_data)
+
     def delete_domain(self, domain):
+        """The command is used to request the deletion of the domain"""
         return self.call('DeleteDomain', domain=domain)
 
+    def delete_contact(self, contact):
+        """Delete a contact"""
+        return self.call('DeleteContact', contact=contact)
+
+    def get_zone_info(self, zone, domain):
+        """Query information about a zone"""
+        return self.call('GetZoneInfo', zone=zone, domain=domain)
+
+    def modify_contact(self, contact, **contact_data):
+        """Modify a contact handle"""
+        return self.call('ModifyContact', contact=contact, **contact_data)
+
     def modify_domain(self, domain, **domain_data):
+        """Modify domain data"""
         return self.call('ModifyDomain', domain=domain, **domain_data)
 
     def renew_domain(self, domain, **domain_data):
+        """Command used to renew domains explicitly for a specified period of time"""
         return self.call('RenewDomain', domain=domain, **domain_data)
 
+    def resend_notification(self, type, object, **extra_data):
+        return self.call('ResendNotification', type=type, object=object, **extra_data)
+
+    def set_auth_code(self, domain, **extra_data):
+        return self.call('SetAuthCode', domain=domain, **extra_data)
+
+    def set_domain_renewal_mode(self, domain, token, renewalmode):
+        return self.call('SetDomainRenewalmode', domain=domain, token=token, renewalmode=renewalmode)
+
+    def status_contact(self, contact, auth):
+        return self.call('StatusContact', contact=contact, auth=auth)
+
     def status_domain(self, domain):
         """Enables you to check the actual status of a Domain in your portfolio"""
         return self.call('StatusDomain', domain=domain)
+
+    def status_domain_transfer(self, domain):
+        return self.call('StatusDomainTransfer', domain=domain)
+
+    def trade_domain(self, domain, **trade_data):
+        """Change registrant of a domain"""
+        return self.call('TradeDomain', domain=domain, **trade_data)
+
+    def transfer_domain(self, domain, action, **transfer_data):
+        """This command allows you to request, approve, deny or cancel a domain transfer."""
+        return self.call('TransferDomain', domain=domain, action=action, **transfer_data)
+
+    def query_zone_list(self):
+        """Query list of activated zones in account"""
+        return self.call('QueryZoneList')

tests/test_rrpproxy_add_contact.py

@@ -4,9 +4,6 @@
 
 
 class TestRRPProxyAddContact(TestRRPProxyBase):
-    def setup(self):
-        super().setUp()
-
     @patch('rrpproxy.RRPProxy.call')
     def test_calls_call_correctly(self, call_mock):
         response = self.proxy.add_contact(title='Mr.', firstName='Alejandro')

tests/test_rrpproxy_add_domain.py

@@ -4,9 +4,6 @@
 
 
 class TestRRPProxyAddDomain(TestRRPProxyBase):
-    def setup(self):
-        super().setUp()
-
     @patch('rrpproxy.RRPProxy.call')
     def test_calls_call_correctly(self, call_mock):
         response = self.proxy.add_domain('example.com', 1, ownerContact0='ABC')

tests/test_rrpproxy_check_contact.py

@@ -0,0 +1,12 @@
+from unittest.mock import patch
+
+from tests.test_rrpproxy_base import TestRRPProxyBase
+
+
+class TestRRPProxyCheckContact(TestRRPProxyBase):
+    @patch('rrpproxy.RRPProxy.call')
+    def test_calls_call_correctly(self, call_mock):
+        response = self.proxy.check_contact('CONTACT-A')
+
+        call_mock.assert_called_once_with('CheckContact', contact='CONTACT-A')
+        self.assertEqual(response, call_mock.return_value)

tests/test_rrpproxy_check_domain.py

@@ -4,9 +4,6 @@
 
 
 class TestRRPProxyCheckDomain(TestRRPProxyBase):
-    def setup(self):
-        super().setUp()
-
     @patch('rrpproxy.RRPProxy.call')
     def test_calls_call_correctly(self, call_mock):
         response = self.proxy.check_domain('example.com')

tests/test_rrpproxy_check_domain_transfer.py

@@ -0,0 +1,12 @@
+from unittest.mock import patch
+
+from tests.test_rrpproxy_base import TestRRPProxyBase
+
+
+class TestRRPProxyCheckDomainTransfer(TestRRPProxyBase):
+    @patch('rrpproxy.RRPProxy.call')
+    def test_calls_call_correctly(self, call_mock):
+        response = self.proxy.check_domain_transfer(domain='hypernode.com', auth='hjd7s', action='approve')
+
+        call_mock.assert_called_once_with('CheckDomainTransfer', domain='hypernode.com', auth='hjd7s', action='approve')
+        self.assertEqual(response, call_mock.return_value)

tests/test_rrpproxy_delete_contact.py

@@ -0,0 +1,12 @@
+from unittest.mock import patch
+
+from tests.test_rrpproxy_base import TestRRPProxyBase
+
+
+class TestRRPProxyDeleteContact(TestRRPProxyBase):
+    @patch('rrpproxy.RRPProxy.call')
+    def test_calls_call_correctly(self, call_mock):
+        response = self.proxy.delete_contact('CONTACT-B')
+
+        call_mock.assert_called_once_with('DeleteContact', contact='CONTACT-B')
+        self.assertEqual(response, call_mock.return_value)

tests/test_rrpproxy_delete_domain.py

@@ -4,9 +4,6 @@
 
 
 class TestRRPProxyDeleteDomain(TestRRPProxyBase):
-    def setup(self):
-        super().setUp()
-
     @patch('rrpproxy.RRPProxy.call')
     def test_calls_call_correctly(self, call_mock):
         response = self.proxy.delete_domain('example.com')

tests/test_rrpproxy_get_zone_info.py

@@ -0,0 +1,12 @@
+from unittest.mock import patch
+
+from tests.test_rrpproxy_base import TestRRPProxyBase
+
+
+class TestRRPProxyGetZoneInfo(TestRRPProxyBase):
+    @patch('rrpproxy.RRPProxy.call')
+    def test_calls_call_correctly(self, call_mock):
+        response = self.proxy.get_zone_info(zone='some_zone', domain='hypernode.com')
+
+        call_mock.assert_called_once_with('GetZoneInfo', zone='some_zone', domain='hypernode.com')
+        self.assertEqual(response, call_mock.return_value)

tests/test_rrpproxy_init.py

@@ -3,9 +3,6 @@
 
 
 class TestRRPProxyInit(TestRRPProxyBase):
-    def setUp(self):
-        super().setUp()
-
     def test_init_sets_correct_api_url_and_query_params_when_using_sandbox_environment(self):
         self.assertEqual(self.proxy.api_url, 'https://api-ote.rrpproxy.net/api/call.cgi')
         self.assertEqual(self.proxy.query_params, {'s_login': self.username, 's_pw': self.password, 's_opmode': 'OTE'})