Add test suite for filtered models

This commit is contained in:
René Stadler 2008-01-10 14:14:12 +02:00 committed by Stefan Sauer
parent 5f3c94afff
commit 48ee98eb5c

201
debug-viewer/tests/test_models.py Executable file
View file

@ -0,0 +1,201 @@
#!/usr/bin/env python
# -*- coding: utf-8; mode: python; -*-
#
# GStreamer Debug Viewer - View and analyze GStreamer debug log files
#
# Copyright (C) 2007 René Stadler <mail@renestadler.de>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation; either version 3 of the License, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program. If not, see <http://www.gnu.org/licenses/>.
"""GStreamer Debug Viewer test suite for the custom tree models."""
import sys
import os
import os.path
from glob import glob
sys.path.insert (0, os.path.join (sys.path[0], os.pardir))
from unittest import TestCase, main as test_main
from GstDebugViewer import Common, Data, GUI
class Model (GUI.LogModelBase):
def __init__ (self):
GUI.LogModelBase.__init__ (self)
for i in range (20):
self.line_offsets.append (i * 100)
self.line_levels.append (Data.debug_level_debug)
def ensure_cached (self, line_offset):
pid = line_offset // 100
if pid % 2 == 0:
category = "EVEN"
else:
category = "ODD"
line_fmt = ("0:00:00.000000000 %5i 0x0000000 DEBUG "
"%20s dummy.c:1:dummy: dummy")
line_str = line_fmt % (pid, category,)
log_line = Data.LogLine.parse_full (line_str)
self.line_cache[line_offset] = log_line
def access_offset (self, line_offset):
return ""
class RandomFilter (GUI.Filter):
def __init__ (self, seed):
import random
rand = random.Random ()
rand.seed (seed)
def filter_func (row):
return rand.choice ((True, False,))
self.filter_func = filter_func
class TestDynamicFilter (TestCase):
def test_filtered_range_refilter (self):
full_model = Model ()
ranged_model = GUI.RangeFilteredLogModel (full_model)
# FIXME: Call to .reset should not be needed.
ranged_model.reset ()
filtered_model = GUI.FilteredLogModel (ranged_model)
row_list = self.__row_list
rows = row_list (full_model)
rows_ranged = row_list (ranged_model)
rows_filtered = row_list (filtered_model)
self.__dump_model (full_model, "full model")
## self.__dump_model (ranged_model, "ranged model")
## self.__dump_model (filtered_model, "filtered model")
self.assertEquals (rows, rows_ranged)
self.assertEquals (rows, rows_filtered)
ranged_model.set_range (5, 15)
self.__dump_model (ranged_model, "ranged model (5, 15)")
filtered_model.super_model_changed_range ()
rows_ranged = row_list (ranged_model)
self.assertEquals (rows_ranged, range (5, 16))
self.__dump_model (filtered_model, "filtered model (nofilter, 5, 15)")
rows_filtered = row_list (filtered_model)
self.assertEquals (rows_ranged, rows_filtered)
filtered_model.add_filter (GUI.CategoryFilter ("EVEN"),
Common.Data.DefaultDispatcher ())
rows_filtered = row_list (filtered_model)
self.assertEquals (rows_filtered, range (5, 16, 2))
self.__dump_model (filtered_model, "filtered model")
ranged_model.set_range (7, 12)
filtered_model.super_model_changed_range ()
self.__dump_model (ranged_model, "ranged model (7, 12)")
rows_ranged = row_list (ranged_model)
self.assertEquals (rows_ranged, range (7, 13))
self.__dump_model (filtered_model, "filtered model (ranged 7, 12)")
rows_filtered = row_list (filtered_model)
self.assertEquals (rows_filtered, range (7, 13, 2))
def test_random_filtered_range_refilter (self):
full_model = Model ()
ranged_model = GUI.RangeFilteredLogModel (full_model)
# FIXME: Call to .reset should not be needed.
ranged_model.reset ()
filtered_model = GUI.FilteredLogModel (ranged_model)
row_list = self.__row_list
self.assertEquals (row_list (full_model), range (20))
self.assertEquals (row_list (ranged_model), range (20))
self.assertEquals (row_list (filtered_model), range (20))
filtered_model.add_filter (RandomFilter (538295943),
Common.Data.DefaultDispatcher ())
random_rows = row_list (filtered_model)
self.__dump_model (filtered_model)
ranged_model.set_range (10, 19)
self.__dump_model (ranged_model, "ranged_model (10, 19)")
self.assertEquals (row_list (ranged_model), range (10, 20))
filtered_model.super_model_changed_range ()
self.__dump_model (filtered_model)
self.assertEquals (row_list (filtered_model), [x for x in range (10, 20) if x in random_rows])
ranged_model.set_range (0, 19)
self.assertEquals (row_list (ranged_model), range (0, 20))
ranged_model = GUI.RangeFilteredLogModel (full_model)
# FIXME: Call to .reset should not be needed.
ranged_model.reset ()
filtered_model = GUI.FilteredLogModel (ranged_model)
filtered_model.add_filter (RandomFilter (538295943),
Common.Data.DefaultDispatcher ())
self.__dump_model (filtered_model, "filtered model")
self.assertEquals (row_list (filtered_model), random_rows)
ranged_model.set_range (0, 9)
self.__dump_model (ranged_model, "ranged model (0, 9)")
filtered_model.super_model_changed_range ()
self.assertEquals (row_list (ranged_model), range (0, 10))
self.__dump_model (filtered_model)
self.assertEquals (row_list (filtered_model), [x for x in range (0, 10) if x in random_rows])
def __row_list (self, model):
return [row[Model.COL_PID] for row in model]
def __dump_model (self, model, comment = None):
# TODO: Provide a command line option to turn this on and off.
return
if not hasattr (model, "super_model"):
# Top model.
print "\t|%s|" % ("|".join ([str (i).rjust (2) for i in self.__row_list (model)]),),
else:
top_model = model.super_model
if hasattr (top_model, "super_model"):
top_model = top_model.super_model
top_indices = self.__row_list (top_model)
positions = self.__row_list (model)
output = [" "] * len (top_indices)
for i, position in enumerate (positions):
output[position] = str (i).rjust (2)
print "\t|%s|" % ("|".join (output),),
if comment is None:
print
else:
print comment
if __name__ == "__main__":
test_main ()