# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
from contrib.shared_storage.page_set import SharedStorageStory
class SharedStorageDocumentSetStory(SharedStorageStory):
NAME = "SharedStorageDocumentSet"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
window.sharedStorage.set('a{{ index }}', 'b{{ index }}');
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}]
class SharedStorageDocumentAppendStory(SharedStorageStory):
NAME = "SharedStorageDocumentAppend"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
window.sharedStorage.append('a{{ index }}', 'b{{ index }}');
window.sharedStorage.append('a{{ index }}', 'c{{ index }}');
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentAppend',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}, {
'type': 'documentAppend',
'params': {
'key': 'a{{ index }}',
'value': 'c{{ index }}'
}
}]
class SharedStorageDocumentDeleteStory(SharedStorageStory):
NAME = "SharedStorageDocumentDelete"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
window.sharedStorage.set('a{{ index }}', 'b{{ index }}');
window.sharedStorage.delete('a{{ index }}');
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}, {
'type': 'documentDelete',
'params': {
'key': 'a{{ index }}'
}
}]
class SharedStorageDocumentClearStory(SharedStorageStory):
NAME = "SharedStorageDocumentClear"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
window.sharedStorage.set('a{{ index }}', 'b{{ index }}')
window.sharedStorage.clear()
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}, {
'type': 'documentClear'
}]
RENAVIGATE_AFTER_ACTION = True
class SharedStorageWorkletRunSetStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunSet"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-set-operation',
{data: {'key': 'a{{ index }}',
'value': 'b{{ index }}'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'workletSet',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}]
class SharedStorageWorkletSelectURLSetStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLSet"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-set-operation',
[{url: 'with_worklet.html'}],
{data: {'key': 'a{{ index }}',
'value': 'b{{ index }}'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'workletSet',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}]
class SharedStorageWorkletRunAppendStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunAppend"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-append-operation',
{data: {'key': 'a{{ index }}',
'value': 'b{{ index }}'},
keepAlive: true});
sharedStorage.run('run-append-operation',
{data: {'key': 'a{{ index }}',
'value': 'c{{ index }}'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'documentRun'
}, {
'type': 'workletAppend',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}, {
'type': 'workletAppend',
'params': {
'key': 'a{{ index }}',
'value': 'c{{ index }}'
}
}]
class SharedStorageWorkletSelectURLAppendStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLAppend"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-append-operation',
[{url: 'with_worklet.html'}],
{data: {'key': 'a{{ index }}',
'value': 'b{{ index }}'},
keepAlive: true});
sharedStorage.selectURL('selecturl-append-operation',
[{url: 'with_worklet.html'}],
{data: {'key': 'a{{ index }}',
'value': 'c{{ index }}'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'documentSelectURL'
}, {
'type': 'workletAppend',
'params': {
'key': 'a{{ index }}',
'value': 'b{{ index }}'
}
}, {
'type': 'workletAppend',
'params': {
'key': 'a{{ index }}',
'value': 'c{{ index }}'
}
}]
class SharedStorageWorkletRunDeleteStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunDelete"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.set('a{{ index }}', 'any');
sharedStorage.run('run-delete-operation',
{data: {'key': 'a{{ index }}'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'any'
}
}, {
'type': 'documentRun'
}, {
'type': 'workletDelete',
'params': {
'key': 'a{{ index }}'
}
}]
class SharedStorageWorkletSelectURLDeleteStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLDelete"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.set('a{{ index }}', 'any');
sharedStorage.selectURL('selecturl-delete-operation',
[{url: 'with_worklet.html'}],
{data: {'key': 'a{{ index }}'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'any'
}
}, {
'type': 'documentSelectURL'
}, {
'type': 'workletDelete',
'params': {
'key': 'a{{ index }}'
}
}]
class SharedStorageWorkletRunClearStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunClear"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.set('a{{ index }}', 'any');
sharedStorage.run('run-clear-operation',
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'any'
}
}, {
'type': 'documentRun'
}, {
'type': 'workletClear'
}]
RENAVIGATE_AFTER_ACTION = True
class SharedStorageWorkletSelectURLClearStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLClear"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.set('a{{ index }}', 'any');
sharedStorage.selectURL('selecturl-clear-operation',
[{url: 'with_worklet.html'}],
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSet',
'params': {
'key': 'a{{ index }}',
'value': 'any'
}
}, {
'type': 'documentSelectURL'
}, {
'type': 'workletClear'
}]
RENAVIGATE_AFTER_ACTION = True
class SharedStorageWorkletRunGetStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunGet"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0');
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-get-operation',
{data: {'key': 'k0'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'workletGet',
'params': {
'key': 'k0'
}
}]
class SharedStorageWorkletSelectURLGetStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLGet"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0')
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-get-operation',
[{url: 'with_worklet.html'}],
{data: {'key': 'k0'},
keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'workletGet',
'params': {
'key': 'k0'
}
}]
class SharedStorageWorkletRunLengthStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunLength"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0')
sharedStorage.set('k1', 'v1')
sharedStorage.set('k2', 'v2')
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k1',
'value': 'v1'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k2',
'value': 'v2'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-length-operation',
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'workletLength'
}]
class SharedStorageWorkletSelectURLLengthStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLLength"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0');
sharedStorage.set('k1', 'v1');
sharedStorage.set('k2', 'v2');
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k1',
'value': 'v1'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k2',
'value': 'v2'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-length-operation',
[{url: 'with_worklet.html'}],
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'workletLength'
}]
class SharedStorageWorkletRunKeysStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunKeys"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0')
sharedStorage.set('k1', 'v1')
sharedStorage.set('k2', 'v2')
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k1',
'value': 'v1'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k2',
'value': 'v2'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-keys-operation',
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'workletKeys'
}]
# Expect to iterate over `3 + self.SIZE` keys, each with a call to
# `GetNextIterationResult()`, then make one last call to
# `GetNextIterationResult()` to terminate the iteration with
# `MakeEndOfIteration()`.
EXPECTED_ITERATOR_HISTOGRAM_COUNT = "4 + self.SIZE"
class SharedStorageWorkletSelectURLKeysStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLKeys"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0');
sharedStorage.set('k1', 'v1');
sharedStorage.set('k2', 'v2');
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k1',
'value': 'v1'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k2',
'value': 'v2'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-keys-operation',
[{url: 'with_worklet.html'}],
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'workletKeys'
}]
# Expect to iterate over `3 + self.SIZE` keys, each with a call to
# `GetNextIterationResult()`, then make one last call to
# `GetNextIterationResult()` to terminate the iteration with
# `MakeEndOfIteration()`.
EXPECTED_ITERATOR_HISTOGRAM_COUNT = "4 + self.SIZE"
class SharedStorageWorkletRunEntriesStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunEntries"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0')
sharedStorage.set('k1', 'v1')
sharedStorage.set('k2', 'v2')
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k1',
'value': 'v1'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k2',
'value': 'v2'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-entries-operation',
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'workletEntries'
}]
# Expect to iterate over `3 + self.SIZE` entries, each with a call to
# `GetNextIterationResult()`, then make one last call to
# `GetNextIterationResult()` to terminate the iteration with
# `MakeEndOfIteration()`.
EXPECTED_ITERATOR_HISTOGRAM_COUNT = "4 + self.SIZE"
class SharedStorageWorkletSelectURLEntriesStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLEntries"
ABSTRACT_STORY = False
SETUP_SCRIPT = """
sharedStorage.set('k0', 'v0');
sharedStorage.set('k1', 'v1');
sharedStorage.set('k2', 'v2');
"""
EXPECTED_SETUP_EVENTS = [{
'type': 'documentSet',
'params': {
'key': 'k0',
'value': 'v0'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k1',
'value': 'v1'
}
}, {
'type': 'documentSet',
'params': {
'key': 'k2',
'value': 'v2'
}
}]
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-entries-operation',
[{url: 'with_worklet.html'}],
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'workletEntries'
}]
# Expect to iterate over `3 + self.SIZE` entries, each with a call to
# `GetNextIterationResult()`, then make one last call to
# `GetNextIterationResult()` to terminate the iteration with
# `MakeEndOfIteration()`.
EXPECTED_ITERATOR_HISTOGRAM_COUNT = "4 + self.SIZE"
class SharedStorageWorkletRunRemainingBudgetStory(SharedStorageStory):
NAME = "SharedStorageWorkletRunRemainingBudget"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.run('run-remainingbudget-operation',
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentRun'
}, {
'type': 'workletRemainingBudget'
}]
class SharedStorageWorkletSelectURLRemainingBudgetStory(SharedStorageStory):
NAME = "SharedStorageWorkletSelectURLRemainingBudget"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """
sharedStorage.selectURL('selecturl-remainingbudget-operation',
[{url: 'with_worklet.html'}],
{keepAlive: true});
"""
EXPECTED_ACTION_EVENTS_TEMPLATE = [{
'type': 'documentSelectURL'
}, {
'type': 'workletRemainingBudget'
}]
class SharedStorageDocumentAddModuleStory(SharedStorageStory):
NAME = "SharedStorageDocumentAddModule"
ABSTRACT_STORY = False
ACTION_SCRIPT_TEMPLATE = """"""
EXPECTED_ACTION_EVENTS_TEMPLATE = []
RENAVIGATE_AFTER_ACTION = True
Codebase Browser
chromium