class Test_tool(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
def test_queue_set_data_successfully(self):
self.client.tool().queue(1)
request = self.client.request
self.assertEqual(1, request.data['id'])
class Test_Custom(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data = {'optionalParam': 'value'}
def test_request_set_data_successfully(self):
self.client.custom().request('/vlan/addServer', 'post', {
'serverId': 1,
'vlanSid': 3
})
request = self.client.request
self.assertEqual(1, request.data['serverId'])
self.assertEqual(3, request.data['vlanSid'])
class Test_os(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data = {'optionalParam': 'value'}
def test_get_params_set_data_successfully(self):
self.client.os().get_params(1)
request = self.client.request
self.assertEqual(1, request.data['serverId'])
def test_get_available_os_versions_set_data_successfully(self):
self.client.os().get_available_os_versions(1)
request = self.client.request
self.assertEqual(1, request.data['serverId'])
class Test_vlan(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data = {'optionalParam': 'value'}
def test_add_server_set_data_successfully(self):
self.client.vlan().add_server(1, 3)
request = self.client.request
self.assertEqual(1, request.data['serverId'])
self.assertEqual(3, request.data['vlanSid'])
def test_remove_server_set_data_successfully(self):
self.client.vlan().remove_server(1, 3)
request = self.client.request
self.assertEqual(1, request.data['serverId'])
self.assertEqual(3, request.data['vlanSid'])
def test_list_set_correct_path(self):
self.client.vlan().list()
request = self.client.request
self.assertEqual('/vlan/list', request.path)
def test_service_method_return_service_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.service()
self.assertEqual('Service', api_resource.__class__.__name__)
def test_client_api_key_attribute_is_required(self):
with self.assertRaises(ValueError):
client = Client()
def test_reseller_client_method_return_reseller_client_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.reseller_client()
self.assertEqual('ResellerClient', api_resource.__class__.__name__)
def test_product_method_return_product_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.product()
self.assertEqual('Product', api_resource.__class__.__name__)
class Test_support(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data = {'optionalParam': 'value'}
def test_get_active_tickets_set_all_fields_correctly(self):
self.client.support().get_active_tickets(self.optional_data)
request = self.client.request
self.assertEqual('value', request.data['optionalParam'])
def test_get_tickets_set_all_fields_correctly(self):
self.client.support().get_tickets(20, self.optional_data)
request = self.client.request
self.assertEqual(20, request.data['limit'])
self.assertEqual('value', request.data['optionalParam'])
def test_search_set_all_fields_correctly(self):
self.client.support().search(self.optional_data)
request = self.client.request
self.assertEqual('value', request.data['optionalParam'])
def test_get_ticket_set_all_fields_correctly(self):
self.client.support().get_ticket(69)
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
def test_reply_ticket_set_all_fields_correctly(self):
self.client.support().reply_ticket(69, 'Done!')
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
self.assertEqual('Done!', request.data['text'])
def test_update_ticket_priority_set_all_fields_correctly(self):
self.client.support().update_ticket_priority(69, 2)
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
self.assertEqual(2, request.data['priority'])
def test_close_ticket_set_all_fields_correctly(self):
self.client.support().close_ticket(69)
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
def test_create_ticket_set_all_fields_correctly(self):
ticket_subject = 'Server is down'
ticket_body = 'The server has a network issue!'
self.client.support().create_ticket(ticket_subject, ticket_body,
self.optional_data)
request = self.client.request
self.assertEqual(ticket_subject, request.data['subject'])
self.assertEqual(ticket_body, request.data['text'])
self.assertEqual('value', request.data['optionalParam'])
def test_update_ticket_set_all_fields_correctly(self):
self.client.support().update_ticket(69, self.optional_data)
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
self.assertEqual('value', request.data['optionalParam'])
def test_forward_to_ph_set_all_fields_correctly(self):
self.client.support().forward_to_ph(69)
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
def test_forward_to_ph_remove_set_all_fields_correctly(self):
self.client.support().forward_to_ph_remove(69)
request = self.client.request
self.assertEqual(69, request.data['ticketId'])
def test_vlan_method_return_vlan_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.vlan()
self.assertEqual('Vlan', api_resource.__class__.__name__)
class Test_user(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data = {'optionalParam': 'value'}
def test_auth_set_data_successfully(self):
self.client.user().auth('benny', 'my-secret-password')
request = self.client.request
self.assertEqual('benny', request.data['login'])
self.assertEqual('my-secret-password', request.data['password'])
def test_ping_path_is_correct(self):
self.client.user().ping()
request = self.client.request
self.assertEqual('/user/ping', request.path)
def test_get_payment_method_path_is_correct(self):
self.client.user().get_payment_methods()
request = self.client.request
self.assertEqual('/user/getPaymentMethods', request.path)
def test_generate_secret_key_set_data_successfully(self):
self.client.user().generate_secret_key('dev-key', 'rw')
request = self.client.request
self.assertEqual('dev-key', request.data['keyName'])
self.assertEqual('rw', request.data['access'])
def test_update_secret_key_set_data_successfully(self):
self.client.user().update_secret_key(1, self.optional_data)
request = self.client.request
self.assertEqual(1, request.data['keyId'])
self.assertEqual('value', request.data['optionalParam'])
def test_delete_secret_key_set_data_successfully(self):
self.client.user().delete_secret_key(1)
request = self.client.request
self.assertEqual(1, request.data['keyId'])
def test_get_secret_keys_set_path_successfully(self):
self.client.user().get_secret_keys()
request = self.client.request
self.assertEqual('/user/getSecretKeys', request.path)
def test_regenerate_secret_key_set_data_successfully(self):
self.client.user().regenerate_secret_key(1, 'r')
request = self.client.request
self.assertEqual(1, request.data['keyId'])
self.assertEqual('r', request.data['access'])
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
class Test_ip(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data={'optionalParam': 'value'}
def test_set_ptr_set_data_successfully(self):
self.client.ip().set_ptr('192.168.0.1', self.optional_data)
request = self.client.request
self.assertEqual('192.168.0.1', request.data['ip'])
self.assertEqual('value', request.data['optionalParam'])
def test_get_ptr_set_data_successfully(self):
self.client.ip().get_ptr('192.168.0.1')
request = self.client.request
self.assertEqual('192.168.0.1', request.data['ip'])
def test_null_route_set_data_successfully(self):
self.client.ip().null_route('192.168.0.1', self.optional_data)
request = self.client.request
self.assertEqual('192.168.0.1', request.data['ip'])
self.assertEqual('value', request.data['optionalParam'])
def test_un_null_route_set_data_successfully(self):
self.client.ip().un_null_route('192.168.0.1')
request = self.client.request
self.assertEqual('192.168.0.1', request.data['ip'])
def test_route_set_data_successfully(self):
self.client.ip().route('192.168.0.1', 1)
request = self.client.request
self.assertEqual('192.168.0.1', request.data['ip'])
self.assertEqual(1, request.data['serverId'])
def test_get_list_set_data_successfully(self):
self.client.ip().get_list(self.optional_data)
request = self.client.request
self.assertEqual('value', request.data['optionalParam'])
def test_get_list6_set_data_successfully(self):
self.client.ip().get_list6(self.optional_data)
request = self.client.request
self.assertEqual('value', request.data['optionalParam'])
def test_support_method_return_support_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.support()
self.assertEqual('Support', api_resource.__class__.__name__)
def test_client_api_user_credentials_can_be_set(self):
my_username = '******'
my_password = '******'
client = Client(username=my_username, password=my_password)
self.assertEqual(my_username, client.username)
self.assertEqual(my_password, client.password)
def test_user_method_return_user_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.user()
self.assertEqual('User', api_resource.__class__.__name__)
def test_ip_method_return_ip_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.ip()
self.assertEqual('IP', api_resource.__class__.__name__)
def test_client_api_keys_can_be_set(self):
my_api_key = 'test-key'
my_api_secret = 'my-secret'
client = Client(api_key=my_api_key, api_secret=my_api_secret)
self.assertEqual(my_api_key, client.api_key)
self.assertEqual(my_api_secret, client.api_secret)
def test_os_method_return_os_resource(self):
client = Client(api_key='a', api_secret='b')
api_resource = client.os()
self.assertEqual('OS', api_resource.__class__.__name__)
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data = {'optionalParam': 'value'}
class Test_product(unittest.TestCase):
def setUp(self):
self.client = Client(api_key='a', api_secret='b')
self.optional_data={'optionalParam': 'value'}
def test_get_product_list_set_data_successfully(self):
self.client.product().get_product_list(self.optional_data)
request = self.client.request
self.assertEqual('value', request.data['optionalParam'])
def test_get_config_options_set_data_successfully(self):
self.client.product().get_config_options(1)
request = self.client.request
self.assertEqual(1, request.data['productId'])
def test_get_addons_set_data_successfully(self):
self.client.product().get_addons(1)
request = self.client.request
self.assertEqual(1, request.data['productId'])
def test_order_products_set_data_successfully(self):
self.client.product().order_products(1, 'card', self.optional_data)
request = self.client.request
self.assertEqual(1, request.data['productId'])
self.assertEqual('card', request.data['paymentMethod'])
self.assertEqual('value', request.data['optionalParam'])
def test_order_history_set_data_successfully(self):
self.client.product().order_history(1)
request = self.client.request
self.assertEqual(1, request.data['orderId'])
def test_cancel_service_set_data_successfully(self):
self.client.product().cancel_service(1, 'My reason', self.optional_data)
request = self.client.request
self.assertEqual(1, request.data['serviceId'])
self.assertEqual('My reason', request.data['reason'])
self.assertEqual('value', request.data['optionalParam'])
def test_cancel_pending_order_set_data_successfully(self):
self.client.product().cancel_pending_order(1)
request = self.client.request
self.assertEqual(1, request.data['orderId'])
def test_upgrade_service_set_data_successfully(self):
self.client.product().upgrade_service(1, '22,69')
request = self.client.request
self.assertEqual(1, request.data['serviceId'])
self.assertEqual('22,69', request.data['configOptions'])
