Exemplo n.º 1
0
 def test_main_no_deployments(self):
     with contextlib.nested(
         mock.patch('setup_chronos_job.parse_args',
                    return_value=self.fake_args,
                    autospec=True),
         mock.patch('paasta_tools.chronos_tools.load_chronos_config', autospec=True),
         mock.patch('paasta_tools.chronos_tools.get_chronos_client',
                    return_value=self.fake_client,
                    autospec=True),
         mock.patch('paasta_tools.chronos_tools.load_chronos_job_config',
                    return_value=self.fake_chronos_job_config,
                    autospec=True,
                    side_effect=NoDeploymentsAvailable),
         mock.patch('setup_chronos_job.setup_job',
                    return_value=(0, 'it_is_finished'),
                    autospec=True),
         mock.patch('setup_chronos_job.load_system_paasta_config', autospec=True),
         mock.patch('setup_chronos_job.send_event', autospec=True),
     ) as (
         parse_args_patch,
         load_chronos_config_patch,
         get_client_patch,
         load_chronos_job_config_patch,
         setup_job_patch,
         load_system_paasta_config_patch,
         send_event_patch
     ):
         load_system_paasta_config_patch.return_value.get_cluster = mock.MagicMock(return_value=self.fake_cluster)
         with raises(SystemExit) as excinfo:
             setup_chronos_job.main()
         assert excinfo.value.code == 0
 def test_main_no_deployments(self):
     with contextlib.nested(
             mock.patch('setup_chronos_job.parse_args',
                        return_value=self.fake_args,
                        autospec=True),
             mock.patch('paasta_tools.chronos_tools.load_chronos_config',
                        autospec=True),
             mock.patch('paasta_tools.chronos_tools.get_chronos_client',
                        return_value=self.fake_client,
                        autospec=True),
             mock.patch(
                 'paasta_tools.chronos_tools.load_chronos_job_config',
                 return_value=self.fake_chronos_job_config,
                 autospec=True,
                 side_effect=NoDeploymentsAvailable),
             mock.patch('setup_chronos_job.setup_job',
                        return_value=(0, 'it_is_finished'),
                        autospec=True),
             mock.patch('setup_chronos_job.load_system_paasta_config',
                        autospec=True),
             mock.patch('setup_chronos_job.send_event', autospec=True),
     ) as (parse_args_patch, load_chronos_config_patch, get_client_patch,
           load_chronos_job_config_patch, setup_job_patch,
           load_system_paasta_config_patch, send_event_patch):
         load_system_paasta_config_patch.return_value.get_cluster = mock.MagicMock(
             return_value=self.fake_cluster)
         with raises(SystemExit) as excinfo:
             setup_chronos_job.main()
         assert excinfo.value.code == 0
Exemplo n.º 3
0
    def test_main_success(self):
        expected_status = 0
        expected_output = "it_is_finished"
        fake_complete_job_config = {"foo": "bar"}
        with contextlib.nested(
            mock.patch("setup_chronos_job.parse_args", return_value=self.fake_args, autospec=True),
            mock.patch("paasta_tools.chronos_tools.load_chronos_config", autospec=True),
            mock.patch("paasta_tools.chronos_tools.get_chronos_client", return_value=self.fake_client, autospec=True),
            mock.patch(
                "paasta_tools.chronos_tools.create_complete_config",
                return_value=fake_complete_job_config,
                autospec=True,
            ),
            mock.patch("setup_chronos_job.setup_job", return_value=(expected_status, expected_output), autospec=True),
            mock.patch("setup_chronos_job.send_event", autospec=True),
            mock.patch("setup_chronos_job.load_system_paasta_config", autospec=True),
            mock.patch("sys.exit", autospec=True),
        ) as (
            parse_args_patch,
            load_chronos_config_patch,
            get_client_patch,
            create_complete_config_patch,
            setup_job_patch,
            send_event_patch,
            load_system_paasta_config_patch,
            sys_exit_patch,
        ):
            load_system_paasta_config_patch.return_value.get_cluster = mock.MagicMock(return_value=self.fake_cluster)
            setup_chronos_job.main()

            parse_args_patch.assert_called_once_with()
            get_client_patch.assert_called_once_with(load_chronos_config_patch.return_value)
            setup_job_patch.assert_called_once_with(
                service=self.fake_service,
                instance=self.fake_instance,
                complete_job_config=fake_complete_job_config,
                client=self.fake_client,
                cluster=self.fake_cluster,
            )
            send_event_patch.assert_called_once_with(
                service=self.fake_service,
                instance=self.fake_instance,
                soa_dir=self.fake_args.soa_dir,
                status=expected_status,
                output=expected_output,
            )
            sys_exit_patch.assert_called_once_with(0)
 def test_main_bad_chronos_job_config_notifies_user(self):
     with contextlib.nested(
             mock.patch('setup_chronos_job.parse_args',
                        return_value=self.fake_args,
                        autospec=True),
             mock.patch('paasta_tools.chronos_tools.load_chronos_config',
                        autospec=True),
             mock.patch('paasta_tools.chronos_tools.get_chronos_client',
                        return_value=self.fake_client,
                        autospec=True),
             mock.patch(
                 'paasta_tools.chronos_tools.load_chronos_job_config',
                 return_value=self.fake_chronos_job_config,
                 autospec=True,
                 side_effect=chronos_tools.InvalidChronosConfigError(
                     'test bad configuration')),
             mock.patch('setup_chronos_job.setup_job',
                        return_value=(0, 'it_is_finished'),
                        autospec=True),
             mock.patch('setup_chronos_job.load_system_paasta_config',
                        autospec=True),
             mock.patch('setup_chronos_job.send_event', autospec=True),
     ) as (
             parse_args_patch,
             load_chronos_config_patch,
             get_client_patch,
             load_chronos_job_config_patch,
             setup_job_patch,
             load_system_paasta_config_patch,
             send_event_patch,
     ):
         load_system_paasta_config_patch.return_value.get_cluster = mock.MagicMock(
             return_value=self.fake_cluster)
         with raises(SystemExit) as excinfo:
             setup_chronos_job.main()
         assert excinfo.value.code == 0
         expected_error_msg = (
             "Could not read chronos configuration file for %s in cluster %s\nError was: test bad configuration"
             % (compose_job_id(self.fake_service,
                               self.fake_instance), self.fake_cluster))
         send_event_patch.assert_called_once_with(
             service=self.fake_service,
             instance=self.fake_instance,
             soa_dir=self.fake_args.soa_dir,
             status=Status.CRITICAL,
             output=expected_error_msg)
