Add tests

anasouma · anasouma · commit f71eb2b2f580 · 2014-04-14T14:17:19.000-07:00
diff --git a/SoftLayer/managers/load_balancer.py b/SoftLayer/managers/load_balancer.py
@@ -289,8 +289,8 @@ def edit_service_group(self, loadbal_id, group_id, allocation=-1, port=0,
         :param int routing_method: The routing method to set on the group
         """
         kwargs = NestedDict({})
-        kwargs['mask'] = 'mask[virtualServers[serviceGroups' \
-                         '[services[groupReferences]]]]'
+        kwargs['mask'] = ('mask[virtualServers[serviceGroups'
+                         '[services[groupReferences]]]]')
 
         load_balancer = self.lb_svc.getObject(id=loadbal_id, **kwargs)
         virtual_servers = load_balancer['virtualServers']
@@ -305,9 +305,10 @@ def edit_service_group(self, loadbal_id, group_id, allocation=-1, port=0,
                     service_group['routingTypeId'] = routing_type
                 if routing_method != 0:
                     service_group['routingMethodId'] = routing_method
+                break
         return self.lb_svc.editObject(load_balancer, id=loadbal_id)
 
-    def reset_service_group(self, loadbal_id, group_id=None):
+    def reset_service_group(self, loadbal_id, group_id):
         """ Resets all the connections on the service group
         :param int loadbal_id: The id of the loadbal
         :param int group_id: The id of the service group to reset
diff --git a/SoftLayer/tests/fixtures/Network_Application_Delivery_Controller_LoadBalancer_VirtualIpAddress.py b/SoftLayer/tests/fixtures/Network_Application_Delivery_Controller_LoadBalancer_VirtualIpAddress.py
@@ -1,5 +1,58 @@
 getBillingItem = {'id': 21370814 }
-getObject = {}
+getObject = {
+    'accountId': 307608,
+    'connectionLimit': 500,
+    'connectionLimitUnits': "connections/second",
+    'dedicatedFlag': False,
+    'highAvailabilityFlag': False,
+    'id': 22348,
+    'ipAddressId': 7303278,
+    'managedResourceFlag': False,
+    'sslActiveFlag': False,
+    'sslEnabledFlag': True,
+    'virtualServers': [
+    {
+        'allocation': 10,
+        'id': 50718,
+        'port': 80,
+        "serviceGroups": [
+            {
+                'id': 51758,
+                'routingMethodId': 10,
+                'routingTypeId': 3,
+                'services': [
+                    {
+                        'enabled': 1,
+                        'id': 1234,
+                        'healthChecks': [
+                            {
+                                'id': 112112
+                            }
+                        ],
+                        'groupReferences': [
+                            {
+                                'serviceGroupId': 51758,
+                                'serviceId': 84986,
+                                'weight': 2
+                            }
+                        ],
+                        'ipAddressId': 14288108,
+                        'port': 8080,
+                        'status': "DOWN"
+                    }
+                ]
+            }
+        ],
+        "virtualIpAddress": {
+            'accountId': 307608,
+            'connectionLimit': 500,
+            'connectionLimitUnits': "connections/second",
+            'id': 22348,
+            'ipAddressId': 7303278,
+        },
+        'virtualIpAddressId': 22348
+    }
+]}
 getVirtualServers = [
     {
         "allocation": 10,
