Python ensure_instance_group_managers_activeの例

コード例 #1

ファイル: parse_test.py プロジェクト: onecityuni/luci-py

    def test_drains_and_reactivates(self):
        """Ensures that the active are drained and the drained are reactivated."""
        template_cfg = config_pb2.InstanceTemplateConfig.InstanceTemplate(
            base_name='base-name', )
        manager_cfgs = config_pb2.InstanceGroupManagerConfig(managers=[
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone1',
            ),
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone2',
            ),
        ], ).managers
        expected_active_keys = [
            parse.get_instance_group_manager_key(template_cfg,
                                                 manager_cfgs[0]),
            parse.get_instance_group_manager_key(template_cfg,
                                                 manager_cfgs[1]),
        ]
        instance_template_revision = models.InstanceTemplateRevision(
            active=[
                ndb.Key(models.InstanceGroupManager, 'fake-key'),
            ],
            drained=expected_active_keys,
        )

        self.failUnless(
            parse.ensure_instance_group_managers_active(
                template_cfg, manager_cfgs, instance_template_revision))
        self.assertItemsEqual(instance_template_revision.active,
                              expected_active_keys)
        self.assertEqual(instance_template_revision.drained[0].id(),
                         'fake-key')

コード例 #2

ファイル: parse_test.py プロジェクト: mellowdistrict/luci-py

  def test_drains_and_reactivates(self):
    """Ensures that the active are drained and the drained are reactivated."""
    template_cfg = config_pb2.InstanceTemplateConfig.InstanceTemplate(
        base_name='base-name',
    )
    manager_cfgs = config_pb2.InstanceGroupManagerConfig(
        managers=[
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone1',
            ),
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone2',
            ),
        ],
    ).managers
    expected_active_keys = [
        parse.get_instance_group_manager_key(template_cfg, manager_cfgs[0]),
        parse.get_instance_group_manager_key(template_cfg, manager_cfgs[1]),
    ]
    instance_template_revision = models.InstanceTemplateRevision(
        active=[
            ndb.Key(models.InstanceGroupManager, 'fake-key'),
        ],
        drained=expected_active_keys,
    )

    self.failUnless(parse.ensure_instance_group_managers_active(
        template_cfg, manager_cfgs, instance_template_revision))
    self.assertItemsEqual(
        instance_template_revision.active, expected_active_keys)
    self.assertEqual(instance_template_revision.drained[0].id(), 'fake-key')

コード例 #3

ファイル: parse_test.py プロジェクト: onecityuni/luci-py

    def test_activates(self):
        """Ensures that the instance group managers are activated."""
        template_cfg = config_pb2.InstanceTemplateConfig.InstanceTemplate(
            base_name='base-name', )
        manager_cfgs = config_pb2.InstanceGroupManagerConfig(managers=[
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone1',
            ),
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone2',
            ),
        ], ).managers
        expected_active_keys = [
            parse.get_instance_group_manager_key(template_cfg,
                                                 manager_cfgs[0]),
            parse.get_instance_group_manager_key(template_cfg,
                                                 manager_cfgs[1]),
        ]
        instance_template_revision = models.InstanceTemplateRevision()

        self.failUnless(
            parse.ensure_instance_group_managers_active(
                template_cfg, manager_cfgs, instance_template_revision))
        self.assertItemsEqual(instance_template_revision.active,
                              expected_active_keys)

コード例 #4

ファイル: parse_test.py プロジェクト: mellowdistrict/luci-py

  def test_activates(self):
    """Ensures that the instance group managers are activated."""
    template_cfg = config_pb2.InstanceTemplateConfig.InstanceTemplate(
        base_name='base-name',
    )
    manager_cfgs = config_pb2.InstanceGroupManagerConfig(
        managers=[
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone1',
            ),
            config_pb2.InstanceGroupManagerConfig.InstanceGroupManager(
                template_base_name='base-name',
                zone='zone2',
            ),
        ],
    ).managers
    expected_active_keys = [
        parse.get_instance_group_manager_key(template_cfg, manager_cfgs[0]),
        parse.get_instance_group_manager_key(template_cfg, manager_cfgs[1]),
    ]
    instance_template_revision = models.InstanceTemplateRevision()

    self.failUnless(parse.ensure_instance_group_managers_active(
        template_cfg, manager_cfgs, instance_template_revision))
    self.assertItemsEqual(
        instance_template_revision.active, expected_active_keys)