Support Websocket profile

Author: zhaoqin-github

f5/bigip/tm/ltm/profile.py

@@ -94,6 +94,7 @@ def __init__(self, ltm):
             Wa_Caches,
             Web_Accelerations,
             Web_Securitys,
+            Websockets,
             Xmls]
 
 
@@ -1207,6 +1208,23 @@ def delete(self, **kwargs):
         )
 
 
+class Websockets(Collection):
+    """BIG-IP® Websocket profile collection."""
+    def __init__(self, profile):
+        super(Websockets, self).__init__(profile)
+        self._meta_data['allowed_lazy_attributes'] = [Websocket]
+        self._meta_data['attribute_registry'] = \
+            {'tm:ltm:profile:websocket:websocketstate': Websocket}
+
+
+class Websocket(Resource):
+    """BIG-IP® Websocket profile resource."""
+    def __init__(self, Websockets):
+        super(Websocket, self).__init__(Websockets)
+        self._meta_data['required_json_kind'] = \
+            'tm:ltm:profile:websocket:websocketstate'
+
+
 class Xmls(Collection):
     """BIG-IP® Xml profile collection."""
     def __init__(self, profile):

f5/bigip/tm/ltm/test/functional/test_profile.py

@@ -1166,6 +1166,18 @@ def test_load(self, request, mgmt_root):
 
 # End Web Security tests
 
+# Begin Websocket tests
+
+
+class TestWebsocket(object):
+    def test_load(self, request, mgmt_root):
+        ws1 = mgmt_root.tm.ltm.profile.\
+            websockets.websocket.load(name='websocket')
+        assert ws1.name == 'websocket'
+
+
+# End Websocket tests
+
 # Begin Xml tests