Add support for trustee service in transfers (#499)

Author: lokst

dnsimple/struct/domain_transfer.py

@@ -9,10 +9,26 @@
 class DomainTransferRequest(dict):
     """DomainTransferRequest represents the attributes you can pass to a register API request."""
 
-    def __init__(self, registrant_id, auth_code, whois_privacy=False, auto_renew=False, extended_attributes=None,
-                 premium_price=None):
-        dict.__init__(self, registrant_id=registrant_id, auth_code=auth_code, whois_privacy=whois_privacy,
-                      auto_renew=auto_renew, extended_attributes=extended_attributes, premium_price=premium_price)
+    def __init__(
+        self,
+        registrant_id,
+        auth_code,
+        whois_privacy=False,
+        auto_renew=False,
+        extended_attributes=None,
+        premium_price=None,
+        trustee_service=None,
+    ):
+        dict.__init__(
+            self,
+            registrant_id=registrant_id,
+            auth_code=auth_code,
+            whois_privacy=whois_privacy,
+            auto_renew=auto_renew,
+            extended_attributes=extended_attributes,
+            premium_price=premium_price,
+            trustee_service=trustee_service,
+        )
 
     def to_json(self):
         return json.dumps(omitempty(self))
@@ -33,6 +49,8 @@ class DomainTransfer(Struct):
     """True if the domain auto-renew was requested"""
     whois_privacy = False
     """True if the domain WHOIS privacy was requested"""
+    trustee_service = False
+    """True if the domain Trustee service was requested"""
     created_at = None
     """When the domain transfer was created in DNSimple"""
     updated_at = None

tests/service/registrar_test.py

@@ -108,6 +108,7 @@ def test_transfer_domain(self):
         self.assertEqual('transferring', domain_transfer.state)
         self.assertFalse(domain_transfer.auto_renew)
         self.assertFalse(domain_transfer.whois_privacy)
+        self.assertFalse(domain_transfer.trustee_service)
         self.assertEqual('2016-12-09T19:43:41Z', domain_transfer.created_at)
         self.assertEqual('2016-12-09T19:43:43Z', domain_transfer.updated_at)
 
@@ -140,6 +141,7 @@ def test_get_domain_transfer(self):
         domain_transfer = self.registrar.get_domain_transfer(1010, 'ruby.codes', 358).data
 
         self.assertEqual('cancelled', domain_transfer.state)
+        self.assertFalse(domain_transfer.trustee_service)
 
     @responses.activate
     def test_cancel_domain_transfer(self):
@@ -149,6 +151,7 @@ def test_cancel_domain_transfer(self):
         domain_transfer = self.registrar.cancel_domain_transfer(1010, 'ruby.codes', 358).data
 
         self.assertEqual('transferring', domain_transfer.state)
+        self.assertFalse(domain_transfer.trustee_service)
 
     @responses.activate
     def test_renew_domain(self):