Exemplo n.º 5
0
 def test_main_bad_chronos_job_config_notifies_user(self):
     with contextlib.nested(
         mock.patch('setup_chronos_job.parse_args',
                    return_value=self.fake_args,
                    autospec=True),
         mock.patch('paasta_tools.chronos_tools.load_chronos_config', autospec=True),
         mock.patch('paasta_tools.chronos_tools.get_chronos_client',
                    return_value=self.fake_client,
                    autospec=True),
         mock.patch('paasta_tools.chronos_tools.load_chronos_job_config',
                    return_value=self.fake_chronos_job_config,
                    autospec=True,
                    side_effect=chronos_tools.InvalidChronosConfigError('test bad configuration')),
         mock.patch('setup_chronos_job.setup_job',
                    return_value=(0, 'it_is_finished'),
                    autospec=True),
         mock.patch('setup_chronos_job.load_system_paasta_config', autospec=True),
         mock.patch('setup_chronos_job.send_event', autospec=True),
     ) as (
         parse_args_patch,
         load_chronos_config_patch,
         get_client_patch,
         load_chronos_job_config_patch,
         setup_job_patch,
         load_system_paasta_config_patch,
         send_event_patch,
     ):
         load_system_paasta_config_patch.return_value.get_cluster = mock.MagicMock(return_value=self.fake_cluster)
         with raises(SystemExit) as excinfo:
             setup_chronos_job.main()
         assert excinfo.value.code == 0
         expected_error_msg = (
             "Could not read chronos configuration file for %s in cluster %s\nError was: test bad configuration"
             % (compose_job_id(self.fake_service, self.fake_instance), self.fake_cluster)
         )
         send_event_patch.assert_called_once_with(
             service=self.fake_service,
             instance=self.fake_instance,
             soa_dir=self.fake_args.soa_dir,
             status=Status.CRITICAL,
             output=expected_error_msg
         )
    def test_main_success(self):
        expected_status = 0
        expected_output = 'it_is_finished'
        fake_complete_job_config = {'foo': 'bar'}
        with contextlib.nested(
                mock.patch('setup_chronos_job.parse_args',
                           return_value=self.fake_args,
                           autospec=True),
                mock.patch('paasta_tools.chronos_tools.load_chronos_config',
                           autospec=True),
                mock.patch('paasta_tools.chronos_tools.get_chronos_client',
                           return_value=self.fake_client,
                           autospec=True),
                mock.patch(
                    'paasta_tools.chronos_tools.load_chronos_job_config',
                    return_value=self.fake_chronos_job_config,
                    autospec=True),
                mock.patch('paasta_tools.chronos_tools.create_complete_config',
                           return_value=fake_complete_job_config,
                           autospec=True),
                mock.patch('setup_chronos_job.setup_job',
                           return_value=(expected_status, expected_output),
                           autospec=True),
                mock.patch('setup_chronos_job.send_event', autospec=True),
                mock.patch('setup_chronos_job.load_system_paasta_config',
                           autospec=True),
                mock.patch('sys.exit', autospec=True),
        ) as (
                parse_args_patch,
                load_chronos_config_patch,
                get_client_patch,
                load_chronos_job_config_patch,
                create_complete_config_patch,
                setup_job_patch,
                send_event_patch,
                load_system_paasta_config_patch,
                sys_exit_patch,
        ):
            load_system_paasta_config_patch.return_value.get_cluster = mock.MagicMock(
                return_value=self.fake_cluster)
            setup_chronos_job.main()

            parse_args_patch.assert_called_once_with()
            get_client_patch.assert_called_once_with(
                load_chronos_config_patch.return_value)
            load_chronos_job_config_patch.assert_called_once_with(
                service=self.fake_service,
                instance=self.fake_instance,
                cluster=self.fake_cluster,
                soa_dir=self.fake_args.soa_dir,
            )
            setup_job_patch.assert_called_once_with(
                service=self.fake_service,
                instance=self.fake_instance,
                chronos_job_config=self.fake_chronos_job_config,
                complete_job_config=fake_complete_job_config,
                client=self.fake_client,
                cluster=self.fake_cluster,
            )
            send_event_patch.assert_called_once_with(
                service=self.fake_service,
                instance=self.fake_instance,
                soa_dir=self.fake_args.soa_dir,
                status=expected_status,
                output=expected_output,
            )
            sys_exit_patch.assert_called_once_with(0)