|
@@ -19,9 +19,10 @@ limitations under the License.
|
|
'''
|
|
'''
|
|
from mock.mock import MagicMock, call, patch
|
|
from mock.mock import MagicMock, call, patch
|
|
from stacks.utils.RMFTestCase import *
|
|
from stacks.utils.RMFTestCase import *
|
|
-import datetime, sys
|
|
|
|
|
|
+import datetime, sys, socket
|
|
import resource_management.libraries.functions
|
|
import resource_management.libraries.functions
|
|
@patch.object(resource_management.libraries.functions, "get_unique_id_and_date", new = MagicMock(return_value=''))
|
|
@patch.object(resource_management.libraries.functions, "get_unique_id_and_date", new = MagicMock(return_value=''))
|
|
|
|
+@patch("socket.socket", new = MagicMock())
|
|
class TestServiceCheck(RMFTestCase):
|
|
class TestServiceCheck(RMFTestCase):
|
|
|
|
|
|
@patch("sys.exit")
|
|
@patch("sys.exit")
|