|
@@ -23,21 +23,18 @@ import urllib2
|
|
|
|
|
|
import logging
|
|
|
from unittest import TestCase
|
|
|
-from only_for_platform import get_platform, PLATFORM_WINDOWS
|
|
|
+from only_for_platform import get_platform, os_distro_value, PLATFORM_WINDOWS
|
|
|
|
|
|
-from mock.mock import patch, MagicMock
|
|
|
+from ambari_commons.os_check import OSCheck
|
|
|
|
|
|
-if get_platform() != PLATFORM_WINDOWS:
|
|
|
- os_distro_value = ('Suse','11','Final')
|
|
|
-else:
|
|
|
- os_distro_value = ('win2012serverr2','6.3','WindowsServer')
|
|
|
+from mock.mock import patch, MagicMock
|
|
|
|
|
|
with patch("platform.linux_distribution", return_value = os_distro_value):
|
|
|
from ambari_commons import OSCheck
|
|
|
- from application_metric_map import ApplicationMetricMap
|
|
|
- from config_reader import Configuration
|
|
|
- from emitter import Emitter
|
|
|
- from stop_handler import bind_signal_handlers
|
|
|
+ from core.application_metric_map import ApplicationMetricMap
|
|
|
+ from core.config_reader import Configuration
|
|
|
+ from core.emitter import Emitter
|
|
|
+ from core.stop_handler import bind_signal_handlers
|
|
|
|
|
|
logger = logging.getLogger()
|
|
|
|