• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1/*
2 * Copyright (C) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16import featureAbility from '@ohos.ability.featureAbility'
17import { UiDriver, BY, PointerMatrix } from '@ohos.uitest'
18const CODECMIMEVALUE = ['video/avc', 'audio/mp4a-latm', 'audio/mpeg']
19
20export async function getPermission(permissionNames) {
21    featureAbility.getContext().requestPermissionsFromUser(permissionNames, 0, async (data) => {
22        console.info("case request success" + JSON.stringify(data));
23    })
24}
25
26export async function driveFn(num) {
27    console.info(`case come in driveFn`);
28    let driver = UiDriver.create();
29    console.info(`driver is ${JSON.stringify(driver)}`);
30    await msleepAsync(2000);
31    console.info(`UiDriver start`);
32    for (let i = 0; i < num; i++) {
33        let button = await driver.findComponent(BY.text('允许'));
34        console.info(`button is ${JSON.stringify(button)}`);
35        await msleepAsync(2000);
36        await button.click();
37    }
38    await msleepAsync(2000);
39}
40
41// File operation
42export async function getFileDescriptor(fileName) {
43    let fileDescriptor = undefined;
44    let mgr = globalThis.abilityContext.resourceManager
45    await mgr.getRawFileDescriptor(fileName).then(value => {
46        fileDescriptor = {fd: value.fd, offset: value.offset, length: value.length};
47        console.log('case getRawFileDescriptor success fileName: ' + fileName);
48    }).catch(error => {
49        console.log('case getRawFileDescriptor err: ' + error);
50    });
51    return fileDescriptor;
52}
53export async function closeFileDescriptor(fileName) {
54    await globalThis.abilityContext.resourceManager.closeRawFileDescriptor(fileName).then(()=> {
55        console.log('case closeRawFileDescriptor ' + fileName);
56    }).catch(error => {
57        console.log('case closeRawFileDescriptor err: ' + error);
58    });
59}
60
61// wait asynchronously
62export async function msleepAsync(ms) {
63    return new Promise((resolve) => setTimeout(resolve, ms));
64}
65
66export function assertErr(opera, err, done) {
67    console.info(`case ${opera} error,errMessage is ${err.message}`);
68    expect().assertFail();
69    done();
70}
71
72// callback function for promise call back error
73export function failureCallback(error) {
74    expect().assertFail();
75    console.info(`case failureCallback promise called,errMessage is ${error.message}`);
76}
77
78// callback function for promise catch error
79export function catchCallback(error) {
80    expect().assertFail();
81    console.info(`case error catch called,errMessage is ${error.message}`);
82}
83
84export function checkDescription(actualDescription, descriptionKey, descriptionValue) {
85    for (let i = 0; i < descriptionKey.length; i++) {
86        let property = actualDescription[descriptionKey[i]];
87        console.info('case key is  '+ descriptionKey[i]);
88        console.info('case actual value is  '+ property);
89        console.info('case hope value is  '+ descriptionValue[i]);
90        if (descriptionKey[i] == 'codec_mime') {
91            expect(property).assertEqual(CODECMIMEVALUE[descriptionValue[i]]);
92        } else {
93            expect(property).assertEqual(descriptionValue[i]);
94        }
95
96    }
97}
98
99export async function getFd(pathName, fileType) {
100    let fdObject = {
101        fileAsset : null,
102        fdNumber : null
103    }
104    let displayName = pathName;
105    console.info('[mediaLibrary] fileType is ' + fileType);
106    const mediaTest = mediaLibrary.getMediaLibrary(globalThis.abilityContext);
107    let fileKeyObj = mediaLibrary.FileKey;
108    let mediaType;
109    let publicPath;
110    if (fileType == 'audio') {
111        mediaType = mediaLibrary.MediaType.AUDIO;
112        publicPath = await mediaTest.getPublicDirectory(mediaLibrary.DirectoryType.DIR_AUDIO);
113    } else if (fileType == "video") {
114        mediaType = mediaLibrary.MediaType.VIDEO;
115        publicPath = await mediaTest.getPublicDirectory(mediaLibrary.DirectoryType.DIR_VIDEO);
116    } else {
117        mediaType = mediaLibrary.MediaType.VIDEO;
118        publicPath = await mediaTest.getPublicDirectory(mediaLibrary.DirectoryType.DIR_VIDEO);
119    }
120    console.info('[mediaLibrary] publicPath is ' + publicPath);
121
122    let dataUri = await mediaTest.createAsset(mediaType, displayName, publicPath);
123    if (dataUri != undefined) {
124        let args = dataUri.id.toString();
125        let fetchOp = {
126            selections : fileKeyObj.ID + "=?",
127            selectionArgs : [args],
128        }
129        let fetchFileResult = await mediaTest.getFileAssets(fetchOp);
130        fdObject.fileAsset = await fetchFileResult.getAllObject();
131        fdObject.fdNumber = await fdObject.fileAsset[0].open('rw');
132        console.info('case getFd number is: ' + fdObject.fdNumber);
133    }
134    return fdObject;
135}
136
137export async function closeFd(fileAsset, fdNumber) {
138    if (fileAsset != null) {
139        await fileAsset[0].close(fdNumber).then(() => {
140            console.info('[mediaLibrary] case close fd success');
141        }).catch((err) => {
142            console.info('[mediaLibrary] case close fd failed');
143        });
144    } else {
145        console.info('[mediaLibrary] case fileAsset is null');
146    }
147}
148