#!/usr/bin/python3
# -*- encoding: utf-8 -*-
import logging
try:
import unittest2 as unittest
except ImportError:
import unittest
import datetime
import clitests
import problem
from abrtcli.filtering import (filter_reported,
filter_not_reported,
filter_since,
filter_since_timestamp,
filter_until,
filter_until_timestamp)
class FilteringTestCase(clitests.TestCase):
'''
Test filtering functionality
'''
def test_filter_since(self):
pl = problem.list()
since = datetime.datetime(2015, 1, 1, 1, 1, 1)
res = filter_since(pl, since)
self.assertEqual(len(res), 3)
def test_filter_since_timestamp(self):
pl = problem.list()
since = datetime.datetime(2015, 1, 1, 1, 1, 1)
since_ts = since.strftime('%s')
res = filter_since_timestamp(pl, since_ts)
self.assertEqual(len(res), 3)
def test_filter_until(self):
pl = problem.list()
until = datetime.datetime(2015, 1, 1, 1, 1, 1)
res = filter_until(pl, until)
self.assertEqual(len(res), 2)
def test_filter_until_timestamp(self):
pl = problem.list()
until = datetime.datetime(2015, 1, 1, 1, 1, 1)
until_ts = until.strftime('%s')
res = filter_until_timestamp(pl, until_ts)
self.assertEqual(len(res), 2)
def test_filter_reported(self):
pl = problem.list()
res = filter_reported(pl)
self.assertEqual(len(res), 1)
def test_filter_not_reported(self):
pl = problem.list()
res = filter_not_reported(pl)
self.assertEqual(len(res), 4)
if __name__ == '__main__':
logging.basicConfig(level=logging.INFO)
unittest.main()