update unit tests

Author: saimonation

tests/ut/edge/test_network.py

@@ -49,6 +49,12 @@ def setUp(self):
         self._proxy_user = 'admin'
         self._proxy_pass = 'password'
 
+        self._timeout = {
+            'timeout': {
+                'sock_read': 120
+            }
+        }
+
     def test_network_status(self):
         get_response = 'Success'
         self._init_filer(get_response=get_response)
@@ -137,7 +143,7 @@ def test_tcp_connect_success(self):
 
         ret = network.Network(self._filer).tcp_connect(TCPService(self._tcp_connect_address, self._tcp_connect_port))
 
-        self._filer.api.execute.assert_called_once_with('/status/network', 'tcpconnect', mock.ANY)
+        self._filer.api.execute.assert_called_once_with('/status/network', 'tcpconnect', mock.ANY, self._timeout)
         self._filer.tasks.wait.assert_called_once_with(self._task_id)
 
         expected_param = self._get_tcp_connect_object()
@@ -157,7 +163,7 @@ def test_tcp_connect_failure(self):
 
         ret = network.Network(self._filer).tcp_connect(TCPService(self._tcp_connect_address, self._tcp_connect_port))
 
-        self._filer.api.execute.assert_called_once_with('/status/network', 'tcpconnect', mock.ANY)
+        self._filer.api.execute.assert_called_once_with('/status/network', 'tcpconnect', mock.ANY, self._timeout)
         self._filer.tasks.wait.assert_called_once_with(self._task_id)
 
         expected_param = self._get_tcp_connect_object()
@@ -200,7 +206,7 @@ def test_tcp_connect_task_error(self):
 
         ret = network.Network(self._filer).tcp_connect(TCPService(self._tcp_connect_address, self._tcp_connect_port))
 
-        self._filer.api.execute.assert_called_once_with('/status/network', 'tcpconnect', mock.ANY)
+        self._filer.api.execute.assert_called_once_with('/status/network', 'tcpconnect', mock.ANY, self._timeout)
         self._filer.tasks.wait.assert_called_once_with(self._task_id)
 
         expected_param = self._get_tcp_connect_object()