/*
 * Copyright (C) 2022 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.
 */

#include "src/trace_processor/importers/common/deobfuscation_mapping_table.h"
#include <string>
#include "perfetto/ext/base/flat_hash_map.h"
#include "test/gtest_and_gmock.h"

namespace perfetto {
namespace trace_processor {
namespace {

using PackageId = DeobfuscationMappingTable::PackageId;

TEST(DeobfuscationMappingTable, EmptyTableByDefault) {
  TraceStorage storage;
  StringId xyz_id = storage.InternString("xyz");

  DeobfuscationMappingTable table;
  EXPECT_EQ(table.TranslateClass(xyz_id), std::nullopt);
  EXPECT_EQ(table.TranslateClass(PackageId{"app", 123}, xyz_id), std::nullopt);
}

TEST(DeobfuscationMappingTable, TranslateClassSingleInsert) {
  TraceStorage storage;
  StringId xyz_id = storage.InternString("xyz");
  StringId abc_id = storage.InternString("abc");
  StringId class_x_id = storage.InternString("class_X");

  DeobfuscationMappingTable table;
  table.AddClassTranslation(PackageId{"app", 123}, xyz_id, class_x_id,
                            base::FlatHashMap<StringId, StringId>{});
  EXPECT_EQ(table.TranslateClass(xyz_id), class_x_id);
  EXPECT_EQ(table.TranslateClass(PackageId{"app", 123}, xyz_id), class_x_id);
  EXPECT_EQ(table.TranslateClass(PackageId{"app", 124}, xyz_id), std::nullopt);
  EXPECT_EQ(table.TranslateClass(PackageId{"app", 123}, abc_id), std::nullopt);
}

TEST(DeobfuscationMappingTable, TranslateClassMultipleInsert) {
  TraceStorage storage;
  StringId xyz_id = storage.InternString("xyz");
  StringId abc_id = storage.InternString("abc");
  StringId class_x_id = storage.InternString("class_X");
  StringId class_y_id = storage.InternString("class_Y");
  StringId class_a_id = storage.InternString("class_A");

  DeobfuscationMappingTable table;
  table.AddClassTranslation(PackageId{"app1", 123}, xyz_id, class_x_id,
                            base::FlatHashMap<StringId, StringId>{});
  table.AddClassTranslation(PackageId{"app2", 123}, xyz_id, class_y_id,
                            base::FlatHashMap<StringId, StringId>{});
  table.AddClassTranslation(PackageId{"app3", 123}, abc_id, class_a_id,
                            base::FlatHashMap<StringId, StringId>{});
  EXPECT_EQ(table.TranslateClass(xyz_id), class_x_id);
  EXPECT_EQ(table.TranslateClass(abc_id), std::nullopt);
  EXPECT_EQ(table.TranslateClass(PackageId{"app1", 123}, xyz_id), class_x_id);
  EXPECT_EQ(table.TranslateClass(PackageId{"app2", 123}, xyz_id), class_y_id);
  EXPECT_EQ(table.TranslateClass(PackageId{"app1", 123}, abc_id), std::nullopt);
}

TEST(DeobfuscationMappingTable, TranslateMember) {
  TraceStorage storage;
  StringId xyz_id = storage.InternString("xyz");
  StringId abc_id = storage.InternString("abc");
  StringId class_x_id = storage.InternString("class_X");
  StringId mmm_1_id = storage.InternString("mmm1");
  StringId mmm_2_id = storage.InternString("mmm2");
  StringId mmm_3_id = storage.InternString("mmm3");
  StringId mmm_4_id = storage.InternString("mmm4");
  StringId member_1_id = storage.InternString("member_1");
  StringId member_2_id = storage.InternString("member_2");
  StringId member_3_id = storage.InternString("member_3");

  base::FlatHashMap<StringId, StringId> members;
  members[mmm_1_id] = member_1_id;
  members[mmm_2_id] = member_2_id;
  members[mmm_3_id] = member_3_id;
  DeobfuscationMappingTable table;
  table.AddClassTranslation(PackageId{"app1", 123}, xyz_id, class_x_id,
                            std::move(members));
  EXPECT_EQ(table.TranslateMember(PackageId{"app1", 123}, xyz_id, mmm_2_id),
            member_2_id);
  EXPECT_EQ(table.TranslateMember(PackageId{"app1", 123}, xyz_id, mmm_4_id),
            std::nullopt);
  EXPECT_EQ(table.TranslateMember(PackageId{"app1", 123}, abc_id, mmm_2_id),
            std::nullopt);
  EXPECT_EQ(table.TranslateMember(PackageId{"app1", 124}, xyz_id, mmm_2_id),
            std::nullopt);
}

}  // namespace
}  // namespace trace_processor
}  // namespace perfetto