Merge pull request #673 from wojtek0806/servers_pr

Adding GTM servers - First one to be merged

Author: caphrim007

f5/bigip/tm/gtm/__init__.py

@@ -31,6 +31,7 @@
 from f5.bigip.resource import OrganizingCollection
 from f5.bigip.tm.gtm.datacenter import Datacenters
 from f5.bigip.tm.gtm.rule import Rules
+from f5.bigip.tm.gtm.server import Servers
 
 
 class Gtm(OrganizingCollection):
@@ -39,5 +40,6 @@ def __init__(self, tm):
         super(Gtm, self).__init__(tm)
         self._meta_data['allowed_lazy_attributes'] = [
             Datacenters,
-            Rules
+            Rules,
+            Servers,
         ]

f5/bigip/tm/gtm/server.py

@@ -0,0 +1,75 @@
+# coding=utf-8
+#
+#  Copyright 2014-2016 F5 Networks Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+"""BIG-IP® Global Traffic Manager™ (GTM®) pool module.
+
+REST URI
+    ``http://localhost/mgmt/tm/gtm/server``
+
+GUI Path
+    ``DNS --> GSLB : Servers``
+
+REST Kind
+    ``tm:gtm:server:*``
+"""
+
+from f5.bigip.resource import Collection
+from f5.bigip.resource import Resource
+
+
+class Servers(Collection):
+    """BIG-IP® GTM server collection"""
+    def __init__(self, gtm):
+        super(Servers, self).__init__(gtm)
+        self._meta_data['allowed_lazy_attributes'] = [Server]
+        self._meta_data['attribute_registry'] = \
+            {'tm:gtm:server:serverstate': Server}
+
+
+class Server(Resource):
+    """BIG-IP® GTM server resource"""
+    def __init__(self, servers):
+        super(Server, self).__init__(servers)
+        self._meta_data['required_json_kind'] = 'tm:gtm:server:serverstate'
+        self._meta_data['required_creation_parameters'].update(
+            ('datacenter', 'addresses'))
+        self._meta_data['attribute_registry'] = {
+            'tm:gtm:server:virtual-servers:virtual-serverscollectionstate':
+                Virtual_Servers_s
+        }
+
+
+class Virtual_Servers_s(Collection):
+    """BIG-IP® GTM virtual server sub-collection"""
+    def __init__(self, server):
+        super(Virtual_Servers_s, self).__init__(server)
+        self._meta_data['allowed_lazy_attributes'] = [Virtual_Servers]
+        self._meta_data['required_json_kind'] = \
+            'tm:gtm:server:virtual-servers:virtual-serverscollectionstate'
+        self._meta_data['attribute_registry'] = {
+            'tm:gtm:server:virtual-servers:virtual-serversstate':
+                Virtual_Servers}
+
+
+class Virtual_Servers(Resource):
+    """BIG-IP® GTM virtual server resource"""
+    def __init__(self, virtual_servers_s):
+        super(Virtual_Servers, self).__init__(virtual_servers_s)
+        self._meta_data['required_creation_parameters'].update((
+            'destination',))
+        self._meta_data['required_json_kind'] = \
+            'tm:gtm:server:virtual-servers:virtual-serversstate'

f5/bigip/tm/gtm/test/test_server.py

@@ -0,0 +1,70 @@
+# Copyright 2015 F5 Networks Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+import mock
+import pytest
+
+from f5.bigip import ManagementRoot
+from f5.bigip.resource import MissingRequiredCreationParameter
+from f5.bigip.tm.gtm.server import Server
+from f5.bigip.tm.gtm.server import Virtual_Servers
+
+
+@pytest.fixture
+def FakeServer():
+    fake_servers = mock.MagicMock()
+    fake_server = Server(fake_servers)
+    return fake_server
+
+
+@pytest.fixture
+def FakeVS():
+    fake_server = mock.MagicMock()
+    fake_vs = Virtual_Servers(fake_server)
+    return fake_vs
+
+
+class TestCreate(object):
+    def test_create_two(self, fakeicontrolsession):
+        b = ManagementRoot('192.168.1.1', 'admin', 'admin')
+        s1 = b.tm.gtm.servers.server
+        s2 = b.tm.gtm.servers.server
+        assert s1 is not s2
+
+    def test_create_no_args(self, FakeServer):
+        with pytest.raises(MissingRequiredCreationParameter):
+            FakeServer.create()
+
+    def test_create_no_datacenter(self, FakeServer):
+        with pytest.raises(MissingRequiredCreationParameter):
+            FakeServer.create(name='fakeserver',
+                              addresses=[{'name': '1.1.1.1'}])
+
+    def test_create_no_address(self, FakeServer):
+        with pytest.raises(MissingRequiredCreationParameter):
+            FakeServer.create(name='fakeserver', datacenter='fakedc')
+
+
+class Test_VS_Subcoll(object):
+    def test_vs_attr_exists(self, fakeicontrolsession):
+        b = ManagementRoot('192.168.1.1', 'admin', 'admin')
+        s = b.tm.gtm.servers.server
+        test_meta = s._meta_data['attribute_registry']
+        kind = 'tm:gtm:server:virtual-servers:virtual-serverscollectionstate'
+        assert kind in test_meta.keys()
+
+    def test_create_no_args(self, FakeVS):
+        with pytest.raises(MissingRequiredCreationParameter):
+            FakeVS.create()