// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
/**
* Gets the metadata of a volume having a specified type.
* @param {string} volumeType volume type for entry.
* @return {!Promise<chrome.fileManagerPrivate.VolumeMetadata>} Volume metadata.
*/
async function getVolumeMetadataByType(volumeType) {
return new Promise(
(resolve,
reject) => {chrome.fileManagerPrivate.getVolumeMetadataList(list => {
if (chrome.runtime.lastError) {
reject(chrome.runtime.lastError.message);
return;
}
resolve(list.find(v => v.volumeType === volumeType));
})});
}
/**
* Gets the file system of specific volume type.
* @param {string} volumeType volume type.
* @return {!Promise<chrome.fileManagerPrivate.FileSystem>} Volume metadata.
*/
async function getFileSystem(volumeType) {
const volume = await getVolumeMetadataByType(volumeType);
return new Promise((resolve, reject) => {
chrome.fileSystem.requestFileSystem(
{volumeId: volume.volumeId, writable: true}, fs => {
if (chrome.runtime.lastError) {
reject(chrome.runtime.lastError.message);
return;
}
resolve(fs);
});
});
}
/**
* Gets an external file entry from a specified path.
* @param {string} volumeType volume type for entry.
* @param {string} path path of entry.
* @return {!Promise<Entry>} specified entry.
*/
async function getFileEntry(volumeType, path) {
const fs = await getFileSystem(volumeType);
return new Promise(resolve => {
fs.root.getFile(path, {}, entry => {
chrome.fileManagerPrivate.resolveIsolatedEntries(
[entry], externalEntries => {
resolve(externalEntries[0]);
});
});
});
}
/**
* Wrapper around getFileEntry() that resolves multiple paths.
* @param {string} volumeType
* @param {Array<string>} paths
* @return {!Promise<Array<Entry>>}
*/
async function getFileEntries(volumeType, paths) {
return Promise.all(paths.map(path => getFileEntry(volumeType, path)));
}
chrome.test.getConfig(config => {
const mode = config.customArg;
if (!mode) {
chrome.test.fail('No mode provided.');
return;
}
switch (mode) {
case 'disabled':
chrome.test.runTests([
async function getDlpMetadata_Disabled() {
const [file] = await getFileEntries('testing', ['blocked_file.txt']);
// Since the DLP feature is disabled, an empty list should be
// returned.
await new Promise((resolve, reject) => file.remove(resolve, reject));
chrome.fileManagerPrivate.getDlpMetadata(
[file], chrome.test.callbackPass(dlpMetadata => {
chrome.test.assertEq([], dlpMetadata);
}))
},
async function getDlpRestrictionDetails_Disabled() {
chrome.fileManagerPrivate.getDlpRestrictionDetails(
'https://example1.com',
chrome.test.callbackPass(dlpRestrictionDetails => {
chrome.test.assertEq([], dlpRestrictionDetails);
}));
},
async function getDlpBlockedComponents_Disabled() {
chrome.fileManagerPrivate.getDlpBlockedComponents(
'https://example1.com',
chrome.test.callbackPass(blockedComponents => {
chrome.test.assertEq([], blockedComponents);
}));
}
]);
break;
case 'error':
chrome.test.runTests([
async function getDlpMetadata_FileDeleted() {
// Get the file.
const [file] = await getFileEntries('testing', ['blocked_file.txt']);
// Delete the file. Even though 'blocked_file.txt' is restricted by
// DLP, once it doesn't exist anymore an empty DlpMetadata object
// should be returned.
await new Promise((resolve, reject) => file.remove(resolve, reject));
chrome.fileManagerPrivate.getDlpMetadata(
[file], chrome.test.callbackPass(dlpMetadata => {
chrome.test.assertEq(
[{
isDlpRestricted: false,
isRestrictedForDestination: false,
sourceUrl: ''
}],
dlpMetadata);
}))
},
async function getDlpMetadata_EmptyList() {
chrome.fileManagerPrivate.getDlpMetadata(
[], chrome.test.callbackPass(dlpMetadata => {
chrome.test.assertEq(0, dlpMetadata.length);
}))
}
]);
break;
case 'restriction_details':
chrome.test.runTests([async function getDlpRestrictionDetails() {
chrome.fileManagerPrivate.getDlpRestrictionDetails(
'https://example1.com',
chrome.test.callbackPass(dlpRestrictionDetails => {
chrome.test.assertEq(
[
{
'components': [
'android_files', 'crostini', 'guest_os', 'removable'
],
'level': 'block',
'urls': ['https://external.com']
},
{
'components': ['drive'],
'level': 'allow',
'urls': ['https://internal.com']
}
],
dlpRestrictionDetails);
}));
}]);
break;
case 'blocked_components':
chrome.test.runTests([async function getDlpBlockedComponents() {
chrome.fileManagerPrivate.getDlpBlockedComponents(
'https://example1.com',
chrome.test.callbackPass(blockedComponents => {
chrome.test.assertEq(
['android_files', 'crostini', 'guest_os', 'removable'],
blockedComponents);
}));
}]);
break;
case 'dismissIOTask':
chrome.test.runTests([
async function dismissIOTask() {
// Valid task id - succeeds and notifies FPNM.
chrome.fileManagerPrivate.dismissIOTask(
1,
chrome.test.callbackPass());
// Invalid task id - succeeds but won't notify FPNM.
chrome.fileManagerPrivate.dismissIOTask(
-5,
chrome.test.callbackFail('Invalid task id'));
chrome.test.succeed();
}
]);
break;
case 'progressPausedTasks':
chrome.test.runTests([
async function progressPausedTasks() {
// Succeeds and notifies FPNM.
chrome.fileManagerPrivate.progressPausedTasks(
chrome.test.callbackPass());
chrome.test.succeed();
}
]);
break;
case 'showPolicyDialog':
chrome.test.runTests([
async function showPolicyDialog() {
// Invalid task id throws an error.
chrome.fileManagerPrivate.showPolicyDialog(
-5,
chrome.fileManagerPrivate.PolicyDialogType.WARNING,
chrome.test.callbackFail('Invalid task id'));
// Valid calls: succeed and notify FPNM.
chrome.fileManagerPrivate.showPolicyDialog(
1,
chrome.fileManagerPrivate.PolicyDialogType.WARNING,
chrome.test.callbackPass());
chrome.fileManagerPrivate.showPolicyDialog(
2,
chrome.fileManagerPrivate.PolicyDialogType.ERROR,
chrome.test.callbackPass());
chrome.test.succeed();
}
]);
break;
case 'getDialogCaller':
chrome.test.runTests([
async function getDialogCaller() {
chrome.fileManagerPrivate.getDialogCaller(
chrome.test.callbackPass(
caller => {
chrome.test.assertEq({url: 'https://example.com/'},
caller)})
);
}
]);
break;
case 'default':
chrome.test.runTests([
async function getDlpMetadata() {
const testEntries = await getFileEntries('testing', [
'blocked_file.txt', 'unrestricted_file.txt', 'untracked_file.txt'
]);
chrome.test.assertEq(3, testEntries.length);
chrome.fileManagerPrivate.getDlpMetadata(
testEntries, chrome.test.callbackPass(dlpMetadata => {
chrome.test.assertEq(
[
{
isDlpRestricted: true,
isRestrictedForDestination: false,
sourceUrl: 'https://example1.com'
},
{
isDlpRestricted: false,
isRestrictedForDestination: false,
sourceUrl: 'https://example2.com'
},
{
isDlpRestricted: false,
isRestrictedForDestination: false,
sourceUrl: ''
}
],
dlpMetadata);
}))
}
]);
break;
default:
chrome.test.notifyFail(`Unrecognized mode ${mode} found.`);
}
});
Codebase Browser
chromium