1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
# -*- coding: utf-8 -*-
#
# Copyright (C) 2006 Edgewall Software
# All rights reserved.
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution. The terms
# are also available at http://trac.edgewall.com/license.html.
#
# This software consists of voluntary contributions made by many
# individuals. For the exact contribution history, see the revision
# history and logs, available at http://projects.edgewall.com/trac/.
from datetime import datetime, timedelta
import time
import unittest
from trac.core import *
from trac.db.sqlite_backend import _to_sql
from trac.test import EnvironmentStub, Mock
from tracspamfilter.model import LogEntry, schema
class LogEntryTestCase(unittest.TestCase):
def setUp(self):
self.env = EnvironmentStub()
db = self.env.get_db_cnx()
cursor = db.cursor()
for table in schema:
for stmt in _to_sql(table):
cursor.execute(stmt)
def test_purge(self):
now = datetime.now()
oneweekago = time.mktime((now - timedelta(weeks=1)).timetuple())
onedayago = time.mktime((now - timedelta(days=1)).timetuple())
LogEntry(self.env, oneweekago, '/foo', 'john', False, '127.0.0.1',
'', 'Test', False, 5, []).insert()
LogEntry(self.env, onedayago, '/foo', 'anonymous', False, '127.0.0.1',
'', 'Test', True, -3, []).insert()
LogEntry.purge(self.env, days=4)
log = list(LogEntry.select(self.env))
self.assertEqual(1, len(log))
entry = log[0]
self.assertEqual('anonymous', entry.author)
def suite():
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(LogEntryTestCase, 'test'))
return suite
if __name__ == '__main__':
unittest.main(defaultTest='suite')
|
