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 { xpowerWifiDataSender } from '../../../../../src/trace/database/data-trafic/xpower/XpowerWifiDataSender'; 17import { threadPool } from '../../../../../src/trace/database/SqlLite'; 18import { TraceRow } from '../../../../../src/trace/component/trace/base/TraceRow'; 19import { XpowerWifiStruct } from '../../../../../src/trace/database/ui-worker/ProcedureWorkerXpowerWifi'; 20jest.mock('../../../../../src/js-heap/model/DatabaseStruct', () => {}); 21jest.mock('../../../../../src/trace/database/ui-worker/ProcedureWorkerSnapshot', () => { 22 return {}; 23}); 24jest.mock('../../../../../src/trace/database/ui-worker/ProcedureWorker', () => { 25 return {}; 26}); 27 28describe('xpowerWifiDataSender Test', () => { 29 let data = { 30 startTime: 5000, 31 rx: 500, 32 tx: 500, 33 }; 34 it('xpowerWifiDataSenderTest01', () => { 35 threadPool.submitProto = jest.fn((query: number, params: any, callback: Function) => { 36 callback(data, 1, true); 37 }); 38 let traceRow = TraceRow.skeleton<XpowerWifiStruct>(); 39 xpowerWifiDataSender(traceRow, 'WIFIPackets').then((result) => { 40 expect(result).toBeTruthy(); 41 }); 42 }); 43 it('xpowerWifiDataSenderTest02', () => { 44 threadPool.submitProto = jest.fn((query: number, params: any, callback: Function) => { 45 callback(data, 1, true); 46 }); 47 let traceRow = TraceRow.skeleton<XpowerWifiStruct>(); 48 xpowerWifiDataSender(traceRow, 'WIFIBytes').then((result) => { 49 expect(result).toBeTruthy(); 50 }); 51 }); 52}); 53