示例#1
0
class WhenTestingPublishStats(unittest.TestCase):
    def setUp(self):
        self.conf = MagicMock()
        self.conf.status_update.worker_status_interval = 60
        self.get_config = MagicMock(return_value=self.conf)
        self.config = WorkerConfiguration(
            personality='worker.correlation',
            personality_module='meniscus.personas.worker.correlation.app',
            worker_id='fgc7104e-8d93-47dc-a49a-8fb0d39e5192',
            worker_token='bbd6307f-8d93-47dc-a49a-8fb0d39e5192',
            coordinator_uri='http://192.168.1.2/v1')
        self.system_info = SystemInfo()
        self.info_format = MagicMock(return_value=self.system_info.format())
        self.request_uri = "{0}/worker/{1}/status".format(
            self.config.coordinator_uri, self.config.worker_id)
        self.req_body = jsonutils.dumps({
            'worker_status': {
                'status': 'online',
                'system_info': self.system_info.format()
            }
        })
        self.get_config = MagicMock(return_value=self.config)
        self.resp = requests.Response()
        self.http_request = MagicMock(return_value=self.resp)

    def test_http_request_called(self):
        with patch.object(
                ConfigCache, 'get_config', self.get_config), patch(
                'meniscus.personas.common.publish_stats.http_request',
                self.http_request), patch(
                'meniscus.personas.common.publish_stats.get_config',
                self.get_config),\
            patch.object(
                SystemInfo, 'format',
                self.info_format):
            publish_worker_stats()
            self.http_request.assert_called_once_with(
                url=self.request_uri,
                json_payload=self.req_body,
                http_verb='PUT'
            )
示例#2
0
class WhenTestingPublishStats(unittest.TestCase):
    def setUp(self):
        self.conf = MagicMock()
        self.conf.status_update.worker_status_interval = 60
        self.get_config = MagicMock(return_value=self.conf)
        self.config = WorkerConfiguration(
            personality='worker.correlation',
            personality_module='meniscus.personas.worker.correlation.app',
            worker_id='fgc7104e-8d93-47dc-a49a-8fb0d39e5192',
            worker_token='bbd6307f-8d93-47dc-a49a-8fb0d39e5192',
            coordinator_uri='http://192.168.1.2/v1')
        self.system_info = SystemInfo()
        self.info_format = MagicMock(return_value=self.system_info.format())
        self.request_uri = "{0}/worker/{1}/status".format(
            self.config.coordinator_uri, self.config.worker_id)
        self.req_body = jsonutils.dumps({
            'worker_status': {
                'status': 'online',
                'system_info': self.system_info.format()
            }
        })
        self.get_config = MagicMock(return_value=self.config)
        self.resp = requests.Response()
        self.http_request = MagicMock(return_value=self.resp)

    def test_http_request_called(self):
        with patch.object(
                ConfigCache, 'get_config', self.get_config), patch(
                'meniscus.personas.common.publish_stats.http_request',
                self.http_request), patch(
                'meniscus.personas.common.publish_stats.get_config',
                self.get_config),\
            patch.object(
                SystemInfo, 'format',
                self.info_format):
            publish_worker_stats()
            self.http_request.assert_called_once_with(
                url=self.request_uri,
                json_payload=self.req_body,
                http_verb='PUT')
示例#3
0
 def setUp(self):
     self.conf = MagicMock()
     self.conf.status_update.worker_status_interval = 60
     self.get_config = MagicMock(return_value=self.conf)
     self.config = WorkerConfiguration(
         personality='worker.correlation',
         personality_module='meniscus.personas.worker.correlation.app',
         worker_id='fgc7104e-8d93-47dc-a49a-8fb0d39e5192',
         worker_token='bbd6307f-8d93-47dc-a49a-8fb0d39e5192',
         coordinator_uri='http://192.168.1.2/v1')
     self.system_info = SystemInfo()
     self.info_format = MagicMock(return_value=self.system_info.format())
     self.request_uri = "{0}/worker/{1}/status".format(
         self.config.coordinator_uri, self.config.worker_id)
     self.req_body = jsonutils.dumps({
         'worker_status': {
             'status': 'online',
             'system_info': self.system_info.format()
         }
     })
     self.get_config = MagicMock(return_value=self.config)
     self.resp = requests.Response()
     self.http_request = MagicMock(return_value=self.resp)
示例#4
0
 def setUp(self):
     self.conf = MagicMock()
     self.conf.status_update.worker_status_interval = 60
     self.get_config = MagicMock(return_value=self.conf)
     self.config = WorkerConfiguration(
         personality='worker.correlation',
         personality_module='meniscus.personas.worker.correlation.app',
         worker_id='fgc7104e-8d93-47dc-a49a-8fb0d39e5192',
         worker_token='bbd6307f-8d93-47dc-a49a-8fb0d39e5192',
         coordinator_uri='http://192.168.1.2/v1')
     self.system_info = SystemInfo()
     self.info_format = MagicMock(return_value=self.system_info.format())
     self.request_uri = "{0}/worker/{1}/status".format(
         self.config.coordinator_uri, self.config.worker_id)
     self.req_body = jsonutils.dumps({
         'worker_status': {
             'status': 'online',
             'system_info': self.system_info.format()
         }
     })
     self.get_config = MagicMock(return_value=self.config)
     self.resp = requests.Response()
     self.http_request = MagicMock(return_value=self.resp)