import unittest
import subprocess
import re

class Trace2009(unittest.TestCase):

  def __iternum(self, num, pattern):
    for index, item in enumerate(num):
      lines = re.findall(pattern, stdout)
      num_recs = int(lines[index])
      self.assertEqual(num_recs, item)

  def testFilters(self):
    """number of filtered records in branch A should be 166
       number of filtered records in branch B should be 166"""
    num = [166, 166]
    pattern = 'No. of Filtered Records: (\d+).*'
    self.__iternum(num, pattern)

  def testGroupers(self):
    """number of groups in branch A should be 32
       number of groups in branch B should be 32"""
    num = [32, 32]
    pattern = 'No. of Groups: (\d+).*'
    self.__iternum(num, pattern)

  def testGroupFilter(self):
    """number of group filters in branch A should be 3
       number of group filters in branch B should be 5"""
    num = [3, 5]
    pattern = 'No. of Filtered Groups: (\d+).*'
    self.__iternum(num, pattern)

  def testMergers(self):
    """number of merged groups should be 3"""
    num = [3]
    pattern = 'No. of Merged Groups: (\d+).*'
    self.__iternum(num, pattern)

  def testUngroupers(self):
    """number of streams should be 3
       number of records in stream 1 should be 24
       number of records in stream 2 should be 62
       number of records in stream 3 should be 56"""
    num = [3]
    pattern = 'No. of Streams: (\d+).*'
    self.__iternum(num, pattern)

    num = [24, 62, 56]
    pattern = 'No. of Records in Stream.+: (\d+).*'
    self.__iternum(num, pattern)

if __name__ == '__main__':
    stdout = subprocess.check_output(
                                      [
                                       'bin/engine',
                                       'examples/query-http-tcp-session.json',
                                       'examples/trace-2009.ft',
                                       '--verbose=1'
                                      ]
                                    )
    unittest.main()