def setUp(self):
     User = get_user_model()
     user = User.objects.create_user('tester', '*****@*****.**', 'tester')
     self.controller = Controller(name='Test App',
                                  marathon_cmd='ping',
                                  owner=user)
     self.controller.save()
     setup_responses_for_log_tests(self.controller)
     self.general_im = GeneralInfrastructureManager()
     self.controller_im = self.controller.infra_manager
예제 #2
0
 def setUp(self):
     User = get_user_model()
     user = User.objects.create_user(
         'tester', '*****@*****.**', 'tester')
     self.controller = Controller(
         name='Test App', marathon_cmd='ping', owner=user)
     self.controller.save()
     setup_responses_for_log_tests(self.controller)
     self.general_im = GeneralInfrastructureManager()
     self.controller_im = self.controller.infra_manager
예제 #3
0
class GeneralInfrastructureManagerTest(TestCase):

    def setUp(self):
        User = get_user_model()
        user = User.objects.create_user(
            'tester', '*****@*****.**', 'tester')
        self.controller = Controller(
            name='Test App', marathon_cmd='ping', owner=user)
        self.controller.save()
        setup_responses_for_log_tests(self.controller)
        self.general_im = GeneralInfrastructureManager()
        self.controller_im = self.controller.infra_manager

    @responses.activate
    def test_get_marathon_app(self):
        app = self.general_im.get_marathon_app(self.controller.app_id)
        self.assertEqual(app['id'], '/%s' % (self.controller.app_id,))

    @responses.activate
    def test_get_marathon_app_tasks(self):
        [task] = self.general_im.get_marathon_app_tasks(self.controller.app_id)
        self.assertEqual(task['appId'], '/%s' % (self.controller.app_id,))
        self.assertEqual(
            task['id'], '%s.the-task-id' % (self.controller.app_id,))
        self.assertEqual(task['ports'], [8898])
        self.assertEqual(task['host'], 'worker-machine-1')

    @responses.activate
    def test_get_marathon_info(self):
        info = self.general_im.get_marathon_info()
        self.assertEqual(info['name'], 'marathon')
        self.assertEqual(info['frameworkId'], 'the-framework-id')

    @responses.activate
    def test_get_worker_info(self):
        worker = self.general_im.get_worker_info('worker-machine-1')
        self.assertEqual(worker['id'], 'worker-machine-id')

    @responses.activate
    def test_get_app_log_info(self):
        [info] = self.general_im.get_app_log_info(self.controller.app_id)
        self.assertEqual(
            info,
            {
                'task_host': 'worker-machine-1',
                'task_id': '%s.the-task-id' % (self.controller.app_id,),
                'task_dir': (
                    'worker-machine-id/frameworks/the-framework-id'
                    '/executors/%s.the-task-id/runs/latest') % (
                        self.controller.app_id,),
            }
        )

    @responses.activate
    def test_get_task_log_info(self):
        info = self.general_im.get_task_log_info(
            self.controller.app_id,
            '%s.the-task-id' % (self.controller.app_id,),
            'worker-machine-1')
        self.assertEqual(
            info,
            {
                'task_host': 'worker-machine-1',
                'task_id': '%s.the-task-id' % (self.controller.app_id,),
                'task_dir': (
                    'worker-machine-id/frameworks/the-framework-id'
                    '/executors/%s.the-task-id/runs/latest') % (
                        self.controller.app_id,),
            }
        )

    @responses.activate
    def test_controller_infra_manager_get_marathon_app(self):
        app = self.controller_im.get_controller_marathon_app()
        self.assertEqual(app['id'], '/%s' % (self.controller.app_id,))

    @responses.activate
    def test_controller_infra_manager_get_controller_log_info(self):
        [info] = self.controller_im.get_controller_log_info()
        self.assertEqual(
            info,
            {
                'task_host': 'worker-machine-1',
                'task_id': '%s.the-task-id' % (self.controller.app_id,),
                'task_dir': (
                    'worker-machine-id/frameworks/the-framework-id'
                    '/executors/%s.the-task-id/runs/latest') % (
                        self.controller.app_id,),
            }
        )

    @responses.activate
    def test_controller_infra_manager_get_controller_task_log_info(self):
        info = self.controller_im.get_controller_task_log_info(
            '%s.the-task-id' % (self.controller.app_id,))
        self.assertEqual(
            info,
            {
                'task_host': 'worker-machine-1',
                'task_id': '%s.the-task-id' % (self.controller.app_id,),
                'task_dir': (
                    'worker-machine-id/frameworks/the-framework-id'
                    '/executors/%s.the-task-id/runs/latest') % (
                        self.controller.app_id,),
            }
        )

    @responses.activate
    def test_controller_infra_manager_get_controller_non_existent(self):
        self.assertRaises(
            InfrastructureError,
            self.controller_im.get_controller_task_log_info,
            'non-existing-task-id')
