Remove the f5-sdk from bigip_gtm_pool_member (#48506)

lib/ansible/modules/network/f5/bigip_gtm_pool_member.py

@@ -142,59 +142,115 @@ options:
 extends_documentation_fragment: f5
 author:
   - Tim Rupp (@caphrim007)
+  - Wojciech Wypior (@wojtek0806)
 '''
 
 EXAMPLES = r'''
-- name: Create a ...
+- name: Create a GTM pool member
   bigip_gtm_pool_member:
-    name: foo
+    pool: pool1
+    server_name: server1
+    virtual_server: vs1
+    type: a
+    provider:
       password: secret
       server: lb.mydomain.com
-    state: present
       user: admin
   delegate_to: localhost
 '''
 
 RETURN = r'''
-param1:
+bits_enabled:
-  description: The new param1 value of the resource.
+  description: Whether the bits limit is enabled.
   returned: changed
   type: bool
-  sample: true
+  sample: yes
-param2:
+bits_limit:
-  description: The new param2 value of the resource.
+  description: The new bits_enabled limit.
+  returned: changed
+  type: int
+  sample: 100
+connections_enabled:
+  description: Whether the connections limit is enabled.
+  returned: changed
+  type: bool
+  sample: yes
+connections_limit:
+  description: The new connections_limit limit.
+  returned: changed
+  type: int
+  sample: 100
+disabled:
+  description: Whether the pool member is disabled or not.
+  returned: changed
+  type: bool
+  sample: yes
+enabled:
+  description: Whether the pool member is enabled or not.
+  returned: changed
+  type: bool
+  sample: yes
+member_order:
+  description: The new order in which the member appears in the pool.
+  returned: changed
+  type: int
+  sample: 2
+monitor:
+  description: The new monitor assigned to the pool member.
+  returned: changed
+  type: string
+  sample: /Common/monitor1
+packets_enabled:
+  description: Whether the packets limit is enabled.
+  returned: changed
+  type: bool
+  sample: yes
+packets_limit:
+  description: The new packets_limit limit.
+  returned: changed
+  type: int
+  sample: 100
+ratio:
+  description: The new weight of the member for load balancing.
+  returned: changed
+  type: int
+  sample: 10
+description:
+  description: The new description of the member.
   returned: changed
   type: string
-  sample: Foo is bar
+  sample: My description
 '''
 
 from ansible.module_utils.basic import AnsibleModule
 from ansible.module_utils.basic import env_fallback
 
 try:
-    from library.module_utils.network.f5.bigip import HAS_F5SDK
+    from library.module_utils.compat.ipaddress import ip_address
-    from library.module_utils.network.f5.bigip import F5Client
+    from library.module_utils.network.f5.bigip import F5RestClient
     from library.module_utils.network.f5.common import F5ModuleError
     from library.module_utils.network.f5.common import AnsibleF5Parameters
     from library.module_utils.network.f5.common import cleanup_tokens
     from library.module_utils.network.f5.common import fq_name
     from library.module_utils.network.f5.common import f5_argument_spec
-    try:
+    from library.module_utils.network.f5.common import transform_name
-        from library.module_utils.network.f5.common import iControlUnexpectedHTTPError
+    from library.module_utils.network.f5.common import exit_json
-    except ImportError:
+    from library.module_utils.network.f5.common import fail_json
-        HAS_F5SDK = False
+    from library.module_utils.network.f5.common import flatten_boolean
+    from library.module_utils.network.f5.icontrol import module_provisioned
 except ImportError:
-    from ansible.module_utils.network.f5.bigip import HAS_F5SDK
+    from ansible.module_utils.compat.ipaddress import ip_address
-    from ansible.module_utils.network.f5.bigip import F5Client
+    from ansible.module_utils.network.f5.bigip import F5RestClient
     from ansible.module_utils.network.f5.common import F5ModuleError
     from ansible.module_utils.network.f5.common import AnsibleF5Parameters
     from ansible.module_utils.network.f5.common import cleanup_tokens
     from ansible.module_utils.network.f5.common import fq_name
     from ansible.module_utils.network.f5.common import f5_argument_spec
-    try:
+    from ansible.module_utils.network.f5.common import transform_name
-        from ansible.module_utils.network.f5.common import iControlUnexpectedHTTPError
+    from ansible.module_utils.network.f5.common import exit_json
-    except ImportError:
+    from ansible.module_utils.network.f5.common import fail_json
-        HAS_F5SDK = False
+    from ansible.module_utils.network.f5.common import flatten_boolean
+    from ansible.module_utils.network.f5.icontrol import module_provisioned
 
 
 class Parameters(AnsibleF5Parameters):
@@ -205,7 +261,7 @@ class Parameters(AnsibleF5Parameters):
         'limitMaxConnectionsStatus': 'connections_enabled',
         'limitMaxPps': 'packets_limit',
         'limitMaxPpsStatus': 'packets_enabled',
-        'memberOrder': 'member_order'
+        'memberOrder': 'member_order',
     }
 
     api_attributes = [
@@ -309,21 +365,6 @@ class ModuleParameters(Parameters):
             return None
         return str(self._values['type'])
 
-    @property
-    def collection(self):
-        type_map = dict(
-            a='a_s',
-            aaaa='aaaas',
-            cname='cnames',
-            mx='mxs',
-            naptr='naptrs',
-            srv='srvs'
-        )
-        if self._values['type'] is None:
-            return None
-        wideip_type = self._values['type']
-        return type_map[wideip_type]
-
     @property
     def enabled(self):
         if self._values['state'] == 'enabled':
@@ -392,7 +433,13 @@ class UsableChanges(Changes):
 
 
 class ReportableChanges(Changes):
-    pass
+    @property
+    def disabled(self):
+        return flatten_boolean(self._values['disabled'])
+
+    @property
+    def enabled(self):
+        return flatten_boolean(self._values['enabled'])
 
 
 class Difference(object):
@@ -480,17 +527,18 @@ class ModuleManager(object):
         return False
 
     def exec_module(self):
+        if not module_provisioned(self.client, 'gtm'):
+            raise F5ModuleError(
+                "GTM must be provisioned to use this module."
+            )
         changed = False
         result = dict()
         state = self.want.state
 
-        try:
         if state in ['present', 'enabled', 'disabled']:
             changed = self.present()
         elif state == "absent":
             changed = self.absent()
-        except iControlUnexpectedHTTPError as e:
-            raise F5ModuleError(str(e))
 
         reportable = ReportableChanges(params=self.changes.to_return())
         changes = reportable.to_return()
@@ -514,18 +562,21 @@ class ModuleManager(object):
             return self.create()
 
     def exists(self):
-        pools = self.client.api.tm.gtm.pools
+        uri = "https://{0}:{1}/mgmt/tm/gtm/pool/{2}/{3}/members/{4}".format(
-        collection = getattr(pools, self.want.collection)
+            self.client.provider['server'],
-        resource = getattr(collection, self.want.type)
+            self.client.provider['server_port'],
-        resource = resource.load(
+            self.want.type,
-            name=self.want.pool,
+            transform_name(self.want.partition, self.want.pool),
-            partition=self.want.partition
+            transform_name(self.want.partition, self.want.name),
-        )
-        result = resource.members_s.member.exists(
-            name=self.want.name,
-            partition=self.want.partition
         )
-        return result
+        resp = self.client.api.get(uri)
+        try:
+            response = resp.json()
+        except ValueError:
+            return False
+        if resp.status == 404 or 'code' in response and response['code'] == 404:
+            return False
+        return True
 
     def update(self):
         self.have = self.read_current_from_device()
@@ -557,33 +608,47 @@ class ModuleManager(object):
 
     def create_on_device(self):
         params = self.changes.api_params()
-        pools = self.client.api.tm.gtm.pools
+        params['name'] = self.want.name
-        collection = getattr(pools, self.want.collection)
+        params['partition'] = self.want.partition
-        resource = getattr(collection, self.want.type)
+        uri = "https://{0}:{1}/mgmt/tm/gtm/pool/{2}/{3}/members/".format(
-        resource = resource.load(
+            self.client.provider['server'],
-            name=self.want.pool,
+            self.client.provider['server_port'],
-            partition=self.want.partition
+            self.want.type,
-        )
+            transform_name(self.want.partition, self.want.pool),
-        resource.members_s.member.create(
-            name=self.want.name,
-            partition=self.want.partition,
-            **params
         )
+        resp = self.client.api.post(uri, json=params)
+        try:
+            response = resp.json()
+        except ValueError as ex:
+            raise F5ModuleError(str(ex))
+
+        if 'code' in response and response['code'] in [400, 403]:
+            if 'message' in response:
+                raise F5ModuleError(response['message'])
+            else:
+                raise F5ModuleError(resp.content)
+        return response['selfLink']
 
     def update_on_device(self):
         params = self.changes.api_params()
-        pools = self.client.api.tm.gtm.pools
+        uri = "https://{0}:{1}/mgmt/tm/gtm/pool/{2}/{3}/members/{4}".format(
-        collection = getattr(pools, self.want.collection)
+            self.client.provider['server'],
-        resource = getattr(collection, self.want.type)
+            self.client.provider['server_port'],
-        resource = resource.load(
+            self.want.type,
-            name=self.want.pool,
+            transform_name(self.want.partition, self.want.pool),
-            partition=self.want.partition
+            transform_name(self.want.partition, self.want.name),
         )
-        resource = resource.members_s.member.load(
+        resp = self.client.api.patch(uri, json=params)
-            name=self.want.name,
+        try:
-            partition=self.want.partition
+            response = resp.json()
-        )
+        except ValueError as ex:
-        resource.modify(**params)
+            raise F5ModuleError(str(ex))
+
+        if 'code' in response and response['code'] == 400:
+            if 'message' in response:
+                raise F5ModuleError(response['message'])
+            else:
+                raise F5ModuleError(resp.content)
 
     def absent(self):
         if self.exists():
@@ -591,34 +656,38 @@ class ModuleManager(object):
         return False
 
     def remove_from_device(self):
-        pools = self.client.api.tm.gtm.pools
+        uri = "https://{0}:{1}/mgmt/tm/gtm/pool/{2}/{3}/members/{4}".format(
-        collection = getattr(pools, self.want.collection)
+            self.client.provider['server'],
-        resource = getattr(collection, self.want.type)
+            self.client.provider['server_port'],
-        resource = resource.load(
+            self.want.type,
-            name=self.want.pool,
+            transform_name(self.want.partition, self.want.pool),
-            partition=self.want.partition
+            transform_name(self.want.partition, self.want.name),
-        )
-        resource = resource.members_s.member.load(
-            name=self.want.name,
-            partition=self.want.partition
         )
-        if resource:
+        response = self.client.api.delete(uri)
-            resource.delete()
+        if response.status == 200:
+            return True
+        raise F5ModuleError(response.content)
 
     def read_current_from_device(self):
-        pools = self.client.api.tm.gtm.pools
+        uri = "https://{0}:{1}/mgmt/tm/gtm/pool/{2}/{3}/members/{4}".format(
-        collection = getattr(pools, self.want.collection)
+            self.client.provider['server'],
-        resource = getattr(collection, self.want.type)
+            self.client.provider['server_port'],
-        resource = resource.load(
+            self.want.type,
-            name=self.want.pool,
+            transform_name(self.want.partition, self.want.pool),
-            partition=self.want.partition
+            transform_name(self.want.partition, self.want.name),
-        )
-        resource = resource.members_s.member.load(
-            name=self.want.name,
-            partition=self.want.partition
         )
-        result = resource.attrs
+        resp = self.client.api.get(uri)
-        return ApiParameters(params=result)
+        try:
+            response = resp.json()
+        except ValueError as ex:
+            raise F5ModuleError(str(ex))
+
+        if 'code' in response and response['code'] == 400:
+            if 'message' in response:
+                raise F5ModuleError(response['message'])
+            else:
+                raise F5ModuleError(resp.content)
+        return ApiParameters(params=response)
 
 
 class ArgumentSpec(object):
@@ -669,20 +738,19 @@ def main():
 
     module = AnsibleModule(
         argument_spec=spec.argument_spec,
-        supports_check_mode=spec.supports_check_mode
+        supports_check_mode=spec.supports_check_mode,
     )
-    if not HAS_F5SDK:
+
-        module.fail_json(msg="The python f5-sdk module is required")
+    client = F5RestClient(**module.params)
 
     try:
-        client = F5Client(**module.params)
         mm = ModuleManager(module=module, client=client)
         results = mm.exec_module()
         cleanup_tokens(client)
-        module.exit_json(**results)
+        exit_json(module, results, client)
     except F5ModuleError as ex:
         cleanup_tokens(client)
-        module.fail_json(msg=str(ex))
+        fail_json(module, ex, client)
 
 
 if __name__ == '__main__':

test/units/modules/network/f5/test_bigip_gtm_pool_member.py

@@ -8,16 +8,12 @@ __metaclass__ = type
 
 import os
 import json
-import pytest
 import sys
 
 from nose.plugins.skip import SkipTest
 if sys.version_info < (2, 7):
     raise SkipTest("F5 Ansible modules require Python >= 2.7")
 
-from units.compat import unittest
-from units.compat.mock import Mock
-from units.compat.mock import patch
 from ansible.module_utils.basic import AnsibleModule
 
 try:
@@ -25,17 +21,25 @@ try:
     from library.modules.bigip_gtm_pool_member import ModuleParameters
     from library.modules.bigip_gtm_pool_member import ModuleManager
     from library.modules.bigip_gtm_pool_member import ArgumentSpec
-    from library.module_utils.network.f5.common import F5ModuleError
+
-    from library.module_utils.network.f5.common import iControlUnexpectedHTTPError
+    # In Ansible 2.8, Ansible changed import paths.
-    from test.unit.modules.utils import set_module_args
+    from test.units.compat import unittest
+    from test.units.compat.mock import Mock
+    from test.units.compat.mock import patch
+
+    from test.units.modules.utils import set_module_args
 except ImportError:
     try:
         from ansible.modules.network.f5.bigip_gtm_pool_member import ApiParameters
         from ansible.modules.network.f5.bigip_gtm_pool_member import ModuleParameters
         from ansible.modules.network.f5.bigip_gtm_pool_member import ModuleManager
         from ansible.modules.network.f5.bigip_gtm_pool_member import ArgumentSpec
-        from ansible.module_utils.network.f5.common import F5ModuleError
+
-        from ansible.module_utils.network.f5.common import iControlUnexpectedHTTPError
+        # Ansible 2.8 imports
+        from units.compat import unittest
+        from units.compat.mock import Mock
+        from units.compat.mock import patch
+
         from units.modules.utils import set_module_args
     except ImportError:
         raise SkipTest("F5 Ansible modules require the f5-sdk Python library")
@@ -112,13 +116,23 @@ class TestParameters(unittest.TestCase):
         assert p.connections_limit == 0
 
 
-@patch('ansible.module_utils.f5_utils.AnsibleF5Client._get_mgmt_root',
-       return_value=True)
 class TestManager(unittest.TestCase):
 
     def setUp(self):
         self.spec = ArgumentSpec()
 
+        try:
+            self.p1 = patch('library.modules.bigip_gtm_pool_member.module_provisioned')
+            self.m1 = self.p1.start()
+            self.m1.return_value = True
+        except Exception:
+            self.p1 = patch('ansible.modules.network.f5.bigip_gtm_pool_member.module_provisioned')
+            self.m1 = self.p1.start()
+            self.m1.return_value = True
+
+    def tearDown(self):
+        self.p1.stop()
+
     def test_create_monitor(self, *args):
         set_module_args(dict(
             pool='pool1',
@@ -153,6 +167,7 @@ class TestManager(unittest.TestCase):
         mm = ModuleManager(module=module)
         mm.exists = Mock(side_effect=[False, True])
         mm.create_on_device = Mock(return_value=True)
+        mm.module_provisioned = Mock(return_value=True)
 
         results = mm.exec_module()