def test_cron_update_buckets_change_reservation(self):
config.Bucket(
id='bucket',
project_id='foo',
revision='deadbeef',
config_content='name: "bucket"',
).put()
buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[project_config_pb2.Bucket(name='bucket')])
self.mock(config_component, 'get_project_configs', mock.Mock())
config_component.get_project_configs.return_value = {
'bar': ('deadbeef', buildbucket_cfg),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
expected = [
config.Bucket(
id='bucket',
project_id='bar',
revision='deadbeef',
config_content='name: "bucket"\n',
)
]
self.assertEqual(actual, expected)
def test_cron_update_buckets_change_reservation(self):
config.Bucket(
id='bucket',
project_id='foo',
revision='deadbeef',
config_content='name: "bucket"',
).put()
buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[project_config_pb2.Bucket(name='bucket')]
)
self.mock(config_component, 'get_project_configs', mock.Mock())
config_component.get_project_configs.return_value = {
'bar': ('deadbeef', buildbucket_cfg),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
expected = [
config.Bucket(
id='bucket',
project_id='bar',
revision='deadbeef',
config_content='name: "bucket"\n',
)
]
self.assertEqual(actual, expected)
def test_cron_update_buckets_change_reservation(self, get_project_configs):
buildbucket_cfg = parse_cfg('''buckets{ name: "bucket" }''')
get_project_configs.return_value = {
'bar': ('deadbeef', buildbucket_cfg, None),
}
config.Bucket(
id='bucket',
project_id='foo',
revision='deadbeef',
config_content='name: "bucket"',
config_content_binary=text_to_binary('name: "bucket"'),
).put()
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
expected = [
config.Bucket(
id='bucket',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='bar',
revision='deadbeef',
config_content='name: "bucket"\n',
config_content_binary=text_to_binary('name: "bucket"\n'),
)
]
self.assertEqual(actual, expected)
def test_cron_update_buckets_with_broken_configs(self, get_project_configs):
config.put_bucket('chromium', 'deadbeef', MASTER_TRYSERVER_CHROMIUM_LINUX)
get_project_configs.return_value = {
'chromium': (
'new!', None, config_component.ConfigFormatError('broken!')
),
}
config.cron_update_buckets()
# We must not delete buckets defined in a project that currently have a
# broken config.
bucket_id = 'chromium/' + MASTER_TRYSERVER_CHROMIUM_LINUX.name
_, actual = config.get_bucket(bucket_id)
self.assertEqual(actual, MASTER_TRYSERVER_CHROMIUM_LINUX)
def test_cron_update_buckets_with_broken_configs(self,
get_project_configs):
bucket = config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT),
)
bucket.put()
get_project_configs.return_value = {
'chromium':
('new!', None, config_component.ConfigFormatError('broken!')),
}
config.cron_update_buckets()
# We must not delete buckets defined in a project that currently have a
# broken config.
actual = bucket.key.get()
self.assertEqual(bucket, actual)
def test_cron_update_buckets_with_existing(self):
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
).put()
# Will not be updated.
config.Bucket(
id='master.tryserver.v8',
project_id='v8',
revision='deadbeef',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
).put()
# Will be deleted.
config.Bucket(
id='master.tryserver.chromium.win',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT,
).put()
chromium_buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[
project_config_pb2.Bucket(
name='master.tryserver.chromium.linux',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.READER, group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER, group='tryjob-access'),
],
),
# Will be added.
project_config_pb2.Bucket(
name='master.tryserver.chromium.mac',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.READER, group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER, group='tryjob-access'),
],
),
])
v8_buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[
# Reservation will fail.
project_config_pb2.Bucket(
name='master.tryserver.chromium.linux',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.WRITER, group='v8-team')
],
),
# Will not be updated.
project_config_pb2.Bucket(
name='master.tryserver.v8',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.WRITER, group='v8-team')
],
),
],
)
self.mock(config_component, 'get_project_configs', mock.Mock())
config_component.get_project_configs.return_value = {
'chromium': ('new!', chromium_buildbucket_cfg),
'v8': ('deadbeef', v8_buildbucket_cfg),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key.id())
expected = [
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='new!',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.chromium.mac',
project_id='chromium',
revision='new!',
config_content=MASTER_TRYSERVER_CHROMIUM_MAC_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.v8',
project_id='v8',
revision='deadbeef',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
),
]
self.assertEqual(actual, expected)
def test_cron_update_buckets(self):
chromium_buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[
project_config_pb2.Bucket(
name='master.tryserver.chromium.linux',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.READER, group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER, group='tryjob-access'),
],
),
project_config_pb2.Bucket(
name='master.tryserver.chromium.win',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.READER, group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER, group='tryjob-access'),
],
),
])
v8_buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[
project_config_pb2.Bucket(
name='master.tryserver.v8',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.WRITER, group='v8-team')
],
),
]
)
test_buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[
project_config_pb2.Bucket(
name='master.tryserver.test',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.WRITER, identity='*****@*****.**')
],
),
]
)
self.mock(config_component, 'get_project_configs', mock.Mock())
config_component.get_project_configs.return_value = {
'chromium': ('deadbeef', chromium_buildbucket_cfg),
'v8': (None, v8_buildbucket_cfg),
'test': ('babe', test_buildbucket_cfg),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key)
expected = [
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.chromium.win',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.test',
project_id='test',
revision='babe',
config_content=MASTER_TRYSERVER_TEST_CONFIG_TEXT
),
config.Bucket(
id='master.tryserver.v8',
project_id='v8',
revision='sha1:cfc761d7a953a72ddea8f3d4c9a28e69777ca22c',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
),
]
self.assertEqual(actual, expected)
def get(self):
config.cron_update_buckets()
def test_cron_update_buckets_with_existing(self, get_project_configs):
chromium_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.chromium.linux"
acls {
role: READER
group: "all"
}
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
buckets {
name: "master.tryserver.chromium.mac"
acls {
role: READER
group: "all"
}
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
''')
v8_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.chromium.linux"
acls {
role: WRITER
group: "v8-team"
}
}
buckets {
name: "master.tryserver.v8"
acls {
role: WRITER
group: "v8-team"
}
}
''')
get_project_configs.return_value = {
'chromium': ('new!', chromium_buildbucket_cfg, None),
'v8': ('deadbeef', v8_buildbucket_cfg, None),
}
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT),
).put()
# Will not be updated.
config.Bucket(
id='master.tryserver.v8',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='v8',
revision='deadbeef',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_V8_CONFIG_TEXT),
).put()
# Will be deleted.
config.Bucket(
id='master.tryserver.chromium.win',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT),
).put()
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key.id())
expected = [
config.Bucket(
id='master.tryserver.chromium.linux',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='chromium',
revision='new!',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT),
),
config.Bucket(
id='master.tryserver.chromium.mac',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='chromium',
revision='new!',
config_content=MASTER_TRYSERVER_CHROMIUM_MAC_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_MAC_CONFIG_TEXT),
),
config.Bucket(
id='master.tryserver.v8',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='v8',
revision='deadbeef',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_V8_CONFIG_TEXT),
),
]
self.assertEqual(actual, expected)
def test_cron_update_buckets(self, get_project_configs):
chromium_buildbucket_cfg = parse_cfg('''
acl_sets {
name: "public"
acls {
role: READER
group: "all"
}
acls {
role: READER
group: "all"
}
}
builder_mixins {
name: "recipe-x"
recipe {
repository: "https://example.com"
name: "x"
}
}
buckets {
name: "luci.chromium.try"
acl_sets: "public"
swarming {
builder_defaults {
dimensions: "pool:default"
}
builders {
name: "release"
mixins: "recipe-x"
dimensions: "os:Linux"
}
}
}
buckets {
name: "master.tryserver.chromium.linux"
acl_sets: "public"
acl_sets: "undefined_acl_set_will_cause_an_error_in_log_but_not_failure"
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
buckets {
name: "master.tryserver.chromium.win"
acls {
role: READER
group: "all"
}
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
''')
v8_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.v8"
acls {
role: WRITER
group: "v8-team"
}
}
''')
test_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.test"
acls {
role: WRITER
identity: "*****@*****.**"
}
}
''')
get_project_configs.return_value = {
'chromium': ('deadbeef', chromium_buildbucket_cfg, None),
'v8': (None, v8_buildbucket_cfg, None),
'test': ('babe', test_buildbucket_cfg, None),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key)
expected = [
config.Bucket(
id='luci.chromium.try',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='chromium',
revision='deadbeef',
config_content=LUCI_CHROMIUM_TRY_CONFIG_TEXT,
config_content_binary=text_to_binary(
LUCI_CHROMIUM_TRY_CONFIG_TEXT),
),
config.Bucket(
id='master.tryserver.chromium.linux',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT),
),
config.Bucket(
id='master.tryserver.chromium.win',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT),
),
config.Bucket(
id='master.tryserver.test',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='test',
revision='babe',
config_content=MASTER_TRYSERVER_TEST_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_TEST_CONFIG_TEXT),
),
config.Bucket(
id='master.tryserver.v8',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
project_id='v8',
revision='sha1:cfc761d7a953a72ddea8f3d4c9a28e69777ca22c',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
config_content_binary=text_to_binary(
MASTER_TRYSERVER_V8_CONFIG_TEXT),
),
]
self.assertEqual(actual, expected)
def get(self): config.cron_update_buckets()
def test_cron_update_buckets_with_existing(self, get_project_configs):
chromium_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.chromium.linux"
acls {
role: READER
group: "all"
}
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
buckets {
name: "master.tryserver.chromium.mac"
acls {
role: READER
group: "all"
}
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
''')
v8_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.v8"
acls {
role: WRITER
group: "v8-team"
}
}
''')
get_project_configs.return_value = {
'chromium': ('new!', chromium_buildbucket_cfg, None),
'v8': ('deadbeef', v8_buildbucket_cfg, None),
}
config.put_bucket('chromium', 'deadbeef',
MASTER_TRYSERVER_CHROMIUM_LINUX)
# Will not be updated.
config.put_bucket('v8', 'deadbeef', MASTER_TRYSERVER_V8)
# Will be deleted.
config.put_bucket('chromium', 'deadbeef',
MASTER_TRYSERVER_CHROMIUM_WIN)
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key)
expected = [
config.Bucket(
parent=ndb.Key(config.Project, 'chromium'),
id='master.tryserver.chromium.linux',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='new!',
config=MASTER_TRYSERVER_CHROMIUM_LINUX,
),
config.Bucket(
parent=ndb.Key(config.Project, 'chromium'),
id='master.tryserver.chromium.mac',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='new!',
config=MASTER_TRYSERVER_CHROMIUM_MAC,
),
config.Bucket(
parent=ndb.Key(config.Project, 'v8'),
id='master.tryserver.v8',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='deadbeef',
config=MASTER_TRYSERVER_V8,
),
]
self.assertEqual(actual, expected)
def test_cron_update_buckets(self, get_project_configs):
chromium_buildbucket_cfg = parse_cfg('''
acl_sets {
name: "public"
acls {
role: READER
group: "all"
}
# Test deduplication.
acls {
group: "all"
}
}
builder_mixins {
name: "recipe-x"
recipe {
cipd_package: "infra/recipe_bundle"
cipd_version: "refs/heads/master"
name: "x"
}
}
buckets {
name: "try"
acl_sets: "public"
acls {
role: SCHEDULER
identity: "*****@*****.**"
}
swarming {
hostname: "swarming.example.com"
task_template_canary_percentage { value: 10 }
builder_defaults {
mixins: "recipe-x"
}
builders {
name: "linux"
dimensions: "os:Linux"
}
}
}
buckets {
name: "master.tryserver.chromium.linux"
acl_sets: "public"
acl_sets: "undefined_acl_set_will_log_an_error_but_not_failure"
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
buckets {
name: "master.tryserver.chromium.win"
acls {
role: READER
group: "all"
}
acls {
role: SCHEDULER
group: "tryjob-access"
}
}
''')
dart_buildbucket_cfg = parse_cfg('''
buckets {
name: "try"
swarming {
builder_defaults {
dimensions: "pool:Dart.LUCI"
recipe {
cipd_package: "infra/recipe_bundle"
cipd_version: "refs/heads/master"
name: "x"
}
}
builders {
name: "linux"
}
}
}
''')
v8_buildbucket_cfg = parse_cfg('''
buckets {
name: "master.tryserver.v8"
acls {
role: WRITER
group: "v8-team"
}
}
''')
get_project_configs.return_value = {
'chromium': ('deadbeef', chromium_buildbucket_cfg, None),
'dart': ('deadbeef', dart_buildbucket_cfg, None),
'v8': (None, v8_buildbucket_cfg, None),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key)
expected = [
config.Bucket(
parent=ndb.Key(config.Project, 'chromium'),
id='master.tryserver.chromium.linux',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='deadbeef',
config=MASTER_TRYSERVER_CHROMIUM_LINUX,
),
config.Bucket(
parent=ndb.Key(config.Project, 'chromium'),
id='master.tryserver.chromium.win',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='deadbeef',
config=MASTER_TRYSERVER_CHROMIUM_WIN,
),
config.Bucket(
parent=ndb.Key(config.Project, 'chromium'),
id='try',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='deadbeef',
config=short_bucket_cfg(LUCI_CHROMIUM_TRY),
),
config.Bucket(
parent=ndb.Key(config.Project, 'dart'),
id='try',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='deadbeef',
config=short_bucket_cfg(LUCI_DART_TRY),
),
config.Bucket(
parent=ndb.Key(config.Project, 'v8'),
id='master.tryserver.v8',
entity_schema_version=config.CURRENT_BUCKET_SCHEMA_VERSION,
revision='sha1:cfc761d7a953a72ddea8f3d4c9a28e69777ca22c',
config=MASTER_TRYSERVER_V8),
]
self.assertEqual(actual, expected)
def test_cron_update_buckets_with_existing(self):
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
).put()
# Will not be updated.
config.Bucket(
id='master.tryserver.v8',
project_id='v8',
revision='deadbeef',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
).put()
# Will be deleted.
config.Bucket(
id='master.tryserver.chromium.win',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT,
).put()
chromium_buildbucket_cfg = project_config_pb2.BuildbucketCfg(buckets=[
project_config_pb2.Bucket(
name='master.tryserver.chromium.linux',
acls=[
project_config_pb2.Acl(role=project_config_pb2.Acl.READER,
group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER,
group='tryjob-access'),
],
),
# Will be added.
project_config_pb2.Bucket(
name='master.tryserver.chromium.mac',
acls=[
project_config_pb2.Acl(role=project_config_pb2.Acl.READER,
group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER,
group='tryjob-access'),
],
),
])
v8_buildbucket_cfg = project_config_pb2.BuildbucketCfg(
buckets=[
# Reservation will fail.
project_config_pb2.Bucket(
name='master.tryserver.chromium.linux',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.WRITER,
group='v8-team')
],
),
# Will not be updated.
project_config_pb2.Bucket(
name='master.tryserver.v8',
acls=[
project_config_pb2.Acl(
role=project_config_pb2.Acl.WRITER,
group='v8-team')
],
),
], )
self.mock(config_component, 'get_project_configs', mock.Mock())
config_component.get_project_configs.return_value = {
'chromium': ('new!', chromium_buildbucket_cfg),
'v8': ('deadbeef', v8_buildbucket_cfg),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key.id())
expected = [
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='new!',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.chromium.mac',
project_id='chromium',
revision='new!',
config_content=MASTER_TRYSERVER_CHROMIUM_MAC_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.v8',
project_id='v8',
revision='deadbeef',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
),
]
self.assertEqual(actual, expected)
def test_cron_update_buckets(self):
chromium_buildbucket_cfg = project_config_pb2.BuildbucketCfg(buckets=[
project_config_pb2.Bucket(
name='master.tryserver.chromium.linux',
acls=[
project_config_pb2.Acl(role=project_config_pb2.Acl.READER,
group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER,
group='tryjob-access'),
],
),
project_config_pb2.Bucket(
name='master.tryserver.chromium.win',
acls=[
project_config_pb2.Acl(role=project_config_pb2.Acl.READER,
group='all'),
project_config_pb2.Acl(
role=project_config_pb2.Acl.SCHEDULER,
group='tryjob-access'),
],
),
])
v8_buildbucket_cfg = project_config_pb2.BuildbucketCfg(buckets=[
project_config_pb2.Bucket(
name='master.tryserver.v8',
acls=[
project_config_pb2.Acl(role=project_config_pb2.Acl.WRITER,
group='v8-team')
],
),
])
test_buildbucket_cfg = project_config_pb2.BuildbucketCfg(buckets=[
project_config_pb2.Bucket(
name='master.tryserver.test',
acls=[
project_config_pb2.Acl(role=project_config_pb2.Acl.WRITER,
identity='*****@*****.**')
],
),
])
self.mock(config_component, 'get_project_configs', mock.Mock())
config_component.get_project_configs.return_value = {
'chromium': ('deadbeef', chromium_buildbucket_cfg),
'v8': (None, v8_buildbucket_cfg),
'test': ('babe', test_buildbucket_cfg),
}
config.cron_update_buckets()
actual = config.Bucket.query().fetch()
actual = sorted(actual, key=lambda b: b.key)
expected = [
config.Bucket(
id='master.tryserver.chromium.linux',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_LINUX_CONFIG_TEXT,
),
config.Bucket(
id='master.tryserver.chromium.win',
project_id='chromium',
revision='deadbeef',
config_content=MASTER_TRYSERVER_CHROMIUM_WIN_CONFIG_TEXT,
),
config.Bucket(id='master.tryserver.test',
project_id='test',
revision='babe',
config_content=MASTER_TRYSERVER_TEST_CONFIG_TEXT),
config.Bucket(
id='master.tryserver.v8',
project_id='v8',
revision='sha1:cfc761d7a953a72ddea8f3d4c9a28e69777ca22c',
config_content=MASTER_TRYSERVER_V8_CONFIG_TEXT,
),
]
self.assertEqual(actual, expected)