class GeneralInfrastructureManagerTest(TestCase):
    def setUp(self):
        User = get_user_model()
        user = User.objects.create_user('tester', '*****@*****.**', 'tester')
        self.controller = Controller(name='Test App',
                                     marathon_cmd='ping',
                                     owner=user)
        self.controller.save()
        setup_responses_for_log_tests(self.controller)
        self.general_im = GeneralInfrastructureManager()
        self.controller_im = self.controller.infra_manager

    @responses.activate
    def test_get_marathon_app(self):
        app = self.general_im.get_marathon_app(self.controller.app_id)
        self.assertEqual(app['id'], '/%s' % (self.controller.app_id, ))

    @responses.activate
    def test_get_marathon_app_tasks(self):
        [task] = self.general_im.get_marathon_app_tasks(self.controller.app_id)
        self.assertEqual(task['appId'], '/%s' % (self.controller.app_id, ))
        self.assertEqual(task['id'],
                         '%s.the-task-id' % (self.controller.app_id, ))
        self.assertEqual(task['ports'], [8898])
        self.assertEqual(task['host'], 'worker-machine-1')

    @responses.activate
    def test_get_marathon_info(self):
        info = self.general_im.get_marathon_info()
        self.assertEqual(info['name'], 'marathon')
        self.assertEqual(info['frameworkId'], 'the-framework-id')

    @responses.activate
    def test_get_worker_info(self):
        worker = self.general_im.get_worker_info('worker-machine-1')
        self.assertEqual(worker['id'], 'worker-machine-id')

    @responses.activate
    def test_get_app_log_info(self):
        [info] = self.general_im.get_app_log_info(self.controller.app_id)
        self.assertEqual(
            info, {
                'task_host':
                'worker-machine-1',
                'task_id':
                '%s.the-task-id' % (self.controller.app_id, ),
                'task_dir': ('worker-machine-id/frameworks/the-framework-id'
                             '/executors/%s.the-task-id/runs/latest') %
                (self.controller.app_id, ),
            })

    @responses.activate
    def test_get_task_log_info(self):
        info = self.general_im.get_task_log_info(
            self.controller.app_id,
            '%s.the-task-id' % (self.controller.app_id, ), 'worker-machine-1')
        self.assertEqual(
            info, {
                'task_host':
                'worker-machine-1',
                'task_id':
                '%s.the-task-id' % (self.controller.app_id, ),
                'task_dir': ('worker-machine-id/frameworks/the-framework-id'
                             '/executors/%s.the-task-id/runs/latest') %
                (self.controller.app_id, ),
            })

    @responses.activate
    def test_controller_infra_manager_get_marathon_app(self):
        app = self.controller_im.get_controller_marathon_app()
        self.assertEqual(app['id'], '/%s' % (self.controller.app_id, ))

    @responses.activate
    def test_controller_infra_manager_get_controller_log_info(self):
        [info] = self.controller_im.get_controller_log_info()
        self.assertEqual(
            info, {
                'task_host':
                'worker-machine-1',
                'task_id':
                '%s.the-task-id' % (self.controller.app_id, ),
                'task_dir': ('worker-machine-id/frameworks/the-framework-id'
                             '/executors/%s.the-task-id/runs/latest') %
                (self.controller.app_id, ),
            })

    @responses.activate
    def test_controller_infra_manager_get_controller_task_log_info(self):
        info = self.controller_im.get_controller_task_log_info(
            '%s.the-task-id' % (self.controller.app_id, ))
        self.assertEqual(
            info, {
                'task_host':
                'worker-machine-1',
                'task_id':
                '%s.the-task-id' % (self.controller.app_id, ),
                'task_dir': ('worker-machine-id/frameworks/the-framework-id'
                             '/executors/%s.the-task-id/runs/latest') %
                (self.controller.app_id, ),
            })

    @responses.activate
    def test_controller_infra_manager_get_controller_non_existent(self):
        self.assertRaises(InfrastructureError,
                          self.controller_im.get_controller_task_log_info,
                          'non-existing-task-id')