#!/usr/bin/env python
# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest

import setup_modules  # pylint: disable=unused-import

import chromium_src.tools.metrics.common.codegen_shared as codegen_shared


class CodegenSharedTest(unittest.TestCase):
  def testHash(self) -> None:
    # Must match those in //base/metrics/metrics_hashes_unittest.cc.
    self.assertEqual(codegen_shared.HashName('Back'), 0x0557FA923DCEE4D0)
    self.assertEqual(codegen_shared.HashName('Forward'), 0x67D2F6740A8EAEBF)
    self.assertEqual(codegen_shared.HashName('NewTab'), 0x290EB683F96572F1)

  def testHashFieldTrialName(self):
    # Must match those in //components/variations/hashing_unittest.cc.
    known_hashes = {
      'a': 937752454,
      '1': 723085877,
      'Trial Name': 2713117220,
      'Group Name': 3201815843,
      'My Favorite Experiment': 3722155194,
      'My Awesome Group Name': 4109503236,
      'abcdefghijklmonpqrstuvwxyz': 787728696,
      '0123456789ABCDEF': 348858318,
    }
    for name, value in known_hashes.items():
      self.assertEqual(codegen_shared.HashFieldTrialName(name), value)


if __name__ == '__main__':
  unittest.main()