#!/usr/bin/env python3
# Copyright (C) 2023 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 Path, DataPath, Metric
from python.generators.diff_tests.testing import Csv, Json, TextProto
from python.generators.diff_tests.testing import DiffTestBlueprint
from python.generators.diff_tests.testing import TestSuite


class AndroidBugreport(TestSuite):

  def test_android_bugreport_battery_stats(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        WITH first_100_events AS (
          SELECT ts, slice.name, dur
          FROM slice
          JOIN track on slice.track_id = track.id
          WHERE track.name LIKE 'battery_stats.%'
          ORDER BY 1, 2, 3 ASC
          LIMIT 100
        ),
        first_100_states AS (
          SELECT ts, name, value
          FROM counter
          JOIN counter_track on counter.track_id = counter_track.id
          WHERE name LIKE 'battery_stats.%'
          ORDER BY 1, 2, 3 ASC
          LIMIT 100
        )
        SELECT * FROM first_100_events
        UNION ALL
        SELECT * FROM first_100_states;
        """,
        out=Path('android_bugreport_battery_stats_test.out'))

  def test_android_bugreport_battery_stats_counts(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        WITH event_counts(type, count) AS ( VALUES
          ('battery_stats_history_events', (
              SELECT
                  COUNT(1)
              FROM slice
              JOIN track
                  ON slice.track_id = track.id
              WHERE
                  track.name LIKE 'battery_stats.%'
            )
          ),
          ('battery_stats_history_states', (
              SELECT
                  COUNT(1)
              FROM counter
              JOIN counter_track
                  ON counter.track_id = counter_track.id
              WHERE
                  name LIKE 'battery_stats.%'
            )
          )
        )
        SELECT * FROM event_counts
        """,
        out=Csv("""
        "type","count"
        "battery_stats_history_events",4237
        "battery_stats_history_states",12245
        """))

  def test_android_bugreport_logs(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        WITH
        initial AS (SELECT
            (SELECT count(*) FROM android_logs) AS cnt,
            ts, prio, tag, msg FROM android_logs
          ORDER BY ts ASC
          LIMIT 100
        ),
        latest AS (SELECT
            (SELECT count(*) FROM android_logs) AS cnt,
            ts, prio, tag, msg FROM android_logs
          ORDER BY ts DESC
          LIMIT 100
        )
        SELECT * FROM initial UNION ALL SELECT * FROM latest;
        """,
        out=Path('android_bugreport_logs_test.out'))

  def test_android_bugreport_dumpstate(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        SELECT section, service, count(line) AS linecount FROM android_dumpstate
        GROUP BY section, service;
        """,
        out=Path('android_bugreport_dumpstate_test.out'))

  def test_android_bugreport_dumpsys(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        SELECT section, service, line FROM android_dumpstate
        WHERE service = 'color_display';
        """,
        out=Path('android_bugreport_dumpsys_test.out'))

  def test_android_bugreport_parse_order(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        SELECT id, parent_id, name, size, trace_type, processing_order
        FROM __intrinsic_trace_file
        WHERE trace_type <> "unknown"
        ORDER BY processing_order
        """,
        out=Csv("""
        "id","parent_id","name","size","trace_type","processing_order"
        0,"[NULL]","[NULL]",6220586,"zip",0
        1,0,"bugreport-crosshatch-SPB5.210812.002-2021-08-24-23-35-40.txt",43132864,"android_dumpstate",1
        16,0,"FS/data/misc/logd/logcat.01",2169697,"android_logcat",2
        15,0,"FS/data/misc/logd/logcat",2152073,"android_logcat",3
        """))

  def test_android_bugreport_trace_types(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        SELECT trace_type, count(*) AS cnt, sum(size) AS total_size
        FROM __intrinsic_trace_file
        GROUP BY trace_type
        ORDER BY trace_type
        """,
        out=Csv("""
        "trace_type","cnt","total_size"
        "android_dumpstate",1,43132864
        "android_logcat",2,4321770
        "unknown",2452,626115
        "zip",1,6220586
        """))

  def test_android_bugreport_trace_types(self):
    return DiffTestBlueprint(
        trace=DataPath('bugreport-crosshatch-SPB5.zip'),
        query="""
        INCLUDE PERFETTO MODULE android.dumpsys.show_map;
        SELECT * FROM android_dumpsys_show_map
        ORDER BY rss_kb DESC
        LIMIT 10;
        """,
        out=Csv("""
                "process_name","pid","vss_kb","rss_kb","pss_kb","shared_clean_kb","shared_dirty_kb","private_clean_kb","private_dirty_kb","swap_kb","swap_pss_kb","anon_huge_pages_kb","shmem_pmd_mapped_kb","file_pmd_mapped_kb","shared_huge_tlb_kb","private_hugetlb_kb","locked_kb","mapping_count","mapped_object"
                "(system_server)",1835,1048576,28068,28068,0,0,0,28068,0,0,0,0,0,0,0,1,0,"space"
    "(com.android.systemui)",2281,524288,8836,8836,0,0,0,8836,0,0,0,0,0,0,0,1,0,"space"
    "(system_server)",1835,7180,7180,1469,0,5832,0,1348,0,0,0,0,0,0,0,1,0,"space]"
    "(com.android.bluetooth)",2261,7180,7180,677,0,6644,0,536,0,0,0,0,0,0,0,1,0,"space]"
    "(com.android.systemui)",2281,7180,7180,1133,0,6176,0,1004,0,0,0,0,0,0,0,1,0,"space]"
    "(com.breel.wallpapers18)",2385,7180,7180,564,0,6760,0,420,0,0,0,0,0,0,0,1,0,"space]"
    "(com.android.networkstack.process)",2407,7180,7180,762,0,6556,0,624,0,0,0,0,0,0,0,1,0,"space]"
    "(.dataservices)",2436,7180,7180,614,0,6708,0,472,0,0,0,0,0,0,0,1,0,"space]"
    "(com.qualcomm.qti.telephonyservice)",2502,7180,7180,553,0,6772,0,408,0,0,0,0,0,0,0,1,0,"space]"
    "(com.google.android.grilservice)",2503,7180,7180,553,0,6772,0,408,0,0,0,0,0,0,0,1,0,"space]"
        """))