#!/usr/bin/env python3
# Copyright (C) 2024 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from python.generators.diff_tests.testing import Csv, DataPath, Path, DiffTestBlueprint, TestSuite
class ExportTests(TestSuite):
def test_to_firefox_profile(self):
return DiffTestBlueprint(
trace=DataPath('zip/perf_track_sym.zip'),
query="""
INCLUDE PERFETTO MODULE export.to_firefox_profile;
SELECT export_to_firefox_profile();
""",
out=Path('firefox_profile.out'))
def test_svg_timeline_simple(self):
return DiffTestBlueprint(
trace=DataPath('example_android_trace_30s.pb'),
query="""
INCLUDE PERFETTO MODULE export.to_svg;
INCLUDE PERFETTO MODULE slices.with_context;
SELECT *
FROM _svg_timeline!(
(SELECT utid, ts, dur, depth, name, 'slice_link' AS href, 1 AS track_group_order FROM thread_slice
WHERE ts >= 89092060412 and ts < 89095305364 AND tid = 5511),
(SELECT ts.utid, ts.ts, ts.dur, ts.state, ts.io_wait, ts.blocked_function, 'thread_state_link' AS href, 0 AS track_group_order FROM thread_state ts
JOIN thread USING(utid)
WHERE ts >= 89092060412 and ts < 89095305364 AND tid = 5511),
utid, -- svg_group_key
utid, -- track_group_key
track_group_order, -- track_order
2000,
500);
""",
out=Csv("""
"svg_group_key","svg"
954,""
"""))
def test_svg_timeline_multi_group(self):
return DiffTestBlueprint(
trace=DataPath('example_android_trace_30s.pb'),
query="""
INCLUDE PERFETTO MODULE export.to_svg;
INCLUDE PERFETTO MODULE slices.with_context;
WITH
slice_data AS (
SELECT
utid,
1 AS tord,
utid, ts, dur, depth, name, 'slice_link' AS href
FROM thread_slice
WHERE ts >= 86625813931 and ts < 86731829201 AND tid IN (5511, 5517)
),
thread_state_data AS (
SELECT
utid,
utid,
0 AS tord,
utid, ts, dur, state, io_wait, blocked_function, 'thread_state_link' AS href
FROM thread_state
JOIN thread USING(utid)
WHERE ts >= 86625813931 and ts < 86731829201 AND tid IN (5511, 5517)
)
SELECT *
FROM _svg_timeline!(
slice_data,
thread_state_data,
utid,
utid,
tord,
2000,
50);
""",
out=Csv("""
"svg_group_key","svg"
954,""
957,""
"""))
def test_svg_combine_tracks(self):
return DiffTestBlueprint(
trace=DataPath('example_android_trace_30s.pb'),
query="""
INCLUDE PERFETTO MODULE export.to_svg;
INCLUDE PERFETTO MODULE slices.with_context;
-- Test the full modular pipeline
WITH
slice_data AS (
SELECT
upid,
utid,
2 AS tord,
utid, ts, dur, depth, name, 'slice_link' AS href
FROM thread_slice
WHERE ts >= 86625813931 and ts < 86731829201 AND tid IN (5511, 5517)
),
thread_state_data AS (
SELECT
upid,
utid,
1 AS tord,
utid, ts, dur, state, io_wait, blocked_function, 'thread_state_link' AS href
FROM thread_state
JOIN thread USING(utid)
WHERE ts >= 86625813931 and ts < 86731829201 AND tid IN (5511, 5517)
),
slice_positions AS (
SELECT * FROM _slice_intervals_to_positions!(
slice_data, upid, utid, tord, 2000, 20
)
),
thread_state_positions AS (
SELECT * FROM _thread_state_intervals_to_positions!(
thread_state_data, upid, utid, tord, 2000, 20
)
),
slice_tracks AS (
SELECT * FROM _generate_tracks_by_group!(
slice_positions, upid, utid, tord, 2, 2, 0
)
),
thread_state_tracks AS (
SELECT * FROM _generate_tracks_by_group_with_label!(
thread_state_positions, upid, utid, tord,
1, 2, 0, utid, 10
)
),
all_tracks AS (
SELECT * FROM slice_tracks
UNION ALL
SELECT * FROM thread_state_tracks
)
SELECT * FROM _combine_track_svgs!(
all_tracks, upid, 2000, 50
);
""",
out=Csv("""
"svg_group_key","svg"
31,""
"""))
def test_svg_timeline_with_counters(self):
return DiffTestBlueprint(
trace=DataPath('example_android_trace_30s.pb'),
query="""
INCLUDE PERFETTO MODULE export.to_svg;
INCLUDE PERFETTO MODULE slices.with_context;
-- Test the svg_timeline_with_counters convenience macro
WITH
slice_data AS (
SELECT
upid,
1 AS tord,
utid, ts, dur, depth, name, 'slice_link' AS href
FROM thread_slice
WHERE ts >= 86625813931 and ts < 86731829201 AND tid IN (5511)
),
thread_state_data AS (
SELECT
upid,
2 AS tord,
utid, ts, dur, state, io_wait, blocked_function, 'thread_state_link' AS href
FROM thread_state
JOIN thread USING(utid)
WHERE ts >= 86625813931 and ts < 86731829201 AND tid IN (5511)
),
counter_data AS (
SELECT
upid,
3 AS tord,
ts,
lead(ts, 1, ts + 1000000) OVER (PARTITION BY track_id ORDER BY ts) - ts AS dur,
process_counter_track.name,
value,
'counter_link' AS href
FROM counter
JOIN process_counter_track ON counter.track_id = process_counter_track.id
JOIN process USING(upid)
WHERE ts >= 86625813931 and ts < 86731829201 AND pid = 5506
AND process_counter_track.name LIKE '%heap size%'
)
SELECT *
FROM _svg_timeline_with_counters!(
slice_data,
thread_state_data,
counter_data,
upid,
upid,
tord,
2000,
50,
1);
""",
out=Csv("""
"svg_group_key","svg"
31,""
"""))