# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest

import mergetraces

class GroupByProcessAndThreadIdTestBasic(unittest.TestCase):
  def runTest(self):
    # (sec, usec, 'pid:tid', function address).
    input_trace = [
        (100, 10, '2000:2001', 0x5),
        (100, 11, '2000:2001', 0x3),
        (100, 13, '2000:1999', 0x8),
        (100, 14, '2000:2000', 0x7),
        (120, 13, '2001:2003', 0x9),
        (150, 12, '2001:2004', 0x6),
        (180, 11, '2000:2000', 0x1),
    ]

    # Functions should be grouped by thread-id and PIDs should not be
    # interleaved.
    expected_trace = [
        (100, 10, '2000:2001', 0x5),
        (100, 11, '2000:2001', 0x3),
        (100, 13, '2000:1999', 0x8),
        (100, 14, '2000:2000', 0x7),
        (180, 11, '2000:2000', 0x1),
        (120, 13, '2001:2003', 0x9),
        (150, 12, '2001:2004', 0x6),
    ]

    grouped_trace = mergetraces.GroupByProcessAndThreadId(input_trace)

    self.assertEqual(grouped_trace, expected_trace)

class GroupByProcessAndThreadIdFailsWithNonUniqueTIDs(unittest.TestCase):
  def runTest(self):
    # (sec, usec, 'pid:tid', function address).
    input_trace = [
        (100, 10, '1999:2001', 0x5),
        (100, 10, '1988:2001', 0x5),
    ]

    try:
      mergetraces.GroupByProcessAndThreadId(input_trace)
    except Exception:
      return

    self.fail('Multiple processes should not have a same thread-ID.')