Migration to JCL
JMC migration to JCL: - Use SQLObject for persistence - Use PyXMPP DataForm implementation - test packages reorganisation Need to update component.py and config.py to finish the migration darcs-hash:20070221173604-86b55-17fb4a530f378b51b6b62a117a6f93c73c5be796.gz
This commit is contained in:
642
coverage.py
642
coverage.py
@@ -6,6 +6,8 @@
|
|||||||
# COVERAGE.PY -- COVERAGE TESTING
|
# COVERAGE.PY -- COVERAGE TESTING
|
||||||
#
|
#
|
||||||
# Gareth Rees, Ravenbrook Limited, 2001-12-04
|
# Gareth Rees, Ravenbrook Limited, 2001-12-04
|
||||||
|
# Ned Batchelder, 2004-12-12
|
||||||
|
# http://nedbatchelder.com/code/modules/coverage.html
|
||||||
#
|
#
|
||||||
#
|
#
|
||||||
# 1. INTRODUCTION
|
# 1. INTRODUCTION
|
||||||
@@ -20,34 +22,49 @@
|
|||||||
# interface and limitations. See [GDR 2001-12-04b] for requirements and
|
# interface and limitations. See [GDR 2001-12-04b] for requirements and
|
||||||
# design.
|
# design.
|
||||||
|
|
||||||
"""Usage:
|
r"""Usage:
|
||||||
|
|
||||||
coverage.py -x MODULE.py [ARG1 ARG2 ...]
|
coverage.py -x [-p] MODULE.py [ARG1 ARG2 ...]
|
||||||
Execute module, passing the given command-line arguments, collecting
|
Execute module, passing the given command-line arguments, collecting
|
||||||
coverage data.
|
coverage data. With the -p option, write to a temporary file containing
|
||||||
|
the machine name and process ID.
|
||||||
|
|
||||||
coverage.py -e
|
coverage.py -e
|
||||||
Erase collected coverage data.
|
Erase collected coverage data.
|
||||||
|
|
||||||
coverage.py -r [-m] FILE1 FILE2 ...
|
coverage.py -c
|
||||||
|
Collect data from multiple coverage files (as created by -p option above)
|
||||||
|
and store it into a single file representing the union of the coverage.
|
||||||
|
|
||||||
|
coverage.py -r [-m] [-o dir1,dir2,...] FILE1 FILE2 ...
|
||||||
Report on the statement coverage for the given files. With the -m
|
Report on the statement coverage for the given files. With the -m
|
||||||
option, show line numbers of the statements that weren't executed.
|
option, show line numbers of the statements that weren't executed.
|
||||||
|
|
||||||
coverage.py -a [-d dir] FILE1 FILE2 ...
|
coverage.py -a [-d dir] [-o dir1,dir2,...] FILE1 FILE2 ...
|
||||||
Make annotated copies of the given files, marking statements that
|
Make annotated copies of the given files, marking statements that
|
||||||
are executed with > and statements that are missed with !. With
|
are executed with > and statements that are missed with !. With
|
||||||
the -d option, make the copies in that directory. Without the -d
|
the -d option, make the copies in that directory. Without the -d
|
||||||
option, make each copy in the same directory as the original.
|
option, make each copy in the same directory as the original.
|
||||||
|
|
||||||
|
-o dir,dir2,...
|
||||||
|
Omit reporting or annotating files when their filename path starts with
|
||||||
|
a directory listed in the omit list.
|
||||||
|
e.g. python coverage.py -i -r -o c:\python23,lib\enthought\traits
|
||||||
|
|
||||||
Coverage data is saved in the file .coverage by default. Set the
|
Coverage data is saved in the file .coverage by default. Set the
|
||||||
COVERAGE_FILE environment variable to save it somewhere else."""
|
COVERAGE_FILE environment variable to save it somewhere else."""
|
||||||
|
|
||||||
|
__version__ = "2.6.20060823" # see detailed history at the end of this file.
|
||||||
|
|
||||||
|
import compiler
|
||||||
|
import compiler.visitor
|
||||||
import os
|
import os
|
||||||
import re
|
import re
|
||||||
import string
|
import string
|
||||||
import sys
|
import sys
|
||||||
|
import threading
|
||||||
import types
|
import types
|
||||||
|
from socket import gethostname
|
||||||
|
|
||||||
# 2. IMPLEMENTATION
|
# 2. IMPLEMENTATION
|
||||||
#
|
#
|
||||||
@@ -69,37 +86,175 @@ import types
|
|||||||
# In the bottleneck of this application it's appropriate to abbreviate
|
# In the bottleneck of this application it's appropriate to abbreviate
|
||||||
# names to increase speed.
|
# names to increase speed.
|
||||||
|
|
||||||
# A dictionary with an entry for (Python source file name, line number
|
class StatementFindingAstVisitor(compiler.visitor.ASTVisitor):
|
||||||
# in that file) if that line has been executed.
|
def __init__(self, statements, excluded, suite_spots):
|
||||||
c = {}
|
compiler.visitor.ASTVisitor.__init__(self)
|
||||||
|
self.statements = statements
|
||||||
|
self.excluded = excluded
|
||||||
|
self.suite_spots = suite_spots
|
||||||
|
self.excluding_suite = 0
|
||||||
|
|
||||||
# t(f, x, y). This method is passed to sys.settrace as a trace
|
def doRecursive(self, node):
|
||||||
# function. See [van Rossum 2001-07-20b, 9.2] for an explanation of
|
self.recordNodeLine(node)
|
||||||
# sys.settrace and the arguments and return value of the trace function.
|
for n in node.getChildNodes():
|
||||||
# See [van Rossum 2001-07-20a, 3.2] for a description of frame and code
|
self.dispatch(n)
|
||||||
# objects.
|
|
||||||
|
|
||||||
def t(f, x, y):
|
visitStmt = visitModule = doRecursive
|
||||||
c[(f.f_code.co_filename, f.f_lineno)] = 1
|
|
||||||
return t
|
def doCode(self, node):
|
||||||
|
if hasattr(node, 'decorators') and node.decorators:
|
||||||
|
self.dispatch(node.decorators)
|
||||||
|
self.recordAndDispatch(node.code)
|
||||||
|
else:
|
||||||
|
self.doSuite(node, node.code)
|
||||||
|
|
||||||
|
visitFunction = visitClass = doCode
|
||||||
|
|
||||||
|
def getFirstLine(self, node):
|
||||||
|
# Find the first line in the tree node.
|
||||||
|
lineno = node.lineno
|
||||||
|
for n in node.getChildNodes():
|
||||||
|
f = self.getFirstLine(n)
|
||||||
|
if lineno and f:
|
||||||
|
lineno = min(lineno, f)
|
||||||
|
else:
|
||||||
|
lineno = lineno or f
|
||||||
|
return lineno
|
||||||
|
|
||||||
|
def getLastLine(self, node):
|
||||||
|
# Find the first line in the tree node.
|
||||||
|
lineno = node.lineno
|
||||||
|
for n in node.getChildNodes():
|
||||||
|
lineno = max(lineno, self.getLastLine(n))
|
||||||
|
return lineno
|
||||||
|
|
||||||
|
def doStatement(self, node):
|
||||||
|
self.recordLine(self.getFirstLine(node))
|
||||||
|
|
||||||
|
visitAssert = visitAssign = visitAssTuple = visitDiscard = visitPrint = \
|
||||||
|
visitPrintnl = visitRaise = visitSubscript = visitDecorators = \
|
||||||
|
doStatement
|
||||||
|
|
||||||
|
def recordNodeLine(self, node):
|
||||||
|
return self.recordLine(node.lineno)
|
||||||
|
|
||||||
|
def recordLine(self, lineno):
|
||||||
|
# Returns a bool, whether the line is included or excluded.
|
||||||
|
if lineno:
|
||||||
|
# Multi-line tests introducing suites have to get charged to their
|
||||||
|
# keyword.
|
||||||
|
if lineno in self.suite_spots:
|
||||||
|
lineno = self.suite_spots[lineno][0]
|
||||||
|
# If we're inside an exluded suite, record that this line was
|
||||||
|
# excluded.
|
||||||
|
if self.excluding_suite:
|
||||||
|
self.excluded[lineno] = 1
|
||||||
|
return 0
|
||||||
|
# If this line is excluded, or suite_spots maps this line to
|
||||||
|
# another line that is exlcuded, then we're excluded.
|
||||||
|
elif self.excluded.has_key(lineno) or \
|
||||||
|
self.suite_spots.has_key(lineno) and \
|
||||||
|
self.excluded.has_key(self.suite_spots[lineno][1]):
|
||||||
|
return 0
|
||||||
|
# Otherwise, this is an executable line.
|
||||||
|
else:
|
||||||
|
self.statements[lineno] = 1
|
||||||
|
return 1
|
||||||
|
return 0
|
||||||
|
|
||||||
|
default = recordNodeLine
|
||||||
|
|
||||||
|
def recordAndDispatch(self, node):
|
||||||
|
self.recordNodeLine(node)
|
||||||
|
self.dispatch(node)
|
||||||
|
|
||||||
|
def doSuite(self, intro, body, exclude=0):
|
||||||
|
exsuite = self.excluding_suite
|
||||||
|
if exclude or (intro and not self.recordNodeLine(intro)):
|
||||||
|
self.excluding_suite = 1
|
||||||
|
self.recordAndDispatch(body)
|
||||||
|
self.excluding_suite = exsuite
|
||||||
|
|
||||||
|
def doPlainWordSuite(self, prevsuite, suite):
|
||||||
|
# Finding the exclude lines for else's is tricky, because they aren't
|
||||||
|
# present in the compiler parse tree. Look at the previous suite,
|
||||||
|
# and find its last line. If any line between there and the else's
|
||||||
|
# first line are excluded, then we exclude the else.
|
||||||
|
lastprev = self.getLastLine(prevsuite)
|
||||||
|
firstelse = self.getFirstLine(suite)
|
||||||
|
for l in range(lastprev+1, firstelse):
|
||||||
|
if self.suite_spots.has_key(l):
|
||||||
|
self.doSuite(None, suite, exclude=self.excluded.has_key(l))
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.doSuite(None, suite)
|
||||||
|
|
||||||
|
def doElse(self, prevsuite, node):
|
||||||
|
if node.else_:
|
||||||
|
self.doPlainWordSuite(prevsuite, node.else_)
|
||||||
|
|
||||||
|
def visitFor(self, node):
|
||||||
|
self.doSuite(node, node.body)
|
||||||
|
self.doElse(node.body, node)
|
||||||
|
|
||||||
|
def visitIf(self, node):
|
||||||
|
# The first test has to be handled separately from the rest.
|
||||||
|
# The first test is credited to the line with the "if", but the others
|
||||||
|
# are credited to the line with the test for the elif.
|
||||||
|
self.doSuite(node, node.tests[0][1])
|
||||||
|
for t, n in node.tests[1:]:
|
||||||
|
self.doSuite(t, n)
|
||||||
|
self.doElse(node.tests[-1][1], node)
|
||||||
|
|
||||||
|
def visitWhile(self, node):
|
||||||
|
self.doSuite(node, node.body)
|
||||||
|
self.doElse(node.body, node)
|
||||||
|
|
||||||
|
def visitTryExcept(self, node):
|
||||||
|
self.doSuite(node, node.body)
|
||||||
|
for i in range(len(node.handlers)):
|
||||||
|
a, b, h = node.handlers[i]
|
||||||
|
if not a:
|
||||||
|
# It's a plain "except:". Find the previous suite.
|
||||||
|
if i > 0:
|
||||||
|
prev = node.handlers[i-1][2]
|
||||||
|
else:
|
||||||
|
prev = node.body
|
||||||
|
self.doPlainWordSuite(prev, h)
|
||||||
|
else:
|
||||||
|
self.doSuite(a, h)
|
||||||
|
self.doElse(node.handlers[-1][2], node)
|
||||||
|
|
||||||
|
def visitTryFinally(self, node):
|
||||||
|
self.doSuite(node, node.body)
|
||||||
|
self.doPlainWordSuite(node.body, node.final)
|
||||||
|
|
||||||
|
def visitGlobal(self, node):
|
||||||
|
# "global" statements don't execute like others (they don't call the
|
||||||
|
# trace function), so don't record their line numbers.
|
||||||
|
pass
|
||||||
|
|
||||||
the_coverage = None
|
the_coverage = None
|
||||||
|
|
||||||
class coverage:
|
class CoverageException(Exception): pass
|
||||||
error = "coverage error"
|
|
||||||
|
|
||||||
|
class coverage:
|
||||||
# Name of the cache file (unless environment variable is set).
|
# Name of the cache file (unless environment variable is set).
|
||||||
cache_default = ".coverage"
|
cache_default = ".coverage"
|
||||||
|
|
||||||
# Environment variable naming the cache file.
|
# Environment variable naming the cache file.
|
||||||
cache_env = "COVERAGE_FILE"
|
cache_env = "COVERAGE_FILE"
|
||||||
|
|
||||||
|
# A dictionary with an entry for (Python source file name, line number
|
||||||
|
# in that file) if that line has been executed.
|
||||||
|
c = {}
|
||||||
|
|
||||||
# A map from canonical Python source file name to a dictionary in
|
# A map from canonical Python source file name to a dictionary in
|
||||||
# which there's an entry for each line number that has been
|
# which there's an entry for each line number that has been
|
||||||
# executed.
|
# executed.
|
||||||
cexecuted = {}
|
cexecuted = {}
|
||||||
|
|
||||||
# Cache of results of calling the analysis() method, so that you can
|
# Cache of results of calling the analysis2() method, so that you can
|
||||||
# specify both -r and -a without doing double work.
|
# specify both -r and -a without doing double work.
|
||||||
analysis_cache = {}
|
analysis_cache = {}
|
||||||
|
|
||||||
@@ -110,10 +265,27 @@ class coverage:
|
|||||||
def __init__(self):
|
def __init__(self):
|
||||||
global the_coverage
|
global the_coverage
|
||||||
if the_coverage:
|
if the_coverage:
|
||||||
raise self.error, "Only one coverage object allowed."
|
raise CoverageException, "Only one coverage object allowed."
|
||||||
self.cache = os.environ.get(self.cache_env, self.cache_default)
|
self.usecache = 1
|
||||||
self.restore()
|
self.cache = None
|
||||||
self.analysis_cache = {}
|
self.exclude_re = ''
|
||||||
|
self.nesting = 0
|
||||||
|
self.cstack = []
|
||||||
|
self.xstack = []
|
||||||
|
self.relative_dir = os.path.normcase(os.path.abspath(os.curdir)+os.path.sep)
|
||||||
|
|
||||||
|
# t(f, x, y). This method is passed to sys.settrace as a trace function.
|
||||||
|
# See [van Rossum 2001-07-20b, 9.2] for an explanation of sys.settrace and
|
||||||
|
# the arguments and return value of the trace function.
|
||||||
|
# See [van Rossum 2001-07-20a, 3.2] for a description of frame and code
|
||||||
|
# objects.
|
||||||
|
|
||||||
|
def t(self, f, w, a): #pragma: no cover
|
||||||
|
if w == 'line':
|
||||||
|
self.c[(f.f_code.co_filename, f.f_lineno)] = 1
|
||||||
|
for c in self.cstack:
|
||||||
|
c[(f.f_code.co_filename, f.f_lineno)] = 1
|
||||||
|
return self.t
|
||||||
|
|
||||||
def help(self, error=None):
|
def help(self, error=None):
|
||||||
if error:
|
if error:
|
||||||
@@ -122,23 +294,26 @@ class coverage:
|
|||||||
print __doc__
|
print __doc__
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
def command_line(self):
|
def command_line(self, argv, help=None):
|
||||||
import getopt
|
import getopt
|
||||||
|
help = help or self.help
|
||||||
settings = {}
|
settings = {}
|
||||||
optmap = {
|
optmap = {
|
||||||
'-a': 'annotate',
|
'-a': 'annotate',
|
||||||
|
'-c': 'collect',
|
||||||
'-d:': 'directory=',
|
'-d:': 'directory=',
|
||||||
'-e': 'erase',
|
'-e': 'erase',
|
||||||
'-h': 'help',
|
'-h': 'help',
|
||||||
'-i': 'ignore-errors',
|
'-i': 'ignore-errors',
|
||||||
'-m': 'show-missing',
|
'-m': 'show-missing',
|
||||||
|
'-p': 'parallel-mode',
|
||||||
'-r': 'report',
|
'-r': 'report',
|
||||||
'-x': 'execute',
|
'-x': 'execute',
|
||||||
|
'-o:': 'omit=',
|
||||||
}
|
}
|
||||||
short_opts = string.join(map(lambda o: o[1:], optmap.keys()), '')
|
short_opts = string.join(map(lambda o: o[1:], optmap.keys()), '')
|
||||||
long_opts = optmap.values()
|
long_opts = optmap.values()
|
||||||
options, args = getopt.getopt(sys.argv[1:], short_opts,
|
options, args = getopt.getopt(argv, short_opts, long_opts)
|
||||||
long_opts)
|
|
||||||
for o, a in options:
|
for o, a in options:
|
||||||
if optmap.has_key(o):
|
if optmap.has_key(o):
|
||||||
settings[optmap[o]] = 1
|
settings[optmap[o]] = 1
|
||||||
@@ -147,86 +322,167 @@ class coverage:
|
|||||||
elif o[2:] in long_opts:
|
elif o[2:] in long_opts:
|
||||||
settings[o[2:]] = 1
|
settings[o[2:]] = 1
|
||||||
elif o[2:] + '=' in long_opts:
|
elif o[2:] + '=' in long_opts:
|
||||||
settings[o[2:]] = a
|
settings[o[2:]+'='] = a
|
||||||
else:
|
else: #pragma: no cover
|
||||||
self.help("Unknown option: '%s'." % o)
|
pass # Can't get here, because getopt won't return anything unknown.
|
||||||
|
|
||||||
if settings.get('help'):
|
if settings.get('help'):
|
||||||
self.help()
|
help()
|
||||||
|
|
||||||
for i in ['erase', 'execute']:
|
for i in ['erase', 'execute']:
|
||||||
for j in ['annotate', 'report']:
|
for j in ['annotate', 'report', 'collect']:
|
||||||
if settings.get(i) and settings.get(j):
|
if settings.get(i) and settings.get(j):
|
||||||
self.help("You can't specify the '%s' and '%s' "
|
help("You can't specify the '%s' and '%s' "
|
||||||
"options at the same time." % (i, j))
|
"options at the same time." % (i, j))
|
||||||
|
|
||||||
args_needed = (settings.get('execute')
|
args_needed = (settings.get('execute')
|
||||||
or settings.get('annotate')
|
or settings.get('annotate')
|
||||||
or settings.get('report'))
|
or settings.get('report'))
|
||||||
action = settings.get('erase') or args_needed
|
action = (settings.get('erase')
|
||||||
|
or settings.get('collect')
|
||||||
|
or args_needed)
|
||||||
if not action:
|
if not action:
|
||||||
self.help("You must specify at least one of -e, -x, -r, "
|
help("You must specify at least one of -e, -x, -c, -r, or -a.")
|
||||||
"or -a.")
|
|
||||||
if not args_needed and args:
|
if not args_needed and args:
|
||||||
self.help("Unexpected arguments %s." % args)
|
help("Unexpected arguments: %s" % " ".join(args))
|
||||||
|
|
||||||
|
self.get_ready(settings.get('parallel-mode'))
|
||||||
|
self.exclude('#pragma[: ]+[nN][oO] [cC][oO][vV][eE][rR]')
|
||||||
|
|
||||||
if settings.get('erase'):
|
if settings.get('erase'):
|
||||||
self.erase()
|
self.erase()
|
||||||
if settings.get('execute'):
|
if settings.get('execute'):
|
||||||
if not args:
|
if not args:
|
||||||
self.help("Nothing to do.")
|
help("Nothing to do.")
|
||||||
sys.argv = args
|
sys.argv = args
|
||||||
self.start()
|
self.start()
|
||||||
import __main__
|
import __main__
|
||||||
sys.path[0] = os.path.dirname(sys.argv[0])
|
sys.path[0] = os.path.dirname(sys.argv[0])
|
||||||
execfile(sys.argv[0], __main__.__dict__)
|
execfile(sys.argv[0], __main__.__dict__)
|
||||||
|
if settings.get('collect'):
|
||||||
|
self.collect()
|
||||||
if not args:
|
if not args:
|
||||||
args = self.cexecuted.keys()
|
args = self.cexecuted.keys()
|
||||||
|
|
||||||
ignore_errors = settings.get('ignore-errors')
|
ignore_errors = settings.get('ignore-errors')
|
||||||
show_missing = settings.get('show-missing')
|
show_missing = settings.get('show-missing')
|
||||||
directory = settings.get('directory=')
|
directory = settings.get('directory=')
|
||||||
if settings.get('report'):
|
|
||||||
self.report(args, show_missing, ignore_errors)
|
|
||||||
if settings.get('annotate'):
|
|
||||||
self.annotate(args, directory, ignore_errors)
|
|
||||||
|
|
||||||
def start(self):
|
omit = settings.get('omit=')
|
||||||
sys.settrace(t)
|
if omit is not None:
|
||||||
|
omit = omit.split(',')
|
||||||
|
else:
|
||||||
|
omit = []
|
||||||
|
|
||||||
|
if settings.get('report'):
|
||||||
|
self.report(args, show_missing, ignore_errors, omit_prefixes=omit)
|
||||||
|
if settings.get('annotate'):
|
||||||
|
self.annotate(args, directory, ignore_errors, omit_prefixes=omit)
|
||||||
|
|
||||||
|
def use_cache(self, usecache, cache_file=None):
|
||||||
|
self.usecache = usecache
|
||||||
|
if cache_file and not self.cache:
|
||||||
|
self.cache_default = cache_file
|
||||||
|
|
||||||
|
def get_ready(self, parallel_mode=False):
|
||||||
|
if self.usecache and not self.cache:
|
||||||
|
self.cache = os.environ.get(self.cache_env, self.cache_default)
|
||||||
|
if parallel_mode:
|
||||||
|
self.cache += "." + gethostname() + "." + str(os.getpid())
|
||||||
|
self.restore()
|
||||||
|
self.analysis_cache = {}
|
||||||
|
|
||||||
|
def start(self, parallel_mode=False):
|
||||||
|
self.get_ready(parallel_mode)
|
||||||
|
if self.nesting == 0: #pragma: no cover
|
||||||
|
sys.settrace(self.t)
|
||||||
|
if hasattr(threading, 'settrace'):
|
||||||
|
threading.settrace(self.t)
|
||||||
|
self.nesting += 1
|
||||||
|
|
||||||
def stop(self):
|
def stop(self):
|
||||||
|
self.nesting -= 1
|
||||||
|
if self.nesting == 0: #pragma: no cover
|
||||||
sys.settrace(None)
|
sys.settrace(None)
|
||||||
|
if hasattr(threading, 'settrace'):
|
||||||
|
threading.settrace(None)
|
||||||
|
|
||||||
def erase(self):
|
def erase(self):
|
||||||
global c
|
self.c = {}
|
||||||
c = {}
|
|
||||||
self.analysis_cache = {}
|
self.analysis_cache = {}
|
||||||
self.cexecuted = {}
|
self.cexecuted = {}
|
||||||
if os.path.exists(self.cache):
|
if self.cache and os.path.exists(self.cache):
|
||||||
os.remove(self.cache)
|
os.remove(self.cache)
|
||||||
|
self.exclude_re = ""
|
||||||
|
|
||||||
|
def exclude(self, re):
|
||||||
|
if self.exclude_re:
|
||||||
|
self.exclude_re += "|"
|
||||||
|
self.exclude_re += "(" + re + ")"
|
||||||
|
|
||||||
|
def begin_recursive(self):
|
||||||
|
self.cstack.append(self.c)
|
||||||
|
self.xstack.append(self.exclude_re)
|
||||||
|
|
||||||
|
def end_recursive(self):
|
||||||
|
self.c = self.cstack.pop()
|
||||||
|
self.exclude_re = self.xstack.pop()
|
||||||
|
|
||||||
# save(). Save coverage data to the coverage cache.
|
# save(). Save coverage data to the coverage cache.
|
||||||
|
|
||||||
def save(self):
|
def save(self):
|
||||||
|
if self.usecache and self.cache:
|
||||||
self.canonicalize_filenames()
|
self.canonicalize_filenames()
|
||||||
cache = open(self.cache, 'wb')
|
cache = open(self.cache, 'wb')
|
||||||
import marshal
|
import marshal
|
||||||
marshal.dump(self.cexecuted, cache)
|
marshal.dump(self.cexecuted, cache)
|
||||||
cache.close()
|
cache.close()
|
||||||
|
|
||||||
# restore(). Restore coverage data from the coverage cache (if it
|
# restore(). Restore coverage data from the coverage cache (if it exists).
|
||||||
# exists).
|
|
||||||
|
|
||||||
def restore(self):
|
def restore(self):
|
||||||
global c
|
self.c = {}
|
||||||
c = {}
|
|
||||||
self.cexecuted = {}
|
self.cexecuted = {}
|
||||||
if not os.path.exists(self.cache):
|
assert self.usecache
|
||||||
return
|
if os.path.exists(self.cache):
|
||||||
|
self.cexecuted = self.restore_file(self.cache)
|
||||||
|
|
||||||
|
def restore_file(self, file_name):
|
||||||
try:
|
try:
|
||||||
cache = open(self.cache, 'rb')
|
cache = open(file_name, 'rb')
|
||||||
import marshal
|
import marshal
|
||||||
cexecuted = marshal.load(cache)
|
cexecuted = marshal.load(cache)
|
||||||
cache.close()
|
cache.close()
|
||||||
if isinstance(cexecuted, types.DictType):
|
if isinstance(cexecuted, types.DictType):
|
||||||
self.cexecuted = cexecuted
|
return cexecuted
|
||||||
|
else:
|
||||||
|
return {}
|
||||||
except:
|
except:
|
||||||
pass
|
return {}
|
||||||
|
|
||||||
|
# collect(). Collect data in multiple files produced by parallel mode
|
||||||
|
|
||||||
|
def collect(self):
|
||||||
|
cache_dir, local = os.path.split(self.cache)
|
||||||
|
for file in os.listdir(cache_dir):
|
||||||
|
if not file.startswith(local):
|
||||||
|
continue
|
||||||
|
|
||||||
|
full_path = os.path.join(cache_dir, file)
|
||||||
|
cexecuted = self.restore_file(full_path)
|
||||||
|
self.merge_data(cexecuted)
|
||||||
|
|
||||||
|
def merge_data(self, new_data):
|
||||||
|
for file_name, file_data in new_data.items():
|
||||||
|
if self.cexecuted.has_key(file_name):
|
||||||
|
self.merge_file_data(self.cexecuted[file_name], file_data)
|
||||||
|
else:
|
||||||
|
self.cexecuted[file_name] = file_data
|
||||||
|
|
||||||
|
def merge_file_data(self, cache_data, new_data):
|
||||||
|
for line_number in new_data.keys():
|
||||||
|
if not cache_data.has_key(line_number):
|
||||||
|
cache_data[line_number] = new_data[line_number]
|
||||||
|
|
||||||
# canonical_filename(filename). Return a canonical filename for the
|
# canonical_filename(filename). Return a canonical filename for the
|
||||||
# file (that is, an absolute path with no redundant components and
|
# file (that is, an absolute path with no redundant components and
|
||||||
@@ -247,25 +503,23 @@ class coverage:
|
|||||||
self.canonical_filename_cache[filename] = cf
|
self.canonical_filename_cache[filename] = cf
|
||||||
return self.canonical_filename_cache[filename]
|
return self.canonical_filename_cache[filename]
|
||||||
|
|
||||||
# canonicalize_filenames(). Copy results from "executed" to
|
# canonicalize_filenames(). Copy results from "c" to "cexecuted",
|
||||||
# "cexecuted", canonicalizing filenames on the way. Clear the
|
# canonicalizing filenames on the way. Clear the "c" map.
|
||||||
# "executed" map.
|
|
||||||
|
|
||||||
def canonicalize_filenames(self):
|
def canonicalize_filenames(self):
|
||||||
global c
|
for filename, lineno in self.c.keys():
|
||||||
for filename, lineno in c.keys():
|
|
||||||
f = self.canonical_filename(filename)
|
f = self.canonical_filename(filename)
|
||||||
if not self.cexecuted.has_key(f):
|
if not self.cexecuted.has_key(f):
|
||||||
self.cexecuted[f] = {}
|
self.cexecuted[f] = {}
|
||||||
self.cexecuted[f][lineno] = 1
|
self.cexecuted[f][lineno] = 1
|
||||||
c = {}
|
self.c = {}
|
||||||
|
|
||||||
# morf_filename(morf). Return the filename for a module or file.
|
# morf_filename(morf). Return the filename for a module or file.
|
||||||
|
|
||||||
def morf_filename(self, morf):
|
def morf_filename(self, morf):
|
||||||
if isinstance(morf, types.ModuleType):
|
if isinstance(morf, types.ModuleType):
|
||||||
if not hasattr(morf, '__file__'):
|
if not hasattr(morf, '__file__'):
|
||||||
raise self.error, "Module has no __file__ attribute."
|
raise CoverageException, "Module has no __file__ attribute."
|
||||||
file = morf.__file__
|
file = morf.__file__
|
||||||
else:
|
else:
|
||||||
file = morf
|
file = morf
|
||||||
@@ -273,9 +527,9 @@ class coverage:
|
|||||||
|
|
||||||
# analyze_morf(morf). Analyze the module or filename passed as
|
# analyze_morf(morf). Analyze the module or filename passed as
|
||||||
# the argument. If the source code can't be found, raise an error.
|
# the argument. If the source code can't be found, raise an error.
|
||||||
# Otherwise, return a pair of (1) the canonical filename of the
|
# Otherwise, return a tuple of (1) the canonical filename of the
|
||||||
# source code for the module, and (2) a list of lines of statements
|
# source code for the module, (2) a list of lines of statements
|
||||||
# in the source code.
|
# in the source code, and (3) a list of lines of excluded statements.
|
||||||
|
|
||||||
def analyze_morf(self, morf):
|
def analyze_morf(self, morf):
|
||||||
if self.analysis_cache.has_key(morf):
|
if self.analysis_cache.has_key(morf):
|
||||||
@@ -284,57 +538,82 @@ class coverage:
|
|||||||
ext = os.path.splitext(filename)[1]
|
ext = os.path.splitext(filename)[1]
|
||||||
if ext == '.pyc':
|
if ext == '.pyc':
|
||||||
if not os.path.exists(filename[0:-1]):
|
if not os.path.exists(filename[0:-1]):
|
||||||
raise self.error, ("No source for compiled code '%s'."
|
raise CoverageException, ("No source for compiled code '%s'."
|
||||||
% filename)
|
% filename)
|
||||||
filename = filename[0:-1]
|
filename = filename[0:-1]
|
||||||
elif ext != '.py':
|
elif ext != '.py':
|
||||||
raise self.error, "File '%s' not Python source." % filename
|
raise CoverageException, "File '%s' not Python source." % filename
|
||||||
source = open(filename, 'r')
|
source = open(filename, 'r')
|
||||||
import parser
|
lines, excluded_lines = self.find_executable_statements(
|
||||||
tree = parser.suite(source.read()).totuple(1)
|
source.read(), exclude=self.exclude_re
|
||||||
|
)
|
||||||
source.close()
|
source.close()
|
||||||
statements = {}
|
result = filename, lines, excluded_lines
|
||||||
self.find_statements(tree, statements)
|
|
||||||
lines = statements.keys()
|
|
||||||
lines.sort()
|
|
||||||
result = filename, lines
|
|
||||||
self.analysis_cache[morf] = result
|
self.analysis_cache[morf] = result
|
||||||
return result
|
return result
|
||||||
|
|
||||||
# find_statements(tree, dict). Find each statement in the parse
|
def get_suite_spots(self, tree, spots):
|
||||||
# tree and record the line on which the statement starts in the
|
|
||||||
# dictionary (by assigning it to 1).
|
|
||||||
#
|
|
||||||
# It works by walking the whole tree depth-first. Every time it
|
|
||||||
# comes across a statement (symbol.stmt -- this includes compound
|
|
||||||
# statements like 'if' and 'while') it calls find_statement, which
|
|
||||||
# descends the tree below the statement to find the first terminal
|
|
||||||
# token in that statement and record the lines on which that token
|
|
||||||
# was found.
|
|
||||||
#
|
|
||||||
# This algorithm may find some lines several times (because of the
|
|
||||||
# grammar production statement -> compound statement -> statement),
|
|
||||||
# but that doesn't matter because we record lines as the keys of the
|
|
||||||
# dictionary.
|
|
||||||
#
|
|
||||||
# See also [GDR 2001-12-04b, 3.2].
|
|
||||||
|
|
||||||
def find_statements(self, tree, dict):
|
|
||||||
import symbol, token
|
import symbol, token
|
||||||
if token.ISNONTERMINAL(tree[0]):
|
for i in range(1, len(tree)):
|
||||||
for t in tree[1:]:
|
if type(tree[i]) == type(()):
|
||||||
self.find_statements(t, dict)
|
if tree[i][0] == symbol.suite:
|
||||||
if tree[0] == symbol.stmt:
|
# Found a suite, look back for the colon and keyword.
|
||||||
self.find_statement(tree[1], dict)
|
lineno_colon = lineno_word = None
|
||||||
elif (tree[0] == token.NAME
|
for j in range(i-1, 0, -1):
|
||||||
and tree[1] in ['elif', 'except', 'finally']):
|
if tree[j][0] == token.COLON:
|
||||||
dict[tree[2]] = 1
|
lineno_colon = tree[j][2]
|
||||||
|
elif tree[j][0] == token.NAME:
|
||||||
|
if tree[j][1] == 'elif':
|
||||||
|
# Find the line number of the first non-terminal
|
||||||
|
# after the keyword.
|
||||||
|
t = tree[j+1]
|
||||||
|
while t and token.ISNONTERMINAL(t[0]):
|
||||||
|
t = t[1]
|
||||||
|
if t:
|
||||||
|
lineno_word = t[2]
|
||||||
|
else:
|
||||||
|
lineno_word = tree[j][2]
|
||||||
|
break
|
||||||
|
elif tree[j][0] == symbol.except_clause:
|
||||||
|
# "except" clauses look like:
|
||||||
|
# ('except_clause', ('NAME', 'except', lineno), ...)
|
||||||
|
if tree[j][1][0] == token.NAME:
|
||||||
|
lineno_word = tree[j][1][2]
|
||||||
|
break
|
||||||
|
if lineno_colon and lineno_word:
|
||||||
|
# Found colon and keyword, mark all the lines
|
||||||
|
# between the two with the two line numbers.
|
||||||
|
for l in range(lineno_word, lineno_colon+1):
|
||||||
|
spots[l] = (lineno_word, lineno_colon)
|
||||||
|
self.get_suite_spots(tree[i], spots)
|
||||||
|
|
||||||
def find_statement(self, tree, dict):
|
def find_executable_statements(self, text, exclude=None):
|
||||||
import token
|
# Find lines which match an exclusion pattern.
|
||||||
while token.ISNONTERMINAL(tree[0]):
|
excluded = {}
|
||||||
tree = tree[1]
|
suite_spots = {}
|
||||||
dict[tree[2]] = 1
|
if exclude:
|
||||||
|
reExclude = re.compile(exclude)
|
||||||
|
lines = text.split('\n')
|
||||||
|
for i in range(len(lines)):
|
||||||
|
if reExclude.search(lines[i]):
|
||||||
|
excluded[i+1] = 1
|
||||||
|
|
||||||
|
import parser
|
||||||
|
tree = parser.suite(text+'\n\n').totuple(1)
|
||||||
|
self.get_suite_spots(tree, suite_spots)
|
||||||
|
|
||||||
|
# Use the compiler module to parse the text and find the executable
|
||||||
|
# statements. We add newlines to be impervious to final partial lines.
|
||||||
|
statements = {}
|
||||||
|
ast = compiler.parse(text+'\n\n')
|
||||||
|
visitor = StatementFindingAstVisitor(statements, excluded, suite_spots)
|
||||||
|
compiler.walk(ast, visitor, walker=visitor)
|
||||||
|
|
||||||
|
lines = statements.keys()
|
||||||
|
lines.sort()
|
||||||
|
excluded_lines = excluded.keys()
|
||||||
|
excluded_lines.sort()
|
||||||
|
return lines, excluded_lines
|
||||||
|
|
||||||
# format_lines(statements, lines). Format a list of line numbers
|
# format_lines(statements, lines). Format a list of line numbers
|
||||||
# for printing by coalescing groups of lines as long as the lines
|
# for printing by coalescing groups of lines as long as the lines
|
||||||
@@ -367,11 +646,15 @@ class coverage:
|
|||||||
return "%d" % start
|
return "%d" % start
|
||||||
else:
|
else:
|
||||||
return "%d-%d" % (start, end)
|
return "%d-%d" % (start, end)
|
||||||
import string
|
|
||||||
return string.join(map(stringify, pairs), ", ")
|
return string.join(map(stringify, pairs), ", ")
|
||||||
|
|
||||||
|
# Backward compatibility with version 1.
|
||||||
def analysis(self, morf):
|
def analysis(self, morf):
|
||||||
filename, statements = self.analyze_morf(morf)
|
f, s, _, m, mf = self.analysis2(morf)
|
||||||
|
return f, s, m, mf
|
||||||
|
|
||||||
|
def analysis2(self, morf):
|
||||||
|
filename, statements, excluded = self.analyze_morf(morf)
|
||||||
self.canonicalize_filenames()
|
self.canonicalize_filenames()
|
||||||
if not self.cexecuted.has_key(filename):
|
if not self.cexecuted.has_key(filename):
|
||||||
self.cexecuted[filename] = {}
|
self.cexecuted[filename] = {}
|
||||||
@@ -379,18 +662,45 @@ class coverage:
|
|||||||
for line in statements:
|
for line in statements:
|
||||||
if not self.cexecuted[filename].has_key(line):
|
if not self.cexecuted[filename].has_key(line):
|
||||||
missing.append(line)
|
missing.append(line)
|
||||||
return (filename, statements, missing,
|
return (filename, statements, excluded, missing,
|
||||||
self.format_lines(statements, missing))
|
self.format_lines(statements, missing))
|
||||||
|
|
||||||
|
def relative_filename(self, filename):
|
||||||
|
""" Convert filename to relative filename from self.relative_dir.
|
||||||
|
"""
|
||||||
|
return filename.replace(self.relative_dir, "")
|
||||||
|
|
||||||
def morf_name(self, morf):
|
def morf_name(self, morf):
|
||||||
|
""" Return the name of morf as used in report.
|
||||||
|
"""
|
||||||
if isinstance(morf, types.ModuleType):
|
if isinstance(morf, types.ModuleType):
|
||||||
return morf.__name__
|
return morf.__name__
|
||||||
else:
|
else:
|
||||||
return os.path.splitext(os.path.basename(morf))[0]
|
return self.relative_filename(os.path.splitext(morf)[0])
|
||||||
|
|
||||||
def report(self, morfs, show_missing=1, ignore_errors=0):
|
def filter_by_prefix(self, morfs, omit_prefixes):
|
||||||
|
""" Return list of morfs where the morf name does not begin
|
||||||
|
with any one of the omit_prefixes.
|
||||||
|
"""
|
||||||
|
filtered_morfs = []
|
||||||
|
for morf in morfs:
|
||||||
|
for prefix in omit_prefixes:
|
||||||
|
if self.morf_name(morf).startswith(prefix):
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
filtered_morfs.append(morf)
|
||||||
|
|
||||||
|
return filtered_morfs
|
||||||
|
|
||||||
|
def morf_name_compare(self, x, y):
|
||||||
|
return cmp(self.morf_name(x), self.morf_name(y))
|
||||||
|
|
||||||
|
def report(self, morfs, show_missing=1, ignore_errors=0, file=None, omit_prefixes=[]):
|
||||||
if not isinstance(morfs, types.ListType):
|
if not isinstance(morfs, types.ListType):
|
||||||
morfs = [morfs]
|
morfs = [morfs]
|
||||||
|
morfs = self.filter_by_prefix(morfs, omit_prefixes)
|
||||||
|
morfs.sort(self.morf_name_compare)
|
||||||
|
|
||||||
max_name = max([5,] + map(len, map(self.morf_name, morfs)))
|
max_name = max([5,] + map(len, map(self.morf_name, morfs)))
|
||||||
fmt_name = "%%- %ds " % max_name
|
fmt_name = "%%- %ds " % max_name
|
||||||
fmt_err = fmt_name + "%s: %s"
|
fmt_err = fmt_name + "%s: %s"
|
||||||
@@ -399,14 +709,16 @@ class coverage:
|
|||||||
if show_missing:
|
if show_missing:
|
||||||
header = header + " Missing"
|
header = header + " Missing"
|
||||||
fmt_coverage = fmt_coverage + " %s"
|
fmt_coverage = fmt_coverage + " %s"
|
||||||
print header
|
if not file:
|
||||||
print "-" * len(header)
|
file = sys.stdout
|
||||||
|
print >>file, header
|
||||||
|
print >>file, "-" * len(header)
|
||||||
total_statements = 0
|
total_statements = 0
|
||||||
total_executed = 0
|
total_executed = 0
|
||||||
for morf in morfs:
|
for morf in morfs:
|
||||||
name = self.morf_name(morf)
|
name = self.morf_name(morf)
|
||||||
try:
|
try:
|
||||||
_, statements, missing, readable = self.analysis(morf)
|
_, statements, _, missing, readable = self.analysis2(morf)
|
||||||
n = len(statements)
|
n = len(statements)
|
||||||
m = n - len(missing)
|
m = n - len(missing)
|
||||||
if n > 0:
|
if n > 0:
|
||||||
@@ -416,17 +728,17 @@ class coverage:
|
|||||||
args = (name, n, m, pc)
|
args = (name, n, m, pc)
|
||||||
if show_missing:
|
if show_missing:
|
||||||
args = args + (readable,)
|
args = args + (readable,)
|
||||||
print fmt_coverage % args
|
print >>file, fmt_coverage % args
|
||||||
total_statements = total_statements + n
|
total_statements = total_statements + n
|
||||||
total_executed = total_executed + m
|
total_executed = total_executed + m
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt: #pragma: no cover
|
||||||
raise
|
raise
|
||||||
except:
|
except:
|
||||||
if not ignore_errors:
|
if not ignore_errors:
|
||||||
type, msg = sys.exc_info()[0:2]
|
type, msg = sys.exc_info()[0:2]
|
||||||
print fmt_err % (name, type, msg)
|
print >>file, fmt_err % (name, type, msg)
|
||||||
if len(morfs) > 1:
|
if len(morfs) > 1:
|
||||||
print "-" * len(header)
|
print >>file, "-" * len(header)
|
||||||
if total_statements > 0:
|
if total_statements > 0:
|
||||||
pc = 100.0 * total_executed / total_statements
|
pc = 100.0 * total_executed / total_statements
|
||||||
else:
|
else:
|
||||||
@@ -434,17 +746,26 @@ class coverage:
|
|||||||
args = ("TOTAL", total_statements, total_executed, pc)
|
args = ("TOTAL", total_statements, total_executed, pc)
|
||||||
if show_missing:
|
if show_missing:
|
||||||
args = args + ("",)
|
args = args + ("",)
|
||||||
print fmt_coverage % args
|
print >>file, fmt_coverage % args
|
||||||
|
|
||||||
# annotate(morfs, ignore_errors).
|
# annotate(morfs, ignore_errors).
|
||||||
|
|
||||||
blank_re = re.compile("\\s*(#|$)")
|
blank_re = re.compile(r"\s*(#|$)")
|
||||||
else_re = re.compile("\\s*else\\s*:\\s*(#|$)")
|
else_re = re.compile(r"\s*else\s*:\s*(#|$)")
|
||||||
|
|
||||||
def annotate(self, morfs, directory=None, ignore_errors=0):
|
def annotate(self, morfs, directory=None, ignore_errors=0, omit_prefixes=[]):
|
||||||
|
morfs = self.filter_by_prefix(morfs, omit_prefixes)
|
||||||
for morf in morfs:
|
for morf in morfs:
|
||||||
try:
|
try:
|
||||||
filename, statements, missing, _ = self.analysis(morf)
|
filename, statements, excluded, missing, _ = self.analysis2(morf)
|
||||||
|
self.annotate_file(filename, statements, excluded, missing, directory)
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
raise
|
||||||
|
except:
|
||||||
|
if not ignore_errors:
|
||||||
|
raise
|
||||||
|
|
||||||
|
def annotate_file(self, filename, statements, excluded, missing, directory=None):
|
||||||
source = open(filename, 'r')
|
source = open(filename, 'r')
|
||||||
if directory:
|
if directory:
|
||||||
dest_file = os.path.join(directory,
|
dest_file = os.path.join(directory,
|
||||||
@@ -471,8 +792,8 @@ class coverage:
|
|||||||
if self.blank_re.match(line):
|
if self.blank_re.match(line):
|
||||||
dest.write(' ')
|
dest.write(' ')
|
||||||
elif self.else_re.match(line):
|
elif self.else_re.match(line):
|
||||||
# Special logic for lines containing only
|
# Special logic for lines containing only 'else:'.
|
||||||
# 'else:'. See [GDR 2001-12-04b, 3.2].
|
# See [GDR 2001-12-04b, 3.2].
|
||||||
if i >= len(statements) and j >= len(missing):
|
if i >= len(statements) and j >= len(missing):
|
||||||
dest.write('! ')
|
dest.write('! ')
|
||||||
elif i >= len(statements) or j >= len(missing):
|
elif i >= len(statements) or j >= len(missing):
|
||||||
@@ -481,6 +802,8 @@ class coverage:
|
|||||||
dest.write('! ')
|
dest.write('! ')
|
||||||
else:
|
else:
|
||||||
dest.write('> ')
|
dest.write('> ')
|
||||||
|
elif lineno in excluded:
|
||||||
|
dest.write('- ')
|
||||||
elif covered:
|
elif covered:
|
||||||
dest.write('> ')
|
dest.write('> ')
|
||||||
else:
|
else:
|
||||||
@@ -488,22 +811,23 @@ class coverage:
|
|||||||
dest.write(line)
|
dest.write(line)
|
||||||
source.close()
|
source.close()
|
||||||
dest.close()
|
dest.close()
|
||||||
except KeyboardInterrupt:
|
|
||||||
raise
|
|
||||||
except:
|
|
||||||
if not ignore_errors:
|
|
||||||
raise
|
|
||||||
|
|
||||||
|
|
||||||
# Singleton object.
|
# Singleton object.
|
||||||
the_coverage = coverage()
|
the_coverage = coverage()
|
||||||
|
|
||||||
# Module functions call methods in the singleton object.
|
# Module functions call methods in the singleton object.
|
||||||
def start(*args): return apply(the_coverage.start, args)
|
def use_cache(*args, **kw): return the_coverage.use_cache(*args, **kw)
|
||||||
def stop(*args): return apply(the_coverage.stop, args)
|
def start(*args, **kw): return the_coverage.start(*args, **kw)
|
||||||
def erase(*args): return apply(the_coverage.erase, args)
|
def stop(*args, **kw): return the_coverage.stop(*args, **kw)
|
||||||
def analysis(*args): return apply(the_coverage.analysis, args)
|
def erase(*args, **kw): return the_coverage.erase(*args, **kw)
|
||||||
def report(*args): return apply(the_coverage.report, args)
|
def begin_recursive(*args, **kw): return the_coverage.begin_recursive(*args, **kw)
|
||||||
|
def end_recursive(*args, **kw): return the_coverage.end_recursive(*args, **kw)
|
||||||
|
def exclude(*args, **kw): return the_coverage.exclude(*args, **kw)
|
||||||
|
def analysis(*args, **kw): return the_coverage.analysis(*args, **kw)
|
||||||
|
def analysis2(*args, **kw): return the_coverage.analysis2(*args, **kw)
|
||||||
|
def report(*args, **kw): return the_coverage.report(*args, **kw)
|
||||||
|
def annotate(*args, **kw): return the_coverage.annotate(*args, **kw)
|
||||||
|
def annotate_file(*args, **kw): return the_coverage.annotate_file(*args, **kw)
|
||||||
|
|
||||||
# Save coverage data when Python exits. (The atexit module wasn't
|
# Save coverage data when Python exits. (The atexit module wasn't
|
||||||
# introduced until Python 2.0, so use sys.exitfunc when it's not
|
# introduced until Python 2.0, so use sys.exitfunc when it's not
|
||||||
@@ -516,18 +840,18 @@ except ImportError:
|
|||||||
|
|
||||||
# Command-line interface.
|
# Command-line interface.
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
the_coverage.command_line()
|
the_coverage.command_line(sys.argv[1:])
|
||||||
|
|
||||||
|
|
||||||
# A. REFERENCES
|
# A. REFERENCES
|
||||||
#
|
#
|
||||||
# [GDR 2001-12-04a] "Statement coverage for Python"; Gareth Rees;
|
# [GDR 2001-12-04a] "Statement coverage for Python"; Gareth Rees;
|
||||||
# Ravenbrook Limited; 2001-12-04;
|
# Ravenbrook Limited; 2001-12-04;
|
||||||
# <http://www.garethrees.org/2001/12/04/python-coverage/>.
|
# <http://www.nedbatchelder.com/code/modules/rees-coverage.html>.
|
||||||
#
|
#
|
||||||
# [GDR 2001-12-04b] "Statement coverage for Python: design and
|
# [GDR 2001-12-04b] "Statement coverage for Python: design and
|
||||||
# analysis"; Gareth Rees; Ravenbrook Limited; 2001-12-04;
|
# analysis"; Gareth Rees; Ravenbrook Limited; 2001-12-04;
|
||||||
# <http://www.garethrees.org/2001/12/04/python-coverage/design.html>.
|
# <http://www.nedbatchelder.com/code/modules/rees-design.html>.
|
||||||
#
|
#
|
||||||
# [van Rossum 2001-07-20a] "Python Reference Manual (releae 2.1.1)";
|
# [van Rossum 2001-07-20a] "Python Reference Manual (releae 2.1.1)";
|
||||||
# Guide van Rossum; 2001-07-20;
|
# Guide van Rossum; 2001-07-20;
|
||||||
@@ -561,10 +885,44 @@ if __name__ == '__main__':
|
|||||||
# so that it matches the value the program would get if it were run on
|
# so that it matches the value the program would get if it were run on
|
||||||
# its own.
|
# its own.
|
||||||
#
|
#
|
||||||
|
# 2004-12-12 NMB Significant code changes.
|
||||||
|
# - Finding executable statements has been rewritten so that docstrings and
|
||||||
|
# other quirks of Python execution aren't mistakenly identified as missing
|
||||||
|
# lines.
|
||||||
|
# - Lines can be excluded from consideration, even entire suites of lines.
|
||||||
|
# - The filesystem cache of covered lines can be disabled programmatically.
|
||||||
|
# - Modernized the code.
|
||||||
#
|
#
|
||||||
|
# 2004-12-14 NMB Minor tweaks. Return 'analysis' to its original behavior
|
||||||
|
# and add 'analysis2'. Add a global for 'annotate', and factor it, adding
|
||||||
|
# 'annotate_file'.
|
||||||
|
#
|
||||||
|
# 2004-12-31 NMB Allow for keyword arguments in the module global functions.
|
||||||
|
# Thanks, Allen.
|
||||||
|
#
|
||||||
|
# 2005-12-02 NMB Call threading.settrace so that all threads are measured.
|
||||||
|
# Thanks Martin Fuzzey. Add a file argument to report so that reports can be
|
||||||
|
# captured to a different destination.
|
||||||
|
#
|
||||||
|
# 2005-12-03 NMB coverage.py can now measure itself.
|
||||||
|
#
|
||||||
|
# 2005-12-04 NMB Adapted Greg Rogers' patch for using relative filenames,
|
||||||
|
# and sorting and omitting files to report on.
|
||||||
|
#
|
||||||
|
# 2006-07-23 NMB Applied Joseph Tate's patch for function decorators.
|
||||||
|
#
|
||||||
|
# 2006-08-21 NMB Applied Sigve Tjora and Mark van der Wal's fixes for argument
|
||||||
|
# handling.
|
||||||
|
#
|
||||||
|
# 2006-08-22 NMB Applied Geoff Bache's parallel mode patch.
|
||||||
|
#
|
||||||
|
# 2006-08-23 NMB Refactorings to improve testability. Fixes to command-line
|
||||||
|
# logic for parallel mode and collect.
|
||||||
|
|
||||||
# C. COPYRIGHT AND LICENCE
|
# C. COPYRIGHT AND LICENCE
|
||||||
#
|
#
|
||||||
# Copyright 2001 Gareth Rees. All rights reserved.
|
# Copyright 2001 Gareth Rees. All rights reserved.
|
||||||
|
# Copyright 2004-2006 Ned Batchelder. All rights reserved.
|
||||||
#
|
#
|
||||||
# Redistribution and use in source and binary forms, with or without
|
# Redistribution and use in source and binary forms, with or without
|
||||||
# modification, are permitted provided that the following conditions are
|
# modification, are permitted provided that the following conditions are
|
||||||
@@ -591,6 +949,4 @@ if __name__ == '__main__':
|
|||||||
# USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
|
# USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
|
||||||
# DAMAGE.
|
# DAMAGE.
|
||||||
#
|
#
|
||||||
#
|
# $Id: coverage.py 47 2006-08-24 01:08:48Z Ned $
|
||||||
#
|
|
||||||
# $Id: coverage.py,v 1.1 2005/07/11 20:39:31 dax Exp $
|
|
||||||
|
|||||||
84
run_test.py
84
run_test.py
@@ -1,10 +1,11 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
##
|
##
|
||||||
## run_test.py
|
## run_tests.py
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
## Login : David Rousselie <dax@happycoders.org>
|
||||||
## Started on Wed May 18 13:33:03 2005 David Rousselie
|
## Started on Wed Aug 9 21:37:35 2006 David Rousselie
|
||||||
## $Id: run_test.py,v 1.2 2005/09/18 20:24:07 David Rousselie Exp $
|
## $Id$
|
||||||
##
|
##
|
||||||
## Copyright (C) 2005 David Rousselie
|
## Copyright (C) 2006 David Rousselie
|
||||||
## This program is free software; you can redistribute it and/or modify
|
## 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
|
## it under the terms of the GNU General Public License as published by
|
||||||
## the Free Software Foundation; either version 2 of the License, or
|
## the Free Software Foundation; either version 2 of the License, or
|
||||||
@@ -23,7 +24,9 @@
|
|||||||
import coverage
|
import coverage
|
||||||
coverage.erase()
|
coverage.erase()
|
||||||
coverage.start()
|
coverage.start()
|
||||||
|
import logging
|
||||||
import unittest
|
import unittest
|
||||||
|
from test import test_support
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
sys.path.append("src")
|
sys.path.append("src")
|
||||||
@@ -32,67 +35,32 @@ sys.setdefaultencoding('utf8')
|
|||||||
del sys.setdefaultencoding
|
del sys.setdefaultencoding
|
||||||
|
|
||||||
import tests
|
import tests
|
||||||
from tests.test_mailconnection import *
|
from tests.jmc.model.test_account import *
|
||||||
from tests.test_mailconnection_factory import *
|
|
||||||
from tests.test_component import *
|
|
||||||
from tests.test_storage import *
|
|
||||||
from tests.test_lang import *
|
|
||||||
from test import test_support
|
|
||||||
import logging
|
|
||||||
import jmc
|
|
||||||
|
|
||||||
|
import jmc
|
||||||
|
import jmc.jabber
|
||||||
|
#import jmc.jabber.component
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
logger.addHandler(logging.StreamHandler())
|
logger.addHandler(logging.StreamHandler())
|
||||||
logger.setLevel(logging.INFO)
|
logger.setLevel(logging.INFO)
|
||||||
mail_connection_suite = unittest.makeSuite(MailConnection_TestCase, \
|
|
||||||
"test")
|
|
||||||
pop3_connection_suite = unittest.makeSuite(POP3Connection_TestCase, \
|
|
||||||
"test")
|
|
||||||
imap_connection_suite = unittest.makeSuite(IMAPConnection_TestCase, \
|
|
||||||
"test")
|
|
||||||
mc_factory_suite = unittest.makeSuite(MailConnectionFactory_TestCase, \
|
|
||||||
"test")
|
|
||||||
component_suite = unittest.makeSuite(MailComponent_TestCase_Basic, \
|
|
||||||
"test")
|
|
||||||
component2_suite = unittest.makeSuite(MailComponent_TestCase_NoReg, \
|
|
||||||
"test")
|
|
||||||
storage_suite = unittest.makeSuite(Storage_TestCase, \
|
|
||||||
"test")
|
|
||||||
dbmstorage_suite = unittest.makeSuite(DBMStorage_TestCase, \
|
|
||||||
"test")
|
|
||||||
sqlitestorage_suite = unittest.makeSuite(SQLiteStorage_TestCase, \
|
|
||||||
"test")
|
|
||||||
lang_suite = unittest.makeSuite(Lang_TestCase, \
|
|
||||||
"test")
|
|
||||||
|
|
||||||
jmc_suite = unittest.TestSuite((mail_connection_suite, \
|
mail_account_suite = unittest.makeSuite(MailAccount_TestCase, "test")
|
||||||
pop3_connection_suite, \
|
imap_account_suite = unittest.makeSuite(IMAPAccount_TestCase, "test")
|
||||||
imap_connection_suite, \
|
pop3_account_suite = unittest.makeSuite(POP3Account_TestCase, "test")
|
||||||
mc_factory_suite, \
|
|
||||||
# component_suite, \
|
jmc_suite = unittest.TestSuite()
|
||||||
# component2_suite, \
|
jmc_suite = unittest.TestSuite((mail_account_suite, \
|
||||||
storage_suite, \
|
imap_account_suite, \
|
||||||
dbmstorage_suite, \
|
pop3_account_suite))
|
||||||
sqlitestorage_suite, \
|
|
||||||
lang_suite))
|
|
||||||
#test_support.run_suite(mail_connection_suite)
|
|
||||||
#test_support.run_suite(pop3_connection_suite)
|
|
||||||
#test_support.run_suite(imap_connection_suite)
|
|
||||||
#test_support.run_suite(mc_factory_suite)
|
|
||||||
#test_support.run_suite(component_suite)
|
|
||||||
#test_support.run_suite(component2_suite)
|
|
||||||
#test_support.run_suite(storage_suite)
|
|
||||||
#test_support.run_suite(sqlitestorage_suite)
|
|
||||||
#test_support.run_suite(dbmstorage_suite)
|
|
||||||
test_support.run_suite(jmc_suite)
|
test_support.run_suite(jmc_suite)
|
||||||
|
|
||||||
|
|
||||||
coverage.stop()
|
coverage.stop()
|
||||||
coverage.analysis(jmc.email.mailconnection_factory)
|
#coverage.analysis(jmc.jabber.component)
|
||||||
coverage.analysis(jmc.email.mailconnection)
|
coverage.analysis(jmc.model.account)
|
||||||
coverage.analysis(jmc.jabber.component)
|
|
||||||
coverage.analysis(jmc.jabber.x)
|
coverage.report([
|
||||||
coverage.analysis(jmc.utils.lang)
|
jmc.model.account])
|
||||||
coverage.report([jmc.email.mailconnection_factory, jmc.email.mailconnection, \
|
|
||||||
jmc.jabber.component, jmc.jabber.x, jmc.utils.lang])
|
|
||||||
|
|||||||
85
src/jmc.py
85
src/jmc.py
@@ -1,12 +1,10 @@
|
|||||||
#!/usr/bin/python -u
|
|
||||||
##
|
##
|
||||||
## Jabber Mail Component
|
|
||||||
## jmc.py
|
## jmc.py
|
||||||
## Login : David Rousselie <david.rousselie@happycoders.org>
|
## Login : <dax@happycoders.org>
|
||||||
## Started on Fri Jan 7 11:06:42 2005
|
## Started on Fri Jan 19 18:14:41 2007 David Rousselie
|
||||||
## $Id: jmc.py,v 1.3 2005/07/11 20:39:31 dax Exp $
|
## $Id$
|
||||||
##
|
##
|
||||||
## Copyright (C) 2005
|
## Copyright (C) 2007 David Rousselie
|
||||||
## This program is free software; you can redistribute it and/or modify
|
## 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
|
## it under the terms of the GNU General Public License as published by
|
||||||
## the Free Software Foundation; either version 2 of the License, or
|
## the Free Software Foundation; either version 2 of the License, or
|
||||||
@@ -22,70 +20,27 @@
|
|||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||||
##
|
##
|
||||||
|
|
||||||
import sys
|
|
||||||
import os.path
|
|
||||||
import logging
|
import logging
|
||||||
|
import sys
|
||||||
|
|
||||||
reload(sys)
|
reload(sys)
|
||||||
sys.setdefaultencoding('utf-8')
|
sys.setdefaultencoding('utf-8')
|
||||||
del sys.setdefaultencoding
|
del sys.setdefaultencoding
|
||||||
from jmc.email import mailconnection
|
|
||||||
from jmc.jabber.component import MailComponent, ComponentFatalError
|
|
||||||
from jmc.utils.config import Config
|
|
||||||
|
|
||||||
def main(config_file = "jmc.xml", isDebug = 0):
|
from sqlobject import *
|
||||||
try:
|
from pyxmpp.message import Message
|
||||||
logger = logging.getLogger()
|
|
||||||
logger.addHandler(logging.StreamHandler())
|
|
||||||
if isDebug > 0:
|
|
||||||
logger.setLevel(logging.DEBUG)
|
|
||||||
try:
|
|
||||||
logger.debug("Loading config file " + config_file)
|
|
||||||
config = Config(config_file)
|
|
||||||
except:
|
|
||||||
print >>sys.stderr, "Couldn't load config file:", \
|
|
||||||
str(sys.exc_value)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
pidfile = open(config.get_content("config/pidfile"), "w")
|
from jmc.jabber.component import MailComponent
|
||||||
pidfile.write(str(os.getpid()))
|
from jmc.model.account import MailPresenceAccount
|
||||||
pidfile.close()
|
|
||||||
mailconnection.default_encoding = config.get_content("config/mail_default_encoding")
|
|
||||||
print "creating component..."
|
|
||||||
mailcomp = MailComponent(config.get_content("config/jabber/service"), \
|
|
||||||
config.get_content("config/jabber/secret"), \
|
|
||||||
config.get_content("config/jabber/server"), \
|
|
||||||
int(config.get_content("config/jabber/port")), \
|
|
||||||
config.get_content("config/jabber/language"), \
|
|
||||||
int(config.get_content("config/check_interval")), \
|
|
||||||
config.get_content("config/spooldir"), \
|
|
||||||
config.get_content("config/storage"), \
|
|
||||||
config.get_content("config/jabber/vCard/FN"))
|
|
||||||
|
|
||||||
print "starting..."
|
|
||||||
mailcomp.run(1)
|
|
||||||
os.remove(config.get_content("config/pidfile"))
|
|
||||||
except ComponentFatalError,e:
|
|
||||||
print e
|
|
||||||
print "Aborting."
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
var_option = 0
|
|
||||||
file_num = 0
|
|
||||||
index = 0
|
|
||||||
debug_level = 0
|
|
||||||
for opt in sys.argv:
|
|
||||||
if var_option == 0 and len(opt) == 2 and opt == "-c":
|
|
||||||
var_option += 1
|
|
||||||
elif (var_option & 1) == 1 and len(opt) > 0:
|
|
||||||
var_option += 1
|
|
||||||
file_num = index
|
|
||||||
if len(opt) == 2 and opt == "-D":
|
|
||||||
debug_level = 1
|
|
||||||
index += 1
|
|
||||||
if (var_option & 2) == 2:
|
|
||||||
main(sys.argv[file_num], debug_level)
|
|
||||||
else:
|
|
||||||
main("/etc/jabber/jmc.xml", debug_level)
|
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.addHandler(logging.StreamHandler())
|
||||||
|
logger.setLevel(logging.DEBUG)
|
||||||
|
component = MailComponent("jmc.localhost", \
|
||||||
|
"secret", \
|
||||||
|
"127.0.0.1", \
|
||||||
|
5349, \
|
||||||
|
"sqlite:///tmp/jmc_test.db")
|
||||||
|
component.account_class = MailPresenceAccount
|
||||||
|
component.run()
|
||||||
|
logger.debug("JMC is exiting")
|
||||||
|
|||||||
@@ -0,0 +1,2 @@
|
|||||||
|
"""JMC module"""
|
||||||
|
__revision__ = ""
|
||||||
|
|||||||
@@ -1,142 +0,0 @@
|
|||||||
##
|
|
||||||
## mailconnection_factory.py
|
|
||||||
## Login : David Rousselie <david.rousselie@happycoders.org>
|
|
||||||
## Started on Fri May 20 10:41:46 2005 David Rousselie
|
|
||||||
## $Id: mailconnection_factory.py,v 1.2 2005/09/18 20:24:07 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import jmc.email.mailconnection as mailconnection
|
|
||||||
from jmc.email.mailconnection import IMAPConnection, POP3Connection
|
|
||||||
|
|
||||||
def get_new_mail_connection(type):
|
|
||||||
""" Static method to return an empty MailConnection object of given type
|
|
||||||
:Parameters:
|
|
||||||
- 'type': type of connection to return : 'imap', 'imaps', 'pop3', 'pop3s'
|
|
||||||
|
|
||||||
:return: MailConnection of given type in parameter, None if unknown type
|
|
||||||
|
|
||||||
:returntype: 'MailConnection'
|
|
||||||
"""
|
|
||||||
if type == "imap":
|
|
||||||
return IMAPConnection()
|
|
||||||
elif type == "imaps":
|
|
||||||
return IMAPConnection(ssl = True)
|
|
||||||
elif type == "pop3":
|
|
||||||
return POP3Connection()
|
|
||||||
elif type == "pop3s":
|
|
||||||
return POP3Connection(ssl = True)
|
|
||||||
raise Exception, "Connection type \"" + type + "\" unknown"
|
|
||||||
|
|
||||||
def str_to_mail_connection(connection_string):
|
|
||||||
""" Static methode to create a MailConnection object filled from string
|
|
||||||
|
|
||||||
:Parameters:
|
|
||||||
- 'connection_string': string containing MailConnection parameters separated
|
|
||||||
by '#'. ex: 'pop3#login#password#host#110#chat_action#online_action#away_action#xa_action#dnd_action#offline_action#check_interval#liv_email_only(#Mailbox)'
|
|
||||||
|
|
||||||
:Types:
|
|
||||||
- 'connection_string': string
|
|
||||||
|
|
||||||
:return: MailConnection of given type found in string parameter
|
|
||||||
|
|
||||||
:returntype: 'MailConnection'
|
|
||||||
"""
|
|
||||||
arg_list = connection_string.split("#")
|
|
||||||
# optionals values must be the at the beginning of the list to pop them
|
|
||||||
# last
|
|
||||||
arg_list.reverse()
|
|
||||||
type = arg_list.pop()
|
|
||||||
login = arg_list.pop()
|
|
||||||
password = arg_list.pop()
|
|
||||||
if password == "/\\":
|
|
||||||
password = None
|
|
||||||
store_password = False
|
|
||||||
else:
|
|
||||||
store_password = True
|
|
||||||
host = arg_list.pop()
|
|
||||||
port = int(arg_list.pop())
|
|
||||||
chat_action = None
|
|
||||||
online_action = None
|
|
||||||
away_action = None
|
|
||||||
xa_action = None
|
|
||||||
dnd_action = None
|
|
||||||
offline_action = None
|
|
||||||
interval = None
|
|
||||||
live_email_only = False
|
|
||||||
result = None
|
|
||||||
if type[0:4] == "imap":
|
|
||||||
if len(arg_list) == 9:
|
|
||||||
chat_action = int(arg_list.pop())
|
|
||||||
online_action = int(arg_list.pop())
|
|
||||||
away_action = int(arg_list.pop())
|
|
||||||
xa_action = int(arg_list.pop())
|
|
||||||
dnd_action = int(arg_list.pop())
|
|
||||||
offline_action = int(arg_list.pop())
|
|
||||||
interval = int(arg_list.pop())
|
|
||||||
live_email_only = (arg_list.pop().lower() == "true")
|
|
||||||
else:
|
|
||||||
retrieve = bool(arg_list.pop() == "True")
|
|
||||||
if retrieve:
|
|
||||||
chat_action = online_action = away_action = xa_action = dnd_action = mailconnection.RETRIEVE
|
|
||||||
else:
|
|
||||||
chat_action = online_action = away_action = xa_action = dnd_action = mailconnection.DIGEST
|
|
||||||
offline_action = mailconnection.DO_NOTHING
|
|
||||||
mailbox = arg_list.pop()
|
|
||||||
result = IMAPConnection(login = login, \
|
|
||||||
password = password, \
|
|
||||||
host = host, \
|
|
||||||
ssl = (len(type) == 5), \
|
|
||||||
port = port, \
|
|
||||||
mailbox = mailbox)
|
|
||||||
elif type[0:4] == "pop3":
|
|
||||||
if len(arg_list) == 8:
|
|
||||||
chat_action = int(arg_list.pop())
|
|
||||||
online_action = int(arg_list.pop())
|
|
||||||
away_action = int(arg_list.pop())
|
|
||||||
xa_action = int(arg_list.pop())
|
|
||||||
dnd_action = int(arg_list.pop())
|
|
||||||
offline_action = int(arg_list.pop())
|
|
||||||
interval = int(arg_list.pop())
|
|
||||||
live_email_only = (arg_list.pop().lower() == "true")
|
|
||||||
else:
|
|
||||||
retrieve = bool(arg_list.pop() == "True")
|
|
||||||
if retrieve:
|
|
||||||
chat_action = online_action = away_action = xa_action = dnd_action = mailconnection.RETRIEVE
|
|
||||||
else:
|
|
||||||
chat_action = online_action = away_action = xa_action = dnd_action = mailconnection.DIGEST
|
|
||||||
offline_action = mailconnection.DO_NOTHING
|
|
||||||
result = POP3Connection(login = login, \
|
|
||||||
password = password, \
|
|
||||||
host = host, \
|
|
||||||
port = port, \
|
|
||||||
ssl = (len(type) == 5))
|
|
||||||
if result is None:
|
|
||||||
raise Exception, "Connection type \"" + type + "\" unknown"
|
|
||||||
result.store_password = store_password
|
|
||||||
result.chat_action = chat_action
|
|
||||||
result.online_action = online_action
|
|
||||||
result.away_action = away_action
|
|
||||||
result.xa_action = xa_action
|
|
||||||
result.dnd_action = dnd_action
|
|
||||||
result.offline_action = offline_action
|
|
||||||
if interval is not None:
|
|
||||||
result.interval = interval
|
|
||||||
result.live_email_only = live_email_only
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
"""Jabber component classes"""
|
||||||
|
__revision__ = ""
|
||||||
|
|||||||
@@ -1,129 +0,0 @@
|
|||||||
##
|
|
||||||
## x.py
|
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
|
||||||
## Started on Fri Jan 7 11:06:42 2005
|
|
||||||
## $Id: x.py,v 1.3 2005/09/18 20:24:07 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import sys
|
|
||||||
from pyxmpp.stanza import common_doc
|
|
||||||
|
|
||||||
class Option(object):
|
|
||||||
def __init__(self, label, value):
|
|
||||||
self.__label = label
|
|
||||||
self.__value = value
|
|
||||||
|
|
||||||
def get_xml(self, parent):
|
|
||||||
if parent is None:
|
|
||||||
option = common_doc.newChild(None, "option", None)
|
|
||||||
else:
|
|
||||||
option = parent.newChild(None, "option", None)
|
|
||||||
option.setProp("label", self.__label)
|
|
||||||
option.newChild(None, "value", self.__value)
|
|
||||||
return option
|
|
||||||
|
|
||||||
class Field(object):
|
|
||||||
def __init__(self, type, label, var, value):
|
|
||||||
self.__type = type
|
|
||||||
self.__label = label
|
|
||||||
self.__var = var
|
|
||||||
self.value = value
|
|
||||||
self.__options = []
|
|
||||||
|
|
||||||
def add_option(self, label, value):
|
|
||||||
option = Option(label, value)
|
|
||||||
self.__options.append(option)
|
|
||||||
return option
|
|
||||||
|
|
||||||
def get_xml(self, parent):
|
|
||||||
if parent is None:
|
|
||||||
raise Exception, "parent field should not be None"
|
|
||||||
else:
|
|
||||||
field = parent.newChild(None, "field", None)
|
|
||||||
field.setProp("type", self.__type)
|
|
||||||
if not self.__label is None:
|
|
||||||
field.setProp("label", self.__label)
|
|
||||||
if not self.__var is None:
|
|
||||||
field.setProp("var", self.__var)
|
|
||||||
if self.value:
|
|
||||||
field.newChild(None, "value", self.value)
|
|
||||||
for option in self.__options:
|
|
||||||
option.get_xml(field)
|
|
||||||
return field
|
|
||||||
|
|
||||||
class X(object):
|
|
||||||
def __init__(self):
|
|
||||||
self.fields = {}
|
|
||||||
self.fields_tab = []
|
|
||||||
self.title = None
|
|
||||||
self.instructions = None
|
|
||||||
self.type = None
|
|
||||||
self.xmlns = None
|
|
||||||
|
|
||||||
def add_field(self, type = "fixed", label = None, var = None, value = ""):
|
|
||||||
field = Field(type, label, var, value)
|
|
||||||
self.fields[var] = field
|
|
||||||
# fields_tab exist to keep added fields order
|
|
||||||
self.fields_tab.append(field)
|
|
||||||
return field
|
|
||||||
|
|
||||||
def attach_xml(self, iq):
|
|
||||||
node = iq.newChild(None, "x", None)
|
|
||||||
_ns = node.newNs(self.xmlns, None)
|
|
||||||
node.setNs(_ns)
|
|
||||||
if not self.title is None:
|
|
||||||
node.newTextChild(None, "title", self.title)
|
|
||||||
if not self.instructions is None:
|
|
||||||
node.newTextChild(None, "instructions", self.instructions)
|
|
||||||
for field in self.fields_tab:
|
|
||||||
field.get_xml(node)
|
|
||||||
return node
|
|
||||||
|
|
||||||
def from_xml(self, node):
|
|
||||||
## TODO : test node type and ns and clean that loop !!!!
|
|
||||||
while node and node.type != "element":
|
|
||||||
node = node.next
|
|
||||||
child = node.children
|
|
||||||
while child:
|
|
||||||
## TODO : test child type (element) and ns (jabber:x:data)
|
|
||||||
if child.type == "element" and child.name == "field":
|
|
||||||
if child.hasProp("type"):
|
|
||||||
type = child.prop("type")
|
|
||||||
else:
|
|
||||||
type = ""
|
|
||||||
|
|
||||||
if child.hasProp("label"):
|
|
||||||
label = child.prop("label")
|
|
||||||
else:
|
|
||||||
label = ""
|
|
||||||
|
|
||||||
if child.hasProp("var"):
|
|
||||||
var = child.prop("var")
|
|
||||||
else:
|
|
||||||
var = ""
|
|
||||||
|
|
||||||
xval = child.children
|
|
||||||
while xval and xval.name != "value":
|
|
||||||
xval = xval.next
|
|
||||||
if xval:
|
|
||||||
value = xval.getContent()
|
|
||||||
else:
|
|
||||||
value = ""
|
|
||||||
field = Field(type, label, var, value)
|
|
||||||
self.fields[var] = field
|
|
||||||
child = child.next
|
|
||||||
2
src/jmc/model/__init__.py
Normal file
2
src/jmc/model/__init__.py
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
"""Contains data model classes"""
|
||||||
|
__revision__ = ""
|
||||||
@@ -1,10 +1,10 @@
|
|||||||
##
|
##
|
||||||
## mailconnection.py
|
## account.py
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
## Login : <dax@happycoders.org>
|
||||||
## Started on Fri Jan 7 11:06:42 2005
|
## Started on Fri Jan 19 18:21:44 2007 David Rousselie
|
||||||
## $Id: mailconnection.py,v 1.11 2005/09/18 20:24:07 dax Exp $
|
## $Id$
|
||||||
##
|
##
|
||||||
## Copyright (C) 2005
|
## Copyright (C) 2007 David Rousselie
|
||||||
## This program is free software; you can redistribute it and/or modify
|
## 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
|
## it under the terms of the GNU General Public License as published by
|
||||||
## the Free Software Foundation; either version 2 of the License, or
|
## the Free Software Foundation; either version 2 of the License, or
|
||||||
@@ -20,7 +20,6 @@
|
|||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||||
##
|
##
|
||||||
|
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
import logging
|
import logging
|
||||||
import email
|
import email
|
||||||
@@ -31,16 +30,15 @@ import poplib
|
|||||||
import imaplib
|
import imaplib
|
||||||
import socket
|
import socket
|
||||||
|
|
||||||
from jmc.utils.lang import Lang
|
from sqlobject.inheritance import InheritableSQLObject
|
||||||
|
from sqlobject.col import StringCol, IntCol, BoolCol
|
||||||
|
|
||||||
|
from jcl.model.account import PresenceAccount
|
||||||
|
from jmc.lang import Lang
|
||||||
|
|
||||||
IMAP4_TIMEOUT = 10
|
IMAP4_TIMEOUT = 10
|
||||||
POP3_TIMEOUT = 10
|
POP3_TIMEOUT = 10
|
||||||
|
|
||||||
DO_NOTHING = 0
|
|
||||||
DIGEST = 1
|
|
||||||
RETRIEVE = 2
|
|
||||||
default_encoding = "iso-8859-1"
|
|
||||||
|
|
||||||
## All MY* classes are implemented to add a timeout (settimeout)
|
## All MY* classes are implemented to add a timeout (settimeout)
|
||||||
## while connecting
|
## while connecting
|
||||||
class MYIMAP4(imaplib.IMAP4):
|
class MYIMAP4(imaplib.IMAP4):
|
||||||
@@ -127,76 +125,83 @@ class MYPOP3_SSL(poplib.POP3_SSL):
|
|||||||
self._debugging = 0
|
self._debugging = 0
|
||||||
self.welcome = self._getresp()
|
self.welcome = self._getresp()
|
||||||
|
|
||||||
class MailConnection(object):
|
class MailAccount(PresenceAccount):
|
||||||
""" Wrapper to mail connection and action.
|
""" Wrapper to mail connection and action.
|
||||||
Abstract class, do not represent real mail connection type"""
|
Abstract class, do not represent real mail connection type"""
|
||||||
|
|
||||||
_logger = logging.getLogger("jmc.MailConnection")
|
# Define constants
|
||||||
|
DIGEST = 1
|
||||||
|
RETRIEVE = 2
|
||||||
|
default_encoding = "iso-8859-1"
|
||||||
|
possibles_actions = [PresenceAccount.DO_NOTHING, \
|
||||||
|
DIGEST, \
|
||||||
|
RETRIEVE]
|
||||||
|
|
||||||
def __init__(self, login = "", password = "", host = "", \
|
login = StringCol(default = "")
|
||||||
port = 110, ssl = False):
|
password = StringCol(default = None)
|
||||||
""" Initialize MailConnection object for common parameters of all
|
host = StringCol(default = "localhost")
|
||||||
connections types
|
port = IntCol(default = 110)
|
||||||
|
ssl = BoolCol(default = False)
|
||||||
|
interval = IntCol(default = 5)
|
||||||
|
store_password = BoolCol(default = True)
|
||||||
|
live_email_only = BoolCol(default = False)
|
||||||
|
|
||||||
:Parameters:
|
lastcheck = IntCol(default = 0)
|
||||||
- 'login': login used to connect mail server
|
waiting_password_reply = BoolCol(default = False)
|
||||||
- 'password': password associated with 'login' to connect mail server
|
in_error = BoolCol(default = False)
|
||||||
- 'host': mail server hostname
|
first_check = BoolCol(default = True)
|
||||||
- 'port': mail server port
|
|
||||||
- 'ssl': activate ssl to connect server
|
|
||||||
|
|
||||||
:Types:
|
def _init(self, *args, **kw):
|
||||||
- 'login': string
|
"""MailAccount init
|
||||||
- 'password': string
|
Initialize class attributes"""
|
||||||
- 'host': string
|
InheritableSQLObject._init(self, *args, **kw)
|
||||||
- 'port': int
|
self.__logger = logging.getLogger("jmc.model.account.MailAccount")
|
||||||
- 'ssl': boolean"""
|
|
||||||
self.login = login
|
|
||||||
self.password = password
|
|
||||||
self.store_password = True
|
|
||||||
self.host = host
|
|
||||||
self.port = port
|
|
||||||
self.ssl = ssl
|
|
||||||
self.status = "offline"
|
|
||||||
self.connection = None
|
self.connection = None
|
||||||
self.chat_action = RETRIEVE
|
self.default_lang_class = Lang.en # TODO: use String
|
||||||
self.online_action = RETRIEVE
|
|
||||||
self.away_action = RETRIEVE
|
|
||||||
self.xa_action = RETRIEVE
|
|
||||||
self.dnd_action = RETRIEVE
|
|
||||||
self.offline_action = DO_NOTHING
|
|
||||||
self.interval = 5
|
|
||||||
self.lastcheck = 0
|
|
||||||
self.default_lang_class = Lang.en
|
|
||||||
self.waiting_password_reply = False
|
|
||||||
self.in_error = False
|
|
||||||
self.live_email_only = False
|
|
||||||
self.first_check = True
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
def _get_register_fields(cls):
|
||||||
return self.get_type() == other.get_type() \
|
"""See Account._get_register_fields
|
||||||
and self.login == other.login \
|
"""
|
||||||
and (not self.store_password or self.password == other.password) \
|
def password_post_func(password):
|
||||||
and self.store_password == other.store_password \
|
if password is None or password == "":
|
||||||
and self.host == other.host \
|
return None
|
||||||
and self.port == other.port \
|
return password
|
||||||
and self.ssl == other.ssl \
|
|
||||||
and self.chat_action == other.chat_action \
|
|
||||||
and self.online_action == other.online_action \
|
|
||||||
and self.away_action == other.away_action \
|
|
||||||
and self.xa_action == other.xa_action \
|
|
||||||
and self.dnd_action == other.dnd_action \
|
|
||||||
and self.offline_action == other.offline_action \
|
|
||||||
and self.interval == other.interval \
|
|
||||||
and self.live_email_only == other.live_email_only
|
|
||||||
|
|
||||||
def __str__(self):
|
return Account.get_register_fields() + \
|
||||||
return self.get_type() + "#" + self.login + "#" + \
|
[("login", "text-single", None, account.string_not_null_post_func, \
|
||||||
(self.store_password and self.password or "/\\") + "#" \
|
account.mandatory_field), \
|
||||||
+ self.host + "#" + str(self.port) + "#" + str(self.chat_action) + "#" \
|
("password", "text-private", None, password_post_func, \
|
||||||
+ str(self.online_action) + "#" + str(self.away_action) + "#" + \
|
(lambda field_name: None)), \
|
||||||
str(self.xa_action) + "#" + str(self.dnd_action) + "#" + \
|
("host", "text-single", None, account.string_not_null_post_func, \
|
||||||
str(self.offline_action) + "#" + str(self.interval) + "#" + str(self.live_email_only)
|
account.mandatory_field), \
|
||||||
|
("port", "text-single", None, account.int_post_func, \
|
||||||
|
account.mandatory_field), \
|
||||||
|
("ssl", "boolean", None, account.default_post_func, \
|
||||||
|
(lambda field_name: None)), \
|
||||||
|
("store_password", "boolean", None, account.default_post_func, \
|
||||||
|
(lambda field_name: True)), \
|
||||||
|
("live_email_only", "boolean", None, account.default_post_func, \
|
||||||
|
lambda field_name: False)]
|
||||||
|
|
||||||
|
get_register_fields = classmethod(_get_register_fields)
|
||||||
|
|
||||||
|
def _get_presence_actions_fields(cls):
|
||||||
|
"""See PresenceAccount._get_presence_actions_fields
|
||||||
|
"""
|
||||||
|
return {'chat_action': (cls.possibles_actions, \
|
||||||
|
RETRIEVE), \
|
||||||
|
'online_action': (cls.possibles_actions, \
|
||||||
|
RETRIEVE), \
|
||||||
|
'away_action': (cls.possibles_actions, \
|
||||||
|
RETRIEVE), \
|
||||||
|
'xa_action': (cls.possibles_actions, \
|
||||||
|
RETRIEVE), \
|
||||||
|
'dnd_action': (cls.possibles_actions, \
|
||||||
|
RETRIEVE), \
|
||||||
|
'offline_action': (cls.possibles_actions, \
|
||||||
|
PresenceAccount.DO_NOTHING)}
|
||||||
|
|
||||||
|
get_presence_actions_fields = classmethod(_get_presence_actions_fields)
|
||||||
|
|
||||||
def get_decoded_part(self, part, charset_hint):
|
def get_decoded_part(self, part, charset_hint):
|
||||||
content_charset = part.get_content_charset()
|
content_charset = part.get_content_charset()
|
||||||
@@ -292,26 +297,26 @@ class MailConnection(object):
|
|||||||
unicode(self.port)
|
unicode(self.port)
|
||||||
|
|
||||||
def connect(self):
|
def connect(self):
|
||||||
pass
|
raise NotImplementedError
|
||||||
|
|
||||||
def disconnect(self):
|
def disconnect(self):
|
||||||
pass
|
raise NotImplementedError
|
||||||
|
|
||||||
def get_mail_list(self):
|
def get_mail_list(self):
|
||||||
return 0
|
raise NotImplementedError
|
||||||
|
|
||||||
def get_mail(self, index):
|
def get_mail(self, index):
|
||||||
return None
|
raise NotImplementedError
|
||||||
|
|
||||||
def get_mail_summary(self, index):
|
def get_mail_summary(self, index):
|
||||||
return None
|
raise NotImplementedError
|
||||||
|
|
||||||
def get_next_mail_index(self, mail_list):
|
def get_next_mail_index(self, mail_list):
|
||||||
pass
|
raise NotImplementedError
|
||||||
|
|
||||||
# Does not modify server state but just internal JMC state
|
# Does not modify server state but just internal JMC state
|
||||||
def mark_all_as_read(self):
|
def mark_all_as_read(self):
|
||||||
pass
|
raise NotImplementedError
|
||||||
|
|
||||||
def get_action(self):
|
def get_action(self):
|
||||||
mapping = {"online": self.online_action,
|
mapping = {"online": self.online_action,
|
||||||
@@ -322,29 +327,35 @@ class MailConnection(object):
|
|||||||
"offline": self.offline_action}
|
"offline": self.offline_action}
|
||||||
if mapping.has_key(self.status):
|
if mapping.has_key(self.status):
|
||||||
return mapping[self.status]
|
return mapping[self.status]
|
||||||
return DO_NOTHING
|
return PresenceAccount.DO_NOTHING
|
||||||
|
|
||||||
action = property(get_action)
|
action = property(get_action)
|
||||||
|
|
||||||
class IMAPConnection(MailConnection):
|
class IMAPAccount(MailAccount):
|
||||||
_logger = logging.getLogger("jmc.IMAPConnection")
|
mailbox = StringCol(default = "INBOX") # TODO : set default INBOX in reg_form (use get_register_fields last field ?)
|
||||||
|
|
||||||
def __init__(self, login = "", password = "", host = "", \
|
def _get_register_fields(cls):
|
||||||
port = None, ssl = False, mailbox = "INBOX"):
|
"""See Account._get_register_fields
|
||||||
if not port:
|
"""
|
||||||
if ssl:
|
def password_post_func(password):
|
||||||
port = 993
|
if password is None or password == "":
|
||||||
else:
|
return None
|
||||||
port = 143
|
return password
|
||||||
MailConnection.__init__(self, login, password, host, port, ssl)
|
|
||||||
self.mailbox = mailbox
|
|
||||||
|
|
||||||
def __eq__(self, other):
|
return MailAccount.get_register_fields() + \
|
||||||
return MailConnection.__eq__(self, other) \
|
[("mailbox", "text-single", None, account.string_not_null_post_func, \
|
||||||
and self.mailbox == other.mailbox
|
(lambda field_name: "INBOX")), \
|
||||||
|
("password", "text-private", None, password_post_func, \
|
||||||
|
(lambda field_name: None)), \
|
||||||
|
("store_password", "boolean", None, account.boolean_post_func, \
|
||||||
|
lambda field_name: True)]
|
||||||
|
|
||||||
def __str__(self):
|
get_register_fields = classmethod(_get_register_fields)
|
||||||
return MailConnection.__str__(self) + "#" + self.mailbox
|
|
||||||
|
|
||||||
|
def _init(self, *args, **kw):
|
||||||
|
MailAccount._init(self, *args, **kw)
|
||||||
|
self.__logger = logging.getLogger("jmc.IMAPConnection")
|
||||||
|
|
||||||
def get_type(self):
|
def get_type(self):
|
||||||
if self.ssl:
|
if self.ssl:
|
||||||
@@ -352,10 +363,10 @@ class IMAPConnection(MailConnection):
|
|||||||
return "imap"
|
return "imap"
|
||||||
|
|
||||||
def get_status(self):
|
def get_status(self):
|
||||||
return MailConnection.get_status(self) + "/" + self.mailbox
|
return MailAccount.get_status(self) + "/" + self.mailbox
|
||||||
|
|
||||||
def connect(self):
|
def connect(self):
|
||||||
IMAPConnection._logger.debug("Connecting to IMAP server " \
|
self.__logger.debug("Connecting to IMAP server " \
|
||||||
+ self.login + "@" + self.host + ":" + str(self.port) \
|
+ self.login + "@" + self.host + ":" + str(self.port) \
|
||||||
+ " (" + self.mailbox + "). SSL=" \
|
+ " (" + self.mailbox + "). SSL=" \
|
||||||
+ str(self.ssl))
|
+ str(self.ssl))
|
||||||
@@ -366,12 +377,12 @@ class IMAPConnection(MailConnection):
|
|||||||
self.connection.login(self.login, self.password)
|
self.connection.login(self.login, self.password)
|
||||||
|
|
||||||
def disconnect(self):
|
def disconnect(self):
|
||||||
IMAPConnection._logger.debug("Disconnecting from IMAP server " \
|
self.__logger.debug("Disconnecting from IMAP server " \
|
||||||
+ self.host)
|
+ self.host)
|
||||||
self.connection.logout()
|
self.connection.logout()
|
||||||
|
|
||||||
def get_mail_list(self):
|
def get_mail_list(self):
|
||||||
IMAPConnection._logger.debug("Getting mail list")
|
self.__logger.debug("Getting mail list")
|
||||||
typ, data = self.connection.select(self.mailbox)
|
typ, data = self.connection.select(self.mailbox)
|
||||||
typ, data = self.connection.search(None, 'RECENT')
|
typ, data = self.connection.search(None, 'RECENT')
|
||||||
if typ == 'OK':
|
if typ == 'OK':
|
||||||
@@ -379,7 +390,7 @@ class IMAPConnection(MailConnection):
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
def get_mail(self, index):
|
def get_mail(self, index):
|
||||||
IMAPConnection._logger.debug("Getting mail " + str(index))
|
self.__logger.debug("Getting mail " + str(index))
|
||||||
typ, data = self.connection.select(self.mailbox, True)
|
typ, data = self.connection.select(self.mailbox, True)
|
||||||
typ, data = self.connection.fetch(index, '(RFC822)')
|
typ, data = self.connection.fetch(index, '(RFC822)')
|
||||||
if typ == 'OK':
|
if typ == 'OK':
|
||||||
@@ -387,7 +398,7 @@ class IMAPConnection(MailConnection):
|
|||||||
return u"Error while fetching mail " + str(index)
|
return u"Error while fetching mail " + str(index)
|
||||||
|
|
||||||
def get_mail_summary(self, index):
|
def get_mail_summary(self, index):
|
||||||
IMAPConnection._logger.debug("Getting mail summary " + str(index))
|
self.__logger.debug("Getting mail summary " + str(index))
|
||||||
typ, data = self.connection.select(self.mailbox, True)
|
typ, data = self.connection.select(self.mailbox, True)
|
||||||
typ, data = self.connection.fetch(index, '(RFC822)')
|
typ, data = self.connection.fetch(index, '(RFC822)')
|
||||||
if typ == 'OK':
|
if typ == 'OK':
|
||||||
@@ -404,27 +415,23 @@ class IMAPConnection(MailConnection):
|
|||||||
|
|
||||||
type = property(get_type)
|
type = property(get_type)
|
||||||
|
|
||||||
class POP3Connection(MailConnection):
|
class POP3Account(MailAccount):
|
||||||
_logger = logging.getLogger("jmc.POP3Connection")
|
nb_mail = IntCol(default = 0)
|
||||||
|
lastmail = IntCol(default = 0)
|
||||||
|
|
||||||
def __init__(self, login = "", password = "", host = "", \
|
def _init(self, *args, **kw):
|
||||||
port = None, ssl = False):
|
MailAccount._init(self, *args, **kw)
|
||||||
if not port:
|
self.__logger = logging.getLogger("jmc.model.account.POP3Account")
|
||||||
if ssl:
|
|
||||||
port = 995
|
|
||||||
else:
|
|
||||||
port = 110
|
|
||||||
MailConnection.__init__(self, login, password, host, port, ssl)
|
|
||||||
self.__nb_mail = 0
|
|
||||||
self.__lastmail = 0
|
|
||||||
|
|
||||||
def get_type(self):
|
def get_type(self):
|
||||||
if self.ssl:
|
if self.ssl:
|
||||||
return "pop3s"
|
return "pop3s"
|
||||||
return "pop3"
|
return "pop3"
|
||||||
|
|
||||||
|
type = property(get_type)
|
||||||
|
|
||||||
def connect(self):
|
def connect(self):
|
||||||
POP3Connection._logger.debug("Connecting to POP3 server " \
|
self.__logger.debug("Connecting to POP3 server " \
|
||||||
+ self.login + "@" + self.host + ":" + str(self.port)\
|
+ self.login + "@" + self.host + ":" + str(self.port)\
|
||||||
+ ". SSL=" + str(self.ssl))
|
+ ". SSL=" + str(self.ssl))
|
||||||
if self.ssl:
|
if self.ssl:
|
||||||
@@ -439,18 +446,18 @@ class POP3Connection(MailConnection):
|
|||||||
|
|
||||||
|
|
||||||
def disconnect(self):
|
def disconnect(self):
|
||||||
POP3Connection._logger.debug("Disconnecting from POP3 server " \
|
self.__logger.debug("Disconnecting from POP3 server " \
|
||||||
+ self.host)
|
+ self.host)
|
||||||
self.connection.quit()
|
self.connection.quit()
|
||||||
|
|
||||||
def get_mail_list(self):
|
def get_mail_list(self):
|
||||||
POP3Connection._logger.debug("Getting mail list")
|
self.__logger.debug("Getting mail list")
|
||||||
count, size = self.connection.stat()
|
count, size = self.connection.stat()
|
||||||
self.__nb_mail = count
|
self.nb_mail = count
|
||||||
return [str(i) for i in range(1, count + 1)]
|
return [str(i) for i in range(1, count + 1)]
|
||||||
|
|
||||||
def get_mail(self, index):
|
def get_mail(self, index):
|
||||||
POP3Connection._logger.debug("Getting mail " + str(index))
|
self.__logger.debug("Getting mail " + str(index))
|
||||||
ret, data, size = self.connection.retr(index)
|
ret, data, size = self.connection.retr(index)
|
||||||
try:
|
try:
|
||||||
self.connection.rset()
|
self.connection.rset()
|
||||||
@@ -461,7 +468,7 @@ class POP3Connection(MailConnection):
|
|||||||
return u"Error while fetching mail " + str(index)
|
return u"Error while fetching mail " + str(index)
|
||||||
|
|
||||||
def get_mail_summary(self, index):
|
def get_mail_summary(self, index):
|
||||||
POP3Connection._logger.debug("Getting mail summary " + str(index))
|
self.__logger.debug("Getting mail summary " + str(index))
|
||||||
ret, data, size = self.connection.retr(index)
|
ret, data, size = self.connection.retr(index)
|
||||||
try:
|
try:
|
||||||
self.connection.rset()
|
self.connection.rset()
|
||||||
@@ -472,16 +479,15 @@ class POP3Connection(MailConnection):
|
|||||||
return u"Error while fetching mail " + str(index)
|
return u"Error while fetching mail " + str(index)
|
||||||
|
|
||||||
def get_next_mail_index(self, mail_list):
|
def get_next_mail_index(self, mail_list):
|
||||||
if self.__nb_mail == self.__lastmail:
|
if self.nb_mail == self.lastmail:
|
||||||
return None
|
return None
|
||||||
if self.__nb_mail < self.__lastmail:
|
if self.nb_mail < self.lastmail:
|
||||||
self.__lastmail = 0
|
self.lastmail = 0
|
||||||
result = int(mail_list[self.__lastmail])
|
result = int(mail_list[self.lastmail])
|
||||||
self.__lastmail += 1
|
self.lastmail += 1
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def mark_all_as_read(self):
|
def mark_all_as_read(self):
|
||||||
self.get_mail_list()
|
self.get_mail_list()
|
||||||
self.__lastmail = self.__nb_mail
|
self.lastmail = self.nb_mail
|
||||||
|
|
||||||
type = property(get_type)
|
|
||||||
@@ -1,297 +0,0 @@
|
|||||||
##
|
|
||||||
## storage.py
|
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
|
||||||
## Started on Wed Jul 20 20:26:53 2005 dax
|
|
||||||
## $Id: storage.py,v 1.1 2005/09/18 20:24:07 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005 dax
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import os
|
|
||||||
import re
|
|
||||||
import os.path
|
|
||||||
import sys
|
|
||||||
import anydbm
|
|
||||||
import logging
|
|
||||||
from UserDict import UserDict
|
|
||||||
|
|
||||||
import jmc.email.mailconnection_factory as mailconnection_factory
|
|
||||||
|
|
||||||
|
|
||||||
class Storage(UserDict):
|
|
||||||
def __init__(self, nb_pk_fields = 1, spool_dir = ".", db_file = None):
|
|
||||||
UserDict.__init__(self)
|
|
||||||
self.nb_pk_fields = nb_pk_fields
|
|
||||||
if db_file is None:
|
|
||||||
self._spool_dir = ""
|
|
||||||
self.set_spool_dir(spool_dir)
|
|
||||||
self.file = self._spool_dir + "/registered.db"
|
|
||||||
else:
|
|
||||||
spool_dir = os.path.dirname(db_file) or "."
|
|
||||||
self.set_spool_dir(spool_dir)
|
|
||||||
self.file = db_file
|
|
||||||
self._registered = self.load()
|
|
||||||
|
|
||||||
def __setitem__(self, pk_tuple, obj):
|
|
||||||
# print "Adding " + "#".join(map(str, pk_tuple)) + " = " + str(obj)
|
|
||||||
self._registered[str("#".join(map(str, pk_tuple)))] = obj
|
|
||||||
|
|
||||||
def __getitem__(self, pk_tuple):
|
|
||||||
# print "Getting " + "#".join(map(str, pk_tuple))
|
|
||||||
if len(pk_tuple) == self.nb_pk_fields:
|
|
||||||
return self._registered[str("#".join(map(str, pk_tuple)))]
|
|
||||||
else:
|
|
||||||
partial_key = str("#".join(map(str, pk_tuple)))
|
|
||||||
regexp = re.compile(partial_key)
|
|
||||||
return [self._registered[key]
|
|
||||||
for key in self._registered.keys()
|
|
||||||
if regexp.search(key)]
|
|
||||||
|
|
||||||
def __delitem__(self, pk_tuple):
|
|
||||||
#print "Deleting " + "#".join(map(str, pk_tuple))
|
|
||||||
del self._registered[str("#".join(map(str, pk_tuple)))]
|
|
||||||
|
|
||||||
def get_spool_dir(self):
|
|
||||||
return self._spool_dir
|
|
||||||
|
|
||||||
def set_spool_dir(self, spool_dir):
|
|
||||||
self._spool_dir = spool_dir
|
|
||||||
if not os.path.isdir(self._spool_dir):
|
|
||||||
os.makedirs(self._spool_dir)
|
|
||||||
|
|
||||||
spool_dir = property(get_spool_dir, set_spool_dir)
|
|
||||||
|
|
||||||
def has_key(self, pk_tuple):
|
|
||||||
if len(pk_tuple) == self.nb_pk_fields:
|
|
||||||
return self._registered.has_key(str("#".join(map(str, pk_tuple))))
|
|
||||||
else:
|
|
||||||
partial_key = str("#".join(map(str, pk_tuple)))
|
|
||||||
regexp = re.compile("^" + partial_key)
|
|
||||||
for key in self._registered.keys():
|
|
||||||
if regexp.search(key):
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
def keys(self, pk_tuple = None):
|
|
||||||
if pk_tuple is None:
|
|
||||||
return [tuple(key.split("#")) for key in self._registered.keys()]
|
|
||||||
else:
|
|
||||||
level = len(pk_tuple)
|
|
||||||
partial_key = str("#".join(map(str, pk_tuple)))
|
|
||||||
regexp = re.compile("^" + partial_key)
|
|
||||||
result = {}
|
|
||||||
for key in self._registered.keys():
|
|
||||||
if regexp.search(key):
|
|
||||||
result[key.split("#")[level]] = None
|
|
||||||
return result.keys()
|
|
||||||
|
|
||||||
def dump(self):
|
|
||||||
for pk in self._registered.keys():
|
|
||||||
print pk + " = " + str(self._registered[pk])
|
|
||||||
|
|
||||||
def load(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
class DBMStorage(Storage):
|
|
||||||
_logger = logging.getLogger("jmc.utils.DBMStorage")
|
|
||||||
|
|
||||||
def __init__(self, nb_pk_fields = 1, spool_dir = ".", db_file = None):
|
|
||||||
# print "DBM INIT"
|
|
||||||
Storage.__init__(self, nb_pk_fields, spool_dir, db_file)
|
|
||||||
|
|
||||||
def __del__(self):
|
|
||||||
# print "DBM STOP"
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def load(self):
|
|
||||||
str_registered = anydbm.open(self.file, \
|
|
||||||
'c')
|
|
||||||
result = {}
|
|
||||||
try:
|
|
||||||
for pk in str_registered.keys():
|
|
||||||
result[pk] = mailconnection_factory.str_to_mail_connection(str_registered[pk])
|
|
||||||
except Exception, e:
|
|
||||||
print >>sys.stderr, "Cannot load registered.db : "
|
|
||||||
print >>sys.stderr, e
|
|
||||||
str_registered.close()
|
|
||||||
return result
|
|
||||||
|
|
||||||
def sync(self):
|
|
||||||
#print "DBM SYNC"
|
|
||||||
self.store()
|
|
||||||
|
|
||||||
def __store(self, nb_pk_fields, registered, pk):
|
|
||||||
if nb_pk_fields > 0:
|
|
||||||
for key in registered.keys():
|
|
||||||
if pk:
|
|
||||||
self.__store(nb_pk_fields - 1, \
|
|
||||||
registered[key], pk + "#" + key)
|
|
||||||
else:
|
|
||||||
self.__store(nb_pk_fields - 1, \
|
|
||||||
registered[key], key)
|
|
||||||
else:
|
|
||||||
self.__str_registered[pk] = str(registered)
|
|
||||||
# print "STORING : " + pk + " = " + str(registered)
|
|
||||||
|
|
||||||
def store(self):
|
|
||||||
# print "DBM STORE"
|
|
||||||
try:
|
|
||||||
str_registered = anydbm.open(self.file, \
|
|
||||||
'n')
|
|
||||||
for pk in self._registered.keys():
|
|
||||||
str_registered[pk] = str(self._registered[pk])
|
|
||||||
except Exception, e:
|
|
||||||
print >>sys.stderr, "Cannot save to registered.db : "
|
|
||||||
print >>sys.stderr, e
|
|
||||||
str_registered.close()
|
|
||||||
|
|
||||||
def __setitem__(self, pk_tuple, obj):
|
|
||||||
Storage.__setitem__(self, pk_tuple, obj)
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def __delitem__(self, pk_tuple):
|
|
||||||
Storage.__delitem__(self, pk_tuple)
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
# Do not fail if pysqlite is not installed
|
|
||||||
try:
|
|
||||||
from pysqlite2 import dbapi2 as sqlite
|
|
||||||
|
|
||||||
class SQLiteStorage(Storage):
|
|
||||||
_logger = logging.getLogger("jmc.utils.SQLiteStorage")
|
|
||||||
|
|
||||||
def __init__(self, nb_pk_fields = 1, spool_dir = ".", db_file = None):
|
|
||||||
self.__connection = None
|
|
||||||
Storage.__init__(self, nb_pk_fields, spool_dir, db_file)
|
|
||||||
|
|
||||||
def create(self):
|
|
||||||
SQLiteStorage._logger.debug("creating new Table")
|
|
||||||
cursor = self.__connection.cursor()
|
|
||||||
cursor.execute("""
|
|
||||||
create table account(
|
|
||||||
jid STRING,
|
|
||||||
name STRING,
|
|
||||||
type STRING,
|
|
||||||
login STRING,
|
|
||||||
password STRING,
|
|
||||||
host STRING,
|
|
||||||
port INTEGER,
|
|
||||||
chat_action INTEGER,
|
|
||||||
online_action INTEGER,
|
|
||||||
away_action INTEGER,
|
|
||||||
xa_action INTEGER,
|
|
||||||
dnd_action INTEGER,
|
|
||||||
offline_action INTEGER,
|
|
||||||
interval INTEGER,
|
|
||||||
live_email_only BOOLEAN,
|
|
||||||
mailbox STRING,
|
|
||||||
PRIMARY KEY(jid, name)
|
|
||||||
)
|
|
||||||
""")
|
|
||||||
self.__connection.commit()
|
|
||||||
cursor.close()
|
|
||||||
|
|
||||||
def __del__(self):
|
|
||||||
self.__connection.close()
|
|
||||||
|
|
||||||
def sync(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def load(self):
|
|
||||||
if not os.path.exists(self.file):
|
|
||||||
self.__connection = sqlite.connect(self.file)
|
|
||||||
self.create()
|
|
||||||
else:
|
|
||||||
self.__connection = sqlite.connect(self.file)
|
|
||||||
cursor = self.__connection.cursor()
|
|
||||||
cursor.execute("""select * from account""")
|
|
||||||
result = {}
|
|
||||||
for row in cursor.fetchall():
|
|
||||||
# print "Creating new " + row[self.nb_pk_fields] + " connection."
|
|
||||||
account_type = row[self.nb_pk_fields]
|
|
||||||
account = result["#".join(row[0:self.nb_pk_fields])] = mailconnection_factory.get_new_mail_connection(account_type)
|
|
||||||
account.login = row[self.nb_pk_fields + 1]
|
|
||||||
account.password = row[self.nb_pk_fields + 2]
|
|
||||||
if account.password is None:
|
|
||||||
account.store_password = False
|
|
||||||
else:
|
|
||||||
account.store_password = True
|
|
||||||
account.host = row[self.nb_pk_fields + 3]
|
|
||||||
account.port = int(row[self.nb_pk_fields + 4])
|
|
||||||
account.chat_action = int(row[self.nb_pk_fields + 5])
|
|
||||||
account.online_action = int(row[self.nb_pk_fields + 6])
|
|
||||||
account.away_action = int(row[self.nb_pk_fields + 7])
|
|
||||||
account.xa_action = int(row[self.nb_pk_fields + 8])
|
|
||||||
account.dnd_action = int(row[self.nb_pk_fields + 9])
|
|
||||||
account.offline_action = int(row[self.nb_pk_fields + 10])
|
|
||||||
account.interval = int(row[self.nb_pk_fields + 11])
|
|
||||||
account.live_email_only = (row[self.nb_pk_fields + 12] == 1)
|
|
||||||
if account_type[0:4] == "imap":
|
|
||||||
account.mailbox = row[self.nb_pk_fields + 13]
|
|
||||||
# for field_index in range(self.nb_pk_fields + 1, len(row)):
|
|
||||||
# print "\tSetting " + str(cursor.description[field_index][0]) + \
|
|
||||||
# " to " + str(row[field_index])
|
|
||||||
# setattr(account,
|
|
||||||
# cursor.description[field_index][0],
|
|
||||||
# row[field_index])
|
|
||||||
cursor.close()
|
|
||||||
return result
|
|
||||||
|
|
||||||
def __setitem__(self, pk_tuple, obj):
|
|
||||||
Storage.__setitem__(self, pk_tuple, obj)
|
|
||||||
cursor = self.__connection.cursor()
|
|
||||||
mailbox = None
|
|
||||||
password = None
|
|
||||||
if obj.type[0:4] == "imap":
|
|
||||||
mailbox = obj.mailbox
|
|
||||||
if obj.store_password == True:
|
|
||||||
password = obj.password
|
|
||||||
cursor.execute("""
|
|
||||||
insert or replace into account values
|
|
||||||
(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
||||||
""",
|
|
||||||
(pk_tuple[0],
|
|
||||||
pk_tuple[1],
|
|
||||||
obj.type,
|
|
||||||
obj.login,
|
|
||||||
password,
|
|
||||||
obj.host,
|
|
||||||
obj.port,
|
|
||||||
obj.chat_action,
|
|
||||||
obj.online_action,
|
|
||||||
obj.away_action,
|
|
||||||
obj.xa_action,
|
|
||||||
obj.dnd_action,
|
|
||||||
obj.offline_action,
|
|
||||||
obj.interval,
|
|
||||||
obj.live_email_only,
|
|
||||||
mailbox))
|
|
||||||
self.__connection.commit()
|
|
||||||
cursor.close()
|
|
||||||
|
|
||||||
def __delitem__(self, pk_tuple):
|
|
||||||
Storage.__delitem__(self, pk_tuple)
|
|
||||||
cursor = self.__connection.cursor()
|
|
||||||
cursor.execute("""
|
|
||||||
delete from account where jid = ? and name = ?
|
|
||||||
""",
|
|
||||||
(pk_tuple[0],
|
|
||||||
pk_tuple[1]))
|
|
||||||
self.__connection.commit()
|
|
||||||
cursor.close()
|
|
||||||
except ImportError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
@@ -1,20 +0,0 @@
|
|||||||
<config>
|
|
||||||
<jabber>
|
|
||||||
<server>127.0.0.1</server>
|
|
||||||
<port>5347</port>
|
|
||||||
<secret>secret</secret>
|
|
||||||
<service>jmc.localhost</service>
|
|
||||||
<connectsleep>5</connectsleep>
|
|
||||||
<language>en</language>
|
|
||||||
<vCard>
|
|
||||||
<FN>Jabber Mail Component</FN>
|
|
||||||
<DESC>A Jabber mail server component</DESC>
|
|
||||||
<URL>http://people.happycoders.org/dax/jabber/jmc/</URL>
|
|
||||||
</vCard>
|
|
||||||
</jabber>
|
|
||||||
<storage>SQLite</storage>
|
|
||||||
<pidfile>jmc.pid</pidfile>
|
|
||||||
<spooldir>.</spooldir>
|
|
||||||
<check_interval>5</check_interval> <!-- in minutes -->
|
|
||||||
<mail_default_encoding>iso-8859-15</mail_default_encoding>
|
|
||||||
</config>
|
|
||||||
2
tests/jmc/__init__.py
Normal file
2
tests/jmc/__init__.py
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
"""JMC test module"""
|
||||||
|
__revision__ = ""
|
||||||
@@ -29,9 +29,28 @@ import socket
|
|||||||
import types
|
import types
|
||||||
import select
|
import select
|
||||||
import xml.dom.minidom
|
import xml.dom.minidom
|
||||||
import utils
|
|
||||||
from pyxmpp import xmlextra
|
from pyxmpp import xmlextra
|
||||||
|
|
||||||
|
def xmldiff(node1, node2):
|
||||||
|
if node1.nodeType == node1.TEXT_NODE:
|
||||||
|
if not node2.nodeType == node2.TEXT_NODE \
|
||||||
|
or re.compile(node2.data + "$").match(node1.data) is None:
|
||||||
|
raise Exception("data in text node " + node1.data + " does not match " + node2.data)
|
||||||
|
elif node1.nodeType == node1.DOCUMENT_NODE:
|
||||||
|
if not node2.nodeType == node2.DOCUMENT_NODE:
|
||||||
|
raise Exception("node1 is Document but not node2 (" + node2.nodeType + ")")
|
||||||
|
elif node1.tagName != node2.tagName:
|
||||||
|
raise Exception("Different tag name : " + node1.tagName + " != " + node2.tagName)
|
||||||
|
else:
|
||||||
|
for attr in node1._get_attributes().keys():
|
||||||
|
if not node2.hasAttribute(attr) \
|
||||||
|
or node1.getAttribute(attr) != node2.getAttribute(attr):
|
||||||
|
raise Exception("(" + node1.tagName + ") Different attributes : " + node1.getAttribute(attr) + " != " + node2.getAttribute(attr))
|
||||||
|
if len(node1.childNodes) != len(node2.childNodes):
|
||||||
|
raise Exception("(" + node1.tagName + ") Different children number : " + str(len(node1.childNodes)) + " != " + str(len(node2.childNodes)))
|
||||||
|
for i in range(len(node1.childNodes)):
|
||||||
|
xmldiff(node1.childNodes[i], node2.childNodes[i])
|
||||||
|
|
||||||
class DummyServer:
|
class DummyServer:
|
||||||
def __init__(self, host, port, responses = None):
|
def __init__(self, host, port, responses = None):
|
||||||
for res in socket.getaddrinfo(host, port, socket.AF_UNSPEC, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
|
for res in socket.getaddrinfo(host, port, socket.AF_UNSPEC, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
|
||||||
@@ -40,7 +59,7 @@ class DummyServer:
|
|||||||
s = socket.socket(af, socktype, proto)
|
s = socket.socket(af, socktype, proto)
|
||||||
except socket.error, msg:
|
except socket.error, msg:
|
||||||
s = None
|
s = None
|
||||||
continue
|
raise socket.error
|
||||||
try:
|
try:
|
||||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||||
s.bind(sa)
|
s.bind(sa)
|
||||||
@@ -48,7 +67,7 @@ class DummyServer:
|
|||||||
except socket.error, msg:
|
except socket.error, msg:
|
||||||
s.close()
|
s.close()
|
||||||
s = None
|
s = None
|
||||||
continue
|
raise socket.error
|
||||||
break
|
break
|
||||||
self.socket = s
|
self.socket = s
|
||||||
self.responses = None
|
self.responses = None
|
||||||
@@ -56,6 +75,7 @@ class DummyServer:
|
|||||||
self.real_queries = []
|
self.real_queries = []
|
||||||
|
|
||||||
def serve(self):
|
def serve(self):
|
||||||
|
conn = None
|
||||||
try:
|
try:
|
||||||
conn, addr = self.socket.accept()
|
conn, addr = self.socket.accept()
|
||||||
rfile = conn.makefile('rb', -1)
|
rfile = conn.makefile('rb', -1)
|
||||||
@@ -77,7 +97,9 @@ class DummyServer:
|
|||||||
else:
|
else:
|
||||||
self.real_queries.append(data)
|
self.real_queries.append(data)
|
||||||
#print >>sys.stderr, 'Receive : ', data
|
#print >>sys.stderr, 'Receive : ', data
|
||||||
|
if conn is not None:
|
||||||
conn.close()
|
conn.close()
|
||||||
|
if self.socket is not None:
|
||||||
self.socket.close()
|
self.socket.close()
|
||||||
self.socket = None
|
self.socket = None
|
||||||
except:
|
except:
|
||||||
2
tests/jmc/jabber/__init__.py
Normal file
2
tests/jmc/jabber/__init__.py
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
"""JMC jabber test module"""
|
||||||
|
__revision__ = ""
|
||||||
@@ -1,10 +1,10 @@
|
|||||||
##
|
##
|
||||||
## release.py
|
## test_component.py
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
## Login : <dax@happycoders.org>
|
||||||
## Started on Mon Jul 24 22:37:00 2006 dax
|
## Started on Wed Feb 14 18:04:49 2007 David Rousselie
|
||||||
## $Id$
|
## $Id$
|
||||||
##
|
##
|
||||||
## Copyright (C) 2006 dax
|
## Copyright (C) 2007 David Rousselie
|
||||||
## This program is free software; you can redistribute it and/or modify
|
## 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
|
## it under the terms of the GNU General Public License as published by
|
||||||
## the Free Software Foundation; either version 2 of the License, or
|
## the Free Software Foundation; either version 2 of the License, or
|
||||||
@@ -20,14 +20,7 @@
|
|||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||||
##
|
##
|
||||||
|
|
||||||
version = "0.2.2"
|
import unittest
|
||||||
author = "David Rousselie"
|
|
||||||
email = "dax@happycoders.org"
|
|
||||||
license = "GPL"
|
|
||||||
long_description = """Jabber Mail Component
|
|
||||||
|
|
||||||
JMC is a jabber service to check email from POP3 and IMAP4 server and retrieve
|
class MailComponent_TestCase(unittest.TestCase):
|
||||||
them or just a notification of new emails. Jabber users can register multiple
|
pass
|
||||||
email accounts.
|
|
||||||
|
|
||||||
"""
|
|
||||||
@@ -1,11 +1,11 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
##
|
##
|
||||||
## mailconnection_test.py
|
## test_account.py
|
||||||
## Login : David Rousselie <david.rousselie@happycoders.org>
|
## Login : <dax@happycoders.org>
|
||||||
## Started on Fri May 13 11:32:51 2005 David Rousselie
|
## Started on Wed Feb 14 08:23:17 2007 David Rousselie
|
||||||
## $Id: test_mailconnection.py,v 1.2 2005/09/18 20:24:07 David Rousselie Exp $
|
## $Id$
|
||||||
##
|
##
|
||||||
## Copyright (C) 2005 David Rousselie
|
## Copyright (C) 2007 David Rousselie
|
||||||
## This program is free software; you can redistribute it and/or modify
|
## 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
|
## it under the terms of the GNU General Public License as published by
|
||||||
## the Free Software Foundation; either version 2 of the License, or
|
## the Free Software Foundation; either version 2 of the License, or
|
||||||
@@ -22,19 +22,44 @@
|
|||||||
##
|
##
|
||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
from jmc.email.mailconnection import IMAPConnection, \
|
import os
|
||||||
POP3Connection, \
|
|
||||||
MailConnection
|
|
||||||
import dummy_server
|
|
||||||
import email_generator
|
|
||||||
import thread
|
import thread
|
||||||
import re
|
|
||||||
import sys
|
|
||||||
import string
|
|
||||||
|
|
||||||
class MailConnection_TestCase(unittest.TestCase):
|
from sqlobject import *
|
||||||
|
from sqlobject.dbconnection import TheURIOpener
|
||||||
|
|
||||||
|
from jcl.model import account
|
||||||
|
from jcl.model.account import Account, PresenceAccount
|
||||||
|
from jmc.model.account import MailAccount, POP3Account, IMAPAccount
|
||||||
|
|
||||||
|
from tests.jmc import email_generator, dummy_server
|
||||||
|
|
||||||
|
DB_PATH = "/tmp/jmc_test.db"
|
||||||
|
DB_URL = DB_PATH # + "?debug=1&debugThreading=1"
|
||||||
|
|
||||||
|
class MailAccount_TestCase(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.connection = MailConnection()
|
if os.path.exists(DB_PATH):
|
||||||
|
os.unlink(DB_PATH)
|
||||||
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
|
Account.createTable(ifNotExists = True)
|
||||||
|
PresenceAccount.createTable(ifNotExists = True)
|
||||||
|
MailAccount.createTable(ifNotExists = True)
|
||||||
|
self.account = MailAccount(user_jid = "user1@test.com", \
|
||||||
|
name = "account1", \
|
||||||
|
jid = "account1@jmc.test.com")
|
||||||
|
del account.hub.threadConnection
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
|
MailAccount.dropTable(ifExists = True)
|
||||||
|
PresenceAccount.dropTable(ifExists = True)
|
||||||
|
Account.dropTable(ifExists = True)
|
||||||
|
del TheURIOpener.cachedURIs['sqlite://' + DB_URL]
|
||||||
|
account.hub.threadConnection.close()
|
||||||
|
del account.hub.threadConnection
|
||||||
|
if os.path.exists(DB_PATH):
|
||||||
|
os.unlink(DB_PATH)
|
||||||
|
|
||||||
def make_test(email_type, tested_func, expected_res):
|
def make_test(email_type, tested_func, expected_res):
|
||||||
def inner(self):
|
def inner(self):
|
||||||
@@ -48,23 +73,23 @@ class MailConnection_TestCase(unittest.TestCase):
|
|||||||
|
|
||||||
test_get_decoded_part_not_encoded = \
|
test_get_decoded_part_not_encoded = \
|
||||||
make_test((False, False, False), \
|
make_test((False, False, False), \
|
||||||
lambda self, email: self.connection.get_decoded_part(email, None), \
|
lambda self, email: self.account.get_decoded_part(email, None), \
|
||||||
u"Not encoded single part")
|
u"Not encoded single part")
|
||||||
|
|
||||||
test_get_decoded_part_encoded = \
|
test_get_decoded_part_encoded = \
|
||||||
make_test((True, False, False), \
|
make_test((True, False, False), \
|
||||||
lambda self, email: self.connection.get_decoded_part(email, None), \
|
lambda self, email: self.account.get_decoded_part(email, None), \
|
||||||
u"Encoded single part with 'iso-8859-15' charset (éàê)")
|
u"Encoded single part with 'iso-8859-15' charset (éàê)")
|
||||||
|
|
||||||
test_format_message_summary_not_encoded = \
|
test_format_message_summary_not_encoded = \
|
||||||
make_test((False, False, True), \
|
make_test((False, False, True), \
|
||||||
lambda self, email: self.connection.format_message_summary(email), \
|
lambda self, email: self.account.format_message_summary(email), \
|
||||||
(u"From : not encoded from\nSubject : not encoded subject\n\n", \
|
(u"From : not encoded from\nSubject : not encoded subject\n\n", \
|
||||||
u"not encoded from"))
|
u"not encoded from"))
|
||||||
|
|
||||||
test_format_message_summary_encoded = \
|
test_format_message_summary_encoded = \
|
||||||
make_test((True, False, True), \
|
make_test((True, False, True), \
|
||||||
lambda self, email: self.connection.format_message_summary(email), \
|
lambda self, email: self.account.format_message_summary(email), \
|
||||||
(u"From : encoded from (éàê)\nSubject : encoded subject " + \
|
(u"From : encoded from (éàê)\nSubject : encoded subject " + \
|
||||||
u"(éàê)\n\n", \
|
u"(éàê)\n\n", \
|
||||||
u"encoded from (éàê)"))
|
u"encoded from (éàê)"))
|
||||||
@@ -78,21 +103,21 @@ class MailConnection_TestCase(unittest.TestCase):
|
|||||||
email.replace_header("From", \
|
email.replace_header("From", \
|
||||||
"\"" + str(email["From"]) \
|
"\"" + str(email["From"]) \
|
||||||
+ "\" not encoded part") or \
|
+ "\" not encoded part") or \
|
||||||
self.connection.format_message_summary(email), \
|
self.account.format_message_summary(email), \
|
||||||
(u"From : \"encoded from (éàê)\" not encoded part\nSubject " + \
|
(u"From : \"encoded from (éàê)\" not encoded part\nSubject " + \
|
||||||
u": \"encoded subject (éàê)\" not encoded part\n\n", \
|
u": \"encoded subject (éàê)\" not encoded part\n\n", \
|
||||||
u"\"encoded from (éàê)\" not encoded part"))
|
u"\"encoded from (éàê)\" not encoded part"))
|
||||||
|
|
||||||
test_format_message_single_not_encoded = \
|
test_format_message_single_not_encoded = \
|
||||||
make_test((False, False, True), \
|
make_test((False, False, True), \
|
||||||
lambda self, email: self.connection.format_message(email), \
|
lambda self, email: self.account.format_message(email), \
|
||||||
(u"From : not encoded from\nSubject : not encoded subject" + \
|
(u"From : not encoded from\nSubject : not encoded subject" + \
|
||||||
u"\n\nNot encoded single part\n", \
|
u"\n\nNot encoded single part\n", \
|
||||||
u"not encoded from"))
|
u"not encoded from"))
|
||||||
|
|
||||||
test_format_message_single_encoded = \
|
test_format_message_single_encoded = \
|
||||||
make_test((True, False, True), \
|
make_test((True, False, True), \
|
||||||
lambda self, email: self.connection.format_message(email), \
|
lambda self, email: self.account.format_message(email), \
|
||||||
(u"From : encoded from (éàê)\nSubject : encoded subject " + \
|
(u"From : encoded from (éàê)\nSubject : encoded subject " + \
|
||||||
u"(éàê)\n\nEncoded single part with 'iso-8859-15' charset" + \
|
u"(éàê)\n\nEncoded single part with 'iso-8859-15' charset" + \
|
||||||
u" (éàê)\n", \
|
u" (éàê)\n", \
|
||||||
@@ -100,14 +125,14 @@ class MailConnection_TestCase(unittest.TestCase):
|
|||||||
|
|
||||||
test_format_message_multi_not_encoded = \
|
test_format_message_multi_not_encoded = \
|
||||||
make_test((False, True, True), \
|
make_test((False, True, True), \
|
||||||
lambda self, email: self.connection.format_message(email), \
|
lambda self, email: self.account.format_message(email), \
|
||||||
(u"From : not encoded from\nSubject : not encoded subject" + \
|
(u"From : not encoded from\nSubject : not encoded subject" + \
|
||||||
u"\n\nNot encoded multipart1\nNot encoded multipart2\n", \
|
u"\n\nNot encoded multipart1\nNot encoded multipart2\n", \
|
||||||
u"not encoded from"))
|
u"not encoded from"))
|
||||||
|
|
||||||
test_format_message_multi_encoded = \
|
test_format_message_multi_encoded = \
|
||||||
make_test((True, True, True), \
|
make_test((True, True, True), \
|
||||||
lambda self, email: self.connection.format_message(email), \
|
lambda self, email: self.account.format_message(email), \
|
||||||
(u"From : encoded from (éàê)\nSubject : encoded subject (éà" + \
|
(u"From : encoded from (éàê)\nSubject : encoded subject (éà" + \
|
||||||
u"ê)\n\nutf-8 multipart1 with no charset (éàê)" + \
|
u"ê)\n\nutf-8 multipart1 with no charset (éàê)" + \
|
||||||
u"\nEncoded multipart2 with 'iso-8859-15' charset (éàê)\n" + \
|
u"\nEncoded multipart2 with 'iso-8859-15' charset (éàê)\n" + \
|
||||||
@@ -115,19 +140,40 @@ class MailConnection_TestCase(unittest.TestCase):
|
|||||||
u"encoded from (éàê)"))
|
u"encoded from (éàê)"))
|
||||||
|
|
||||||
|
|
||||||
class POP3Connection_TestCase(unittest.TestCase):
|
class POP3Account_TestCase(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.server = dummy_server.DummyServer("localhost", 1110)
|
self.server = dummy_server.DummyServer("localhost", 1110)
|
||||||
thread.start_new_thread(self.server.serve, ())
|
thread.start_new_thread(self.server.serve, ())
|
||||||
self.pop3connection = POP3Connection("login", \
|
if os.path.exists(DB_PATH):
|
||||||
"pass", \
|
os.unlink(DB_PATH)
|
||||||
"localhost", \
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
1110, \
|
Account.createTable(ifNotExists = True)
|
||||||
ssl = False)
|
PresenceAccount.createTable(ifNotExists = True)
|
||||||
|
MailAccount.createTable(ifNotExists = True)
|
||||||
|
POP3Account.createTable(ifNotExists = True)
|
||||||
|
self.pop3_account = POP3Account(user_jid = "user1@test.com", \
|
||||||
|
name = "account1", \
|
||||||
|
jid = "account1@jmc.test.com", \
|
||||||
|
login = "login")
|
||||||
|
self.pop3_account.password = "pass"
|
||||||
|
self.pop3_account.host = "localhost"
|
||||||
|
self.pop3_account.port = 1110
|
||||||
|
self.pop3_account.ssl = False
|
||||||
|
del account.hub.threadConnection
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
|
POP3Account.dropTable(ifExists = True)
|
||||||
|
MailAccount.dropTable(ifExists = True)
|
||||||
|
PresenceAccount.dropTable(ifExists = True)
|
||||||
|
Account.dropTable(ifExists = True)
|
||||||
|
del TheURIOpener.cachedURIs['sqlite://' + DB_URL]
|
||||||
|
account.hub.threadConnection.close()
|
||||||
|
del account.hub.threadConnection
|
||||||
|
if os.path.exists(DB_PATH):
|
||||||
|
os.unlink(DB_PATH)
|
||||||
self.server = None
|
self.server = None
|
||||||
self.pop3connection = None
|
self.pop3_account = None
|
||||||
|
|
||||||
def make_test(responses = None, queries = None, core = None):
|
def make_test(responses = None, queries = None, core = None):
|
||||||
def inner(self):
|
def inner(self):
|
||||||
@@ -141,12 +187,14 @@ class POP3Connection_TestCase(unittest.TestCase):
|
|||||||
if queries:
|
if queries:
|
||||||
self.server.queries += queries
|
self.server.queries += queries
|
||||||
self.server.queries += ["QUIT\r\n"]
|
self.server.queries += ["QUIT\r\n"]
|
||||||
self.pop3connection.connect()
|
self.pop3_account.connect()
|
||||||
self.failUnless(self.pop3connection.connection, \
|
self.failUnless(self.pop3_account.connection, \
|
||||||
"Cannot establish connection")
|
"Cannot establish connection")
|
||||||
if core:
|
if core:
|
||||||
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
core(self)
|
core(self)
|
||||||
self.pop3connection.disconnect()
|
del account.hub.threadConnection
|
||||||
|
self.pop3_account.disconnect()
|
||||||
self.failUnless(self.server.verify_queries(), \
|
self.failUnless(self.server.verify_queries(), \
|
||||||
"Sended queries does not match expected queries.")
|
"Sended queries does not match expected queries.")
|
||||||
return inner
|
return inner
|
||||||
@@ -157,7 +205,7 @@ class POP3Connection_TestCase(unittest.TestCase):
|
|||||||
make_test(["+OK 2 20\r\n"], \
|
make_test(["+OK 2 20\r\n"], \
|
||||||
["STAT\r\n"], \
|
["STAT\r\n"], \
|
||||||
lambda self: \
|
lambda self: \
|
||||||
self.assertEquals(self.pop3connection.get_mail_list(), \
|
self.assertEquals(self.pop3_account.get_mail_list(), \
|
||||||
["1", "2"]))
|
["1", "2"]))
|
||||||
|
|
||||||
test_get_mail_summary = \
|
test_get_mail_summary = \
|
||||||
@@ -169,7 +217,7 @@ class POP3Connection_TestCase(unittest.TestCase):
|
|||||||
["RETR 1\r\n",
|
["RETR 1\r\n",
|
||||||
"RSET\r\n"], \
|
"RSET\r\n"], \
|
||||||
lambda self: \
|
lambda self: \
|
||||||
self.assertEquals(self.pop3connection.get_mail_summary(1), \
|
self.assertEquals(self.pop3_account.get_mail_summary(1), \
|
||||||
(u"From : user@test.com\n" + \
|
(u"From : user@test.com\n" + \
|
||||||
u"Subject : subject test\n\n", \
|
u"Subject : subject test\n\n", \
|
||||||
u"user@test.com")))
|
u"user@test.com")))
|
||||||
@@ -183,7 +231,7 @@ class POP3Connection_TestCase(unittest.TestCase):
|
|||||||
["RETR 1\r\n",
|
["RETR 1\r\n",
|
||||||
"RSET\r\n"], \
|
"RSET\r\n"], \
|
||||||
lambda self: \
|
lambda self: \
|
||||||
self.assertEquals(self.pop3connection.get_mail(1), \
|
self.assertEquals(self.pop3_account.get_mail(1), \
|
||||||
(u"From : user@test.com\n" + \
|
(u"From : user@test.com\n" + \
|
||||||
u"Subject : subject test\n\n" + \
|
u"Subject : subject test\n\n" + \
|
||||||
u"mymessage\n", \
|
u"mymessage\n", \
|
||||||
@@ -198,7 +246,7 @@ class POP3Connection_TestCase(unittest.TestCase):
|
|||||||
["RETR 1\r\n",
|
["RETR 1\r\n",
|
||||||
"RSET\r\n"], \
|
"RSET\r\n"], \
|
||||||
lambda self: \
|
lambda self: \
|
||||||
self.assertEquals(self.pop3connection.get_mail_summary(1), \
|
self.assertEquals(self.pop3_account.get_mail_summary(1), \
|
||||||
(u"From : user@test.com\n" + \
|
(u"From : user@test.com\n" + \
|
||||||
u"Subject : subject test\n\n", \
|
u"Subject : subject test\n\n", \
|
||||||
u"user@test.com")))
|
u"user@test.com")))
|
||||||
@@ -212,26 +260,47 @@ class POP3Connection_TestCase(unittest.TestCase):
|
|||||||
["RETR 1\r\n",
|
["RETR 1\r\n",
|
||||||
"RSET\r\n"], \
|
"RSET\r\n"], \
|
||||||
lambda self: \
|
lambda self: \
|
||||||
self.assertEquals(self.pop3connection.get_mail(1), \
|
self.assertEquals(self.pop3_account.get_mail(1), \
|
||||||
(u"From : user@test.com\n" + \
|
(u"From : user@test.com\n" + \
|
||||||
u"Subject : subject test\n\n" + \
|
u"Subject : subject test\n\n" + \
|
||||||
u"mymessage\n", \
|
u"mymessage\n", \
|
||||||
u"user@test.com")))
|
u"user@test.com")))
|
||||||
|
|
||||||
|
|
||||||
class IMAPConnection_TestCase(unittest.TestCase):
|
class IMAPAccount_TestCase(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.server = dummy_server.DummyServer("localhost", 1143)
|
self.server = dummy_server.DummyServer("localhost", 1143)
|
||||||
thread.start_new_thread(self.server.serve, ())
|
thread.start_new_thread(self.server.serve, ())
|
||||||
self.imap_connection = IMAPConnection("login", \
|
if os.path.exists(DB_PATH):
|
||||||
"pass", \
|
os.unlink(DB_PATH)
|
||||||
"localhost", \
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
1143, \
|
Account.createTable(ifNotExists = True)
|
||||||
ssl = False)
|
PresenceAccount.createTable(ifNotExists = True)
|
||||||
|
MailAccount.createTable(ifNotExists = True)
|
||||||
|
IMAPAccount.createTable(ifNotExists = True)
|
||||||
|
self.imap_account = IMAPAccount(user_jid = "user1@test.com", \
|
||||||
|
name = "account1", \
|
||||||
|
jid = "account1@jmc.test.com", \
|
||||||
|
login = "login")
|
||||||
|
self.imap_account.password = "pass"
|
||||||
|
self.imap_account.host = "localhost"
|
||||||
|
self.imap_account.port = 1143
|
||||||
|
self.imap_account.ssl = False
|
||||||
|
del account.hub.threadConnection
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
|
IMAPAccount.dropTable(ifExists = True)
|
||||||
|
MailAccount.dropTable(ifExists = True)
|
||||||
|
PresenceAccount.dropTable(ifExists = True)
|
||||||
|
Account.dropTable(ifExists = True)
|
||||||
|
del TheURIOpener.cachedURIs['sqlite://' + DB_URL]
|
||||||
|
account.hub.threadConnection.close()
|
||||||
|
del account.hub.threadConnection
|
||||||
|
if os.path.exists(DB_PATH):
|
||||||
|
os.unlink(DB_PATH)
|
||||||
self.server = None
|
self.server = None
|
||||||
self.imap_connection = None
|
self.imap_account = None
|
||||||
|
|
||||||
def make_test(responses = None, queries = None, core = None):
|
def make_test(responses = None, queries = None, core = None):
|
||||||
def inner(self):
|
def inner(self):
|
||||||
@@ -250,12 +319,14 @@ class IMAPConnection_TestCase(unittest.TestCase):
|
|||||||
if queries:
|
if queries:
|
||||||
self.server.queries += queries
|
self.server.queries += queries
|
||||||
self.server.queries += ["^[^ ]* LOGOUT"]
|
self.server.queries += ["^[^ ]* LOGOUT"]
|
||||||
self.imap_connection.connect()
|
self.imap_account.connect()
|
||||||
self.failUnless(self.imap_connection.connection, \
|
self.failUnless(self.imap_account.connection, \
|
||||||
"Cannot establish connection")
|
"Cannot establish connection")
|
||||||
if core:
|
if core:
|
||||||
|
account.hub.threadConnection = connectionForURI('sqlite://' + DB_URL)
|
||||||
core(self)
|
core(self)
|
||||||
self.imap_connection.disconnect()
|
del account.hub.threadConnection
|
||||||
|
self.imap_account.disconnect()
|
||||||
self.failUnless(self.server.verify_queries())
|
self.failUnless(self.server.verify_queries())
|
||||||
return inner
|
return inner
|
||||||
|
|
||||||
@@ -271,7 +342,7 @@ class IMAPConnection_TestCase(unittest.TestCase):
|
|||||||
["^[^ ]* SELECT INBOX", \
|
["^[^ ]* SELECT INBOX", \
|
||||||
"^[^ ]* SEARCH RECENT"], \
|
"^[^ ]* SEARCH RECENT"], \
|
||||||
lambda self: \
|
lambda self: \
|
||||||
self.assertEquals(self.imap_connection.get_mail_list(), ['9', '10']))
|
self.assertEquals(self.imap_account.get_mail_list(), ['9', '10']))
|
||||||
|
|
||||||
test_get_mail_summary = make_test([lambda data: "* 42 EXISTS\r\n* 1 RECENT\r\n* OK" +\
|
test_get_mail_summary = make_test([lambda data: "* 42 EXISTS\r\n* 1 RECENT\r\n* OK" +\
|
||||||
" [UNSEEN 9]\r\n* FLAGS (\Deleted \Seen\*)\r\n*" +\
|
" [UNSEEN 9]\r\n* FLAGS (\Deleted \Seen\*)\r\n*" +\
|
||||||
@@ -283,7 +354,7 @@ class IMAPConnection_TestCase(unittest.TestCase):
|
|||||||
data.split()[0] + " OK FETCH completed\r\n"], \
|
data.split()[0] + " OK FETCH completed\r\n"], \
|
||||||
["^[^ ]* EXAMINE INBOX", \
|
["^[^ ]* EXAMINE INBOX", \
|
||||||
"^[^ ]* FETCH 1 \(RFC822\)"], \
|
"^[^ ]* FETCH 1 \(RFC822\)"], \
|
||||||
lambda self: self.assertEquals(self.imap_connection.get_mail_summary(1), \
|
lambda self: self.assertEquals(self.imap_account.get_mail_summary(1), \
|
||||||
(u"From : None\nSubject : None\n\n", \
|
(u"From : None\nSubject : None\n\n", \
|
||||||
u"None")))
|
u"None")))
|
||||||
|
|
||||||
@@ -297,7 +368,7 @@ class IMAPConnection_TestCase(unittest.TestCase):
|
|||||||
data.split()[0] + " OK FETCH completed\r\n"], \
|
data.split()[0] + " OK FETCH completed\r\n"], \
|
||||||
["^[^ ]* EXAMINE INBOX", \
|
["^[^ ]* EXAMINE INBOX", \
|
||||||
"^[^ ]* FETCH 1 \(RFC822\)",], \
|
"^[^ ]* FETCH 1 \(RFC822\)",], \
|
||||||
lambda self: self.assertEquals(self.imap_connection.get_mail(1), \
|
lambda self: self.assertEquals(self.imap_account.get_mail(1), \
|
||||||
(u"From : None\nSubject : None\n\nbody text\r\n\n", \
|
(u"From : None\nSubject : None\n\nbody text\r\n\n", \
|
||||||
u"None")))
|
u"None")))
|
||||||
|
|
||||||
@@ -1,355 +0,0 @@
|
|||||||
##
|
|
||||||
## test_mailconnection_factory.py
|
|
||||||
## Login : David Rousselie <david.rousselie@happycoders.org>
|
|
||||||
## Started on Fri May 20 10:46:58 2005
|
|
||||||
## $Id: test_component.py,v 1.2 2005/09/18 20:24:07 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import thread
|
|
||||||
import unittest
|
|
||||||
import dummy_server
|
|
||||||
import time
|
|
||||||
import traceback
|
|
||||||
from pyxmpp import xmlextra
|
|
||||||
from jmc.jabber.component import *
|
|
||||||
from jmc.utils.config import Config
|
|
||||||
|
|
||||||
class TestStreamHandler(xmlextra.StreamHandler):
|
|
||||||
def __init__(self, expected_balises = []):
|
|
||||||
xmlextra.StreamHandler.__init__(self)
|
|
||||||
self.expected_balises = expected_balises
|
|
||||||
|
|
||||||
def stream_start(self, doc):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def stream_end(self, doc):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def stanza(self, notused, node):
|
|
||||||
pass
|
|
||||||
|
|
||||||
class MailComponent_TestCase_NoConnection(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.mail_component = MailComponent(Config("tests/jmc-test.xml"))
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
self.mail_component = None
|
|
||||||
|
|
||||||
def test_get_reg_form(self):
|
|
||||||
reg_form = self.mail_component.get_reg_form()
|
|
||||||
reg_form2 = self.mail_component.get_reg_form()
|
|
||||||
self.assertTrue(reg_form is reg_form2)
|
|
||||||
|
|
||||||
#TODO
|
|
||||||
class MailComponent_TestCase_Basic(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.handler = TestStreamHandler()
|
|
||||||
self.mail_component = MailComponent(Config("tests/jmc-test.xml"))
|
|
||||||
self.server = dummy_server.XMLDummyServer("localhost", 55555, None, self.handler)
|
|
||||||
thread.start_new_thread(self.server.serve, ())
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
self.server = None
|
|
||||||
self.mail_component = None
|
|
||||||
os.remove("./registered.db")
|
|
||||||
|
|
||||||
def test_run(self):
|
|
||||||
self.server.responses = ["<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:component:accept' id='4258238724' from='localhost'>", \
|
|
||||||
"<handshake/></stream:stream>"]
|
|
||||||
self.server.queries = ["<?xml version='1.0' encoding='UTF-8'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:component:accept' to='jmc.localhost' version='1.0'>", \
|
|
||||||
"<handshake>[0-9abcdef]*</handshake>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.mail_component.run(1)
|
|
||||||
self.failUnless(self.server.verify_queries())
|
|
||||||
# TODO : more assertion
|
|
||||||
|
|
||||||
class MailComponent_TestCase_NoReg(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.handler = TestStreamHandler()
|
|
||||||
self.mail_component = MailComponent(Config("tests/jmc-test.xml"))
|
|
||||||
self.server = dummy_server.XMLDummyServer("localhost", 55555, None, self.handler)
|
|
||||||
thread.start_new_thread(self.server.serve, ())
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
self.server = None
|
|
||||||
self.mail_component = None
|
|
||||||
## TODO : be storage independant
|
|
||||||
os.remove("./registered.db")
|
|
||||||
|
|
||||||
def test_disco_get_items(self):
|
|
||||||
self.server.responses = ["<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:component:accept' id='4258238724' from='localhost'>",
|
|
||||||
"<handshake/><iq type='get' to='jmc.localhost' id='aabca'><query xmlns='http://jabber.org/protocol/disco#info'/></iq>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.server.queries = ["<?xml version=\"1.0\" encoding=\"UTF-8\"?>" + \
|
|
||||||
"<stream:stream xmlns:stream=\"http://etherx.jabber.org/streams\" xmlns=\"jabber:component:accept\" to=\"jmc.localhost\" version=\"1.0\">", \
|
|
||||||
"<handshake>[0-9abcdef]*</handshake>",
|
|
||||||
"<iq from=\"jmc.localhost\" type=\"result\" id=\"aabca\"><query xmlns=\"http://jabber.org/protocol/disco#info\"><feature var=\"jabber:iq:version\"/><feature var=\"jabber:iq:register\"/><identity name=\"Jabber Mail Component\" category=\"headline\" type=\"mail\"/></query></iq>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.mail_component.run(1)
|
|
||||||
self.failUnless(self.server.verify_queries())
|
|
||||||
|
|
||||||
def test_get_register(self):
|
|
||||||
self.server.responses = ["<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:component:accept' id='4258238724' from='localhost'>",
|
|
||||||
"<handshake/><iq type='get' to='jmc.localhost' from='test@localhost/test' id='aad9a'><query xmlns='jabber:iq:register'/></iq>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.server.queries = ["<?xml version=\"1.0\" encoding=\"UTF-8\"?>" + \
|
|
||||||
"<stream:stream xmlns:stream=\"http://etherx.jabber.org/streams\" xmlns=\"jabber:component:accept\" to=\"jmc.localhost\" version=\"1.0\">", \
|
|
||||||
"<handshake>[0-9abcdef]*</handshake>",
|
|
||||||
"<iq from=\"jmc.localhost\" to=\"test@localhost/test\" type=\"result\" id=\"aad9a\">" + \
|
|
||||||
"<query xmlns=\"jabber:iq:register\">" + \
|
|
||||||
"<x xmlns=\"jabber:x:data\">" + \
|
|
||||||
"<title>Jabber Mail connection registration</title>" + \
|
|
||||||
"<instructions>Enter anything below</instructions>" + \
|
|
||||||
"<field type=\"text-single\" label=\"Connection name\" var=\"name\"/>" + \
|
|
||||||
"<field type=\"text-single\" label=\"Login\" var=\"login\"/>" + \
|
|
||||||
"<field type=\"text-private\" label=\"Password\" var=\"password\"/>" + \
|
|
||||||
"<field type=\"text-single\" label=\"Host\" var=\"host\"/>" + \
|
|
||||||
"<field type=\"text-single\" label=\"Port\" var=\"port\"/>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Mailbox type\" var=\"type\">" + \
|
|
||||||
"<option label=\"POP3\">" + \
|
|
||||||
"<value>pop3</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"POP3S\">" + \
|
|
||||||
"<value>pop3s</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"IMAP\">" + \
|
|
||||||
"<value>imap</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"IMAPS\">" + \
|
|
||||||
"<value>imaps</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"text-single\" label=\"Mailbox (IMAP)\" var=\"mailbox\">" + \
|
|
||||||
"<value>INBOX</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Action when state is 'Free For Chat'\" var=\"chat_action\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"<option label=\"Do nothing\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Send mail digest\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Retrieve mail\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Action when state is 'Online'\" var=\"online_action\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"<option label=\"Do nothing\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Send mail digest\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Retrieve mail\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Action when state is 'Away'\" var=\"away_action\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"<option label=\"Do nothing\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Send mail digest\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Retrieve mail\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Action when state is 'Not Available'\" var=\"xa_action\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"<option label=\"Do nothing\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Send mail digest\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Retrieve mail\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Action when state is 'Do not Disturb'\" var=\"dnd_action\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"<option label=\"Do nothing\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Send mail digest\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Retrieve mail\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"list-single\" label=\"Action when state is 'Offline'\" var=\"offline_action\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"<option label=\"Do nothing\">" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Send mail digest\">" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"<option label=\"Retrieve mail\">" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</option>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type=\"text-single\" label=\"Mail check interval (in minutes)\" var=\"interval\">" + \
|
|
||||||
"<value>5</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"</x>" + \
|
|
||||||
"</query>" + \
|
|
||||||
"</iq>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.mail_component.run(1)
|
|
||||||
self.failUnless(self.server.verify_queries())
|
|
||||||
|
|
||||||
|
|
||||||
def test_disco_get_info(self):
|
|
||||||
self.server.responses = ["<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:component:accept' id='4258238724' from='localhost'>",
|
|
||||||
"<handshake/><iq type='get' to='jmc.localhost' from='test@localhost/test' id='aad9a'><query xmlns='http://jabber.org/protocol/disco#info'/></iq>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.server.queries = ["<?xml version=\"1.0\" encoding=\"UTF-8\"?>" + \
|
|
||||||
"<stream:stream xmlns:stream=\"http://etherx.jabber.org/streams\" xmlns=\"jabber:component:accept\" to=\"jmc.localhost\" version=\"1.0\">",
|
|
||||||
"<handshake>[0-9abcdef]*</handshake>",
|
|
||||||
"<iq from=\"jmc.localhost\" to=\"test@localhost/test\" type=\"result\" id=\"aad9a\">" + \
|
|
||||||
"<query xmlns=\"http://jabber.org/protocol/disco#info\">" + \
|
|
||||||
"<feature var=\"jabber:iq:version\"/>" + \
|
|
||||||
"<feature var=\"jabber:iq:register\"/>" + \
|
|
||||||
"<identity name=\"Jabber Mail Component\" category=\"headline\" type=\"mail\"/>" + \
|
|
||||||
"</query></iq>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.mail_component.run(1)
|
|
||||||
self.failUnless(self.server.verify_queries())
|
|
||||||
|
|
||||||
def test_set_register(self):
|
|
||||||
self.server.responses = ["<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' xmlns='jabber:component:accept' id='4258238724' from='localhost'>", \
|
|
||||||
"<handshake/>" + \
|
|
||||||
"<iq from='test@localhost/test' to='jmc.localhost' type='set' id='aacaa'>" + \
|
|
||||||
"<query xmlns='jabber:iq:register'>" + \
|
|
||||||
"<x xmlns='jabber:x:data' type='submit'>" + \
|
|
||||||
"<field type='text-single' var='name'>" + \
|
|
||||||
"<value>test</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='text-single' var='login'>" + \
|
|
||||||
"<value>logintest</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='text-private' var='password'>" + \
|
|
||||||
"<value>passtest</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='text-single' var='host'>" + \
|
|
||||||
"<value>hosttest</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='text-single' var='port'>" + \
|
|
||||||
"<value>993</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='type'>" + \
|
|
||||||
"<value>imaps</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='text-single' var='mailbox'>" + \
|
|
||||||
"<value>INBOX</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='chat_action'>" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='online_action'>" + \
|
|
||||||
"<value>2</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='away_action'>" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='xa_action'>" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='dnd_action'>" + \
|
|
||||||
"<value>1</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='list-single' var='offline_action'>" + \
|
|
||||||
"<value>0</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"<field type='text-single' var='interval'>" + \
|
|
||||||
"<value>5</value>" + \
|
|
||||||
"</field>" + \
|
|
||||||
"</x>" + \
|
|
||||||
"</query></iq>",
|
|
||||||
lambda x: None,
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.server.queries = ["<?xml version=\"1.0\" encoding=\"UTF-8\"?>" + \
|
|
||||||
"<stream:stream xmlns:stream=\"http://etherx.jabber.org/streams\" xmlns=\"jabber:component:accept\" to=\"jmc.localhost\" version=\"1.0\">", \
|
|
||||||
"<handshake>[0-9abcdef]*</handshake>", \
|
|
||||||
"<iq from=\"jmc.localhost\" to=\"test@localhost/test\" type=\"result\" id=\"aacaa\"/>",
|
|
||||||
"<presence from=\"jmc.localhost\" to=\"test@localhost\" type=\"subscribe\"/>",
|
|
||||||
"<message from=\"jmc.localhost\" to=\"test@localhost/test\" type=\"message\"><body>New imaps connection \\'test\\': Registered with username \\'logintest\\' and password \\'passtest\\' on \\'hosttest:993\\'</body></message>",
|
|
||||||
"<presence from=\"test@jmc.localhost\" to=\"test@localhost\" type=\"subscribe\"/>",
|
|
||||||
"</stream:stream>"]
|
|
||||||
self.mail_component.run(1)
|
|
||||||
self.failUnless(self.server.verify_queries())
|
|
||||||
|
|
||||||
class MailComponent_TestCase_Reg(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.mail_component = MailComponent(Config("tests/jmc-test.xml"))
|
|
||||||
self.mail_component.set_register(iq = None)
|
|
||||||
|
|
||||||
def test_get_reg_form_init(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_get_reg_form_init_2pass(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_disco_get_items(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_get_register(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_set_register_update(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_set_register_remove(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_available_transport(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_available_mailbox(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_unavailable_transport(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_unavailable_mailbox(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_subscribe(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_subscribed_transport(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_subscribed_mailbox(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_unsubscribe(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_presence_unsubscribed(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def test_check_mail(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
@@ -1,204 +0,0 @@
|
|||||||
##
|
|
||||||
## test_mailconnection_factory.py
|
|
||||||
## Login : David Rousselie <david.rousselie@happycoders.org>
|
|
||||||
## Started on Fri May 20 10:46:58 2005
|
|
||||||
## $Id: test_mailconnection_factory.py,v 1.1 2005/07/11 20:39:31 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import unittest
|
|
||||||
from jmc.email.mailconnection_factory import *
|
|
||||||
from jmc.email.mailconnection import *
|
|
||||||
import jmc.email.mailconnection as mailconnection
|
|
||||||
|
|
||||||
class MailConnectionFactory_TestCase(unittest.TestCase):
|
|
||||||
def test_new_mail_connection_imap(self):
|
|
||||||
mc = get_new_mail_connection("imap")
|
|
||||||
self.assertEquals(mc, IMAPConnection())
|
|
||||||
|
|
||||||
def test_new_mail_connection_imaps(self):
|
|
||||||
mc = get_new_mail_connection("imaps")
|
|
||||||
self.assertEquals(mc, IMAPConnection(ssl = True))
|
|
||||||
|
|
||||||
def test_new_mail_connection_pop3(self):
|
|
||||||
mc = get_new_mail_connection("pop3")
|
|
||||||
self.assertEquals(mc, POP3Connection())
|
|
||||||
|
|
||||||
def test_new_mail_connection_pop3s(self):
|
|
||||||
mc = get_new_mail_connection("pop3s")
|
|
||||||
self.assertEquals(mc, POP3Connection(ssl = True))
|
|
||||||
|
|
||||||
def test_new_mail_connection_unknown(self):
|
|
||||||
self.assertRaises(Exception, get_new_mail_connection, "unknown")
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_imap_v01_v02(self):
|
|
||||||
mc = str_to_mail_connection("imap#login#passwd#host#193#False#INBOX")
|
|
||||||
self.assertEquals(mc.get_type(), "imap")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.store_password, True)
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 193)
|
|
||||||
self.assertEquals(mc.mailbox, "INBOX")
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.interval, 5)
|
|
||||||
self.assertEquals(mc.live_email_only, False)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_imap_v01_v02_retrieve(self):
|
|
||||||
mc = str_to_mail_connection("imap#login#passwd#host#193#True#INBOX")
|
|
||||||
self.assertEquals(mc.get_type(), "imap")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.store_password, True)
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 193)
|
|
||||||
self.assertEquals(mc.mailbox, "INBOX")
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.interval, 5)
|
|
||||||
self.assertEquals(mc.live_email_only, False)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_pop3_v01_v02(self):
|
|
||||||
mc = str_to_mail_connection("pop3#login#passwd#host#110#False")
|
|
||||||
self.assertEquals(mc.get_type(), "pop3")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.store_password, True)
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 110)
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.interval, 5)
|
|
||||||
self.assertEquals(mc.live_email_only, False)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_pop3_v01_v02_retrieve(self):
|
|
||||||
mc = str_to_mail_connection("pop3#login#passwd#host#110#True")
|
|
||||||
self.assertEquals(mc.get_type(), "pop3")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.store_password, True)
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 110)
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.interval, 5)
|
|
||||||
self.assertEquals(mc.live_email_only, False)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_imap(self):
|
|
||||||
mc = str_to_mail_connection("imap#login#passwd#host#193#0#0#0#1#1#2#4#True#INBOX")
|
|
||||||
self.assertEquals(mc.get_type(), "imap")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 193)
|
|
||||||
self.assertEquals(mc.mailbox, "INBOX")
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.interval, 4)
|
|
||||||
self.assertEquals(mc.live_email_only, True)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_no_password(self):
|
|
||||||
mc = str_to_mail_connection("imap#login#/\\#host#193#0#0#0#1#1#2#4#False#INBOX")
|
|
||||||
self.assertEquals(mc.get_type(), "imap")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, None)
|
|
||||||
self.assertEquals(mc.store_password, False)
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 193)
|
|
||||||
self.assertEquals(mc.mailbox, "INBOX")
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.interval, 4)
|
|
||||||
self.assertEquals(mc.live_email_only, False)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_imaps(self):
|
|
||||||
mc = str_to_mail_connection("imaps#login#passwd#host#993#0#0#0#1#1#2#4#True#INBOX.SubDir")
|
|
||||||
self.assertEquals(mc.get_type(), "imaps")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 993)
|
|
||||||
self.assertEquals(mc.mailbox, "INBOX.SubDir")
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.interval, 4)
|
|
||||||
self.assertEquals(mc.live_email_only, True)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_pop3(self):
|
|
||||||
mc = str_to_mail_connection("pop3#login#passwd#host#110#0#0#0#1#1#2#4#False")
|
|
||||||
self.assertEquals(mc.get_type(), "pop3")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 110)
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.interval, 4)
|
|
||||||
self.assertEquals(mc.live_email_only, False)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_pop3s(self):
|
|
||||||
mc = str_to_mail_connection("pop3s#login#passwd#host#995#0#0#0#1#1#2#4#True")
|
|
||||||
self.assertEquals(mc.get_type(), "pop3s")
|
|
||||||
self.assertEquals(mc.login, "login")
|
|
||||||
self.assertEquals(mc.password, "passwd")
|
|
||||||
self.assertEquals(mc.host, "host")
|
|
||||||
self.assertEquals(mc.port, 995)
|
|
||||||
self.assertEquals(mc.chat_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.online_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.away_action, mailconnection.DO_NOTHING)
|
|
||||||
self.assertEquals(mc.xa_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.dnd_action, mailconnection.DIGEST)
|
|
||||||
self.assertEquals(mc.offline_action, mailconnection.RETRIEVE)
|
|
||||||
self.assertEquals(mc.interval, 4)
|
|
||||||
self.assertEquals(mc.live_email_only, True)
|
|
||||||
|
|
||||||
def test_str_to_mail_connection_unknown(self):
|
|
||||||
self.assertRaises(Exception, str_to_mail_connection, ("unknown#login#passwd#host#995#0#0#0#1#1#2#4#True"))
|
|
||||||
|
|
||||||
@@ -1,224 +0,0 @@
|
|||||||
##
|
|
||||||
## test_storage.py
|
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
|
||||||
## Started on Fri May 20 10:46:58 2005 dax
|
|
||||||
## $Id: test_component.py,v 1.1 2005/07/11 20:39:31 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005 adro8400
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import os
|
|
||||||
import unittest
|
|
||||||
import dummy_server
|
|
||||||
from jmc.utils.storage import *
|
|
||||||
from jmc.email.mailconnection import *
|
|
||||||
import jmc.email.mailconnection as mailconnection
|
|
||||||
|
|
||||||
class Storage_TestCase(unittest.TestCase):
|
|
||||||
def test_init(self):
|
|
||||||
spool_dir = "./spool/test"
|
|
||||||
self._storage = Storage(spool_dir = spool_dir)
|
|
||||||
self.assertTrue(os.access(spool_dir, os.F_OK))
|
|
||||||
self.assertEquals(self._storage.spool_dir, spool_dir)
|
|
||||||
os.removedirs(spool_dir)
|
|
||||||
|
|
||||||
class DBMStorage_TestCase(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
spool_dir = "./spool/test"
|
|
||||||
self._storage = DBMStorage(nb_pk_fields = 2, spool_dir = spool_dir)
|
|
||||||
self._account1 = IMAPConnection(login = "login1",
|
|
||||||
password = "password1",
|
|
||||||
host = "host1",
|
|
||||||
port = 993,
|
|
||||||
ssl = True,
|
|
||||||
mailbox = "INBOX.box1")
|
|
||||||
self._account1.chat_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.online_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.away_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.xa_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.dnd_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.offline_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.interval = 4
|
|
||||||
self._account2 = POP3Connection(login = "login2",
|
|
||||||
password = "password2",
|
|
||||||
host = "host2",
|
|
||||||
port = 1110,
|
|
||||||
ssl = False)
|
|
||||||
self._account2.chat_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.online_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.away_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.xa_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.dnd_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.offline_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.interval = 4
|
|
||||||
self._account2.store_password = False
|
|
||||||
self._account2.live_email_only = True
|
|
||||||
|
|
||||||
def tearDown(self):
|
|
||||||
db_file = self._storage.file
|
|
||||||
self._storage = None
|
|
||||||
os.remove(db_file)
|
|
||||||
|
|
||||||
def test_set_get(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
self.assertEquals(self._storage[("test@localhost", "account1")],
|
|
||||||
self._account1)
|
|
||||||
self.assertEquals(self._storage[("test@localhost", "account2")],
|
|
||||||
self._account2)
|
|
||||||
|
|
||||||
def test_set_sync_get(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
loaded_storage = DBMStorage(nb_pk_fields = 2, spool_dir = "./spool/test")
|
|
||||||
self.assertEquals(loaded_storage[("test@localhost", "account1")],
|
|
||||||
self._account1)
|
|
||||||
self.assertEquals(loaded_storage[("test@localhost", "account2")],
|
|
||||||
self._account2)
|
|
||||||
|
|
||||||
def test_set_del_get(self):
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
del self._storage[("test@localhost", "account2")]
|
|
||||||
try:
|
|
||||||
self._storage[("test@localhost", "account2")]
|
|
||||||
except KeyError:
|
|
||||||
return
|
|
||||||
self.fail("KeyError was expected")
|
|
||||||
|
|
||||||
def test_haskey(self):
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
self.assertTrue(self._storage.has_key((u"test@localhost", u"account2")))
|
|
||||||
|
|
||||||
def test_partial_haskey(self):
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
self.assertTrue(self._storage.has_key((u"test@localhost",)))
|
|
||||||
|
|
||||||
def test_get_filtered(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
result = self._storage[("test@localhost",)]
|
|
||||||
self.assertEquals(type(result), list)
|
|
||||||
self.assertEquals(len(result), 2)
|
|
||||||
self.assertEquals(result[1], self._account1)
|
|
||||||
self.assertEquals(result[0], self._account2)
|
|
||||||
|
|
||||||
def test_get_filtered2(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
result = self._storage[("account1",)]
|
|
||||||
self.assertEquals(type(result), list)
|
|
||||||
self.assertEquals(len(result), 1)
|
|
||||||
self.assertEquals(result[0], self._account1)
|
|
||||||
|
|
||||||
def test_keys(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
result = self._storage.keys()
|
|
||||||
self.assertEquals(type(result), list)
|
|
||||||
self.assertEquals(len(result), 2)
|
|
||||||
self.assertEquals(type(result[1]), tuple)
|
|
||||||
self.assertEquals(len(result[1]), 2)
|
|
||||||
self.assertEquals(result[1][0], "test@localhost")
|
|
||||||
self.assertEquals(result[1][1], "account1")
|
|
||||||
self.assertEquals(type(result[0]), tuple)
|
|
||||||
self.assertEquals(len(result[0]), 2)
|
|
||||||
self.assertEquals(result[0][0], "test@localhost")
|
|
||||||
self.assertEquals(result[0][1], "account2")
|
|
||||||
|
|
||||||
def test_keys_filtered(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
result = self._storage.keys(())
|
|
||||||
self.assertEquals(type(result), list)
|
|
||||||
self.assertEquals(len(result), 1)
|
|
||||||
self.assertEquals(result[0], "test@localhost")
|
|
||||||
|
|
||||||
def test_keys_filtered2(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
result = self._storage.keys(("test@localhost",))
|
|
||||||
self.assertEquals(type(result), list)
|
|
||||||
self.assertEquals(len(result), 2)
|
|
||||||
self.assertEquals(result[0], "account2")
|
|
||||||
self.assertEquals(result[1], "account1")
|
|
||||||
|
|
||||||
def test_del_sync_get(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
del self._storage[("test@localhost", "account2")]
|
|
||||||
loaded_storage = DBMStorage(nb_pk_fields = 2, spool_dir = "./spool/test")
|
|
||||||
self.assertEquals(len(loaded_storage.keys()),
|
|
||||||
1)
|
|
||||||
self.assertEquals(loaded_storage[("test@localhost", "account1")],
|
|
||||||
self._account1)
|
|
||||||
|
|
||||||
class SQLiteStorage_TestCase(DBMStorage_TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
spool_dir = "./spool/test"
|
|
||||||
self._storage = SQLiteStorage(nb_pk_fields = 2, spool_dir = spool_dir)
|
|
||||||
self._account1 = IMAPConnection(login = "login1",
|
|
||||||
password = "password1",
|
|
||||||
host = "host1",
|
|
||||||
port = 993,
|
|
||||||
ssl = True,
|
|
||||||
mailbox = "INBOX.box1")
|
|
||||||
self._account1.chat_action = mailconnection.DIGEST
|
|
||||||
self._account1.online_action = mailconnection.DIGEST
|
|
||||||
self._account1.away_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.xa_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.dnd_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.offline_action = mailconnection.DO_NOTHING
|
|
||||||
self._account1.interval = 4
|
|
||||||
self._account2 = POP3Connection(login = "login2",
|
|
||||||
password = "password2",
|
|
||||||
host = "host2",
|
|
||||||
port = 1993,
|
|
||||||
ssl = False)
|
|
||||||
self._account2.chat_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.online_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.away_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.xa_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.dnd_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.offline_action = mailconnection.DO_NOTHING
|
|
||||||
self._account2.interval = 4
|
|
||||||
self._account2.store_password = False
|
|
||||||
self._account2.live_email_only = True
|
|
||||||
|
|
||||||
# def tearDown(self):
|
|
||||||
# os.remove(self._storage.file)
|
|
||||||
# self._storage = None
|
|
||||||
|
|
||||||
|
|
||||||
def test_set_sync_get(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
self._account2.password = None
|
|
||||||
loaded_storage = SQLiteStorage(nb_pk_fields = 2, spool_dir = "./spool/test")
|
|
||||||
self.assertEquals(loaded_storage[("test@localhost", "account1")],
|
|
||||||
self._account1)
|
|
||||||
self.assertEquals(loaded_storage[("test@localhost", "account2")],
|
|
||||||
self._account2)
|
|
||||||
|
|
||||||
def test_del_sync_get(self):
|
|
||||||
self._storage[("test@localhost", "account1")] = self._account1
|
|
||||||
self._storage[("test@localhost", "account2")] = self._account2
|
|
||||||
del self._storage[("test@localhost", "account2")]
|
|
||||||
loaded_storage = SQLiteStorage(nb_pk_fields = 2, spool_dir = "./spool/test")
|
|
||||||
self.assertEquals(len(loaded_storage.keys()),
|
|
||||||
1)
|
|
||||||
self.assertEquals(loaded_storage[("test@localhost", "account1")],
|
|
||||||
self._account1)
|
|
||||||
|
|
||||||
@@ -1,72 +0,0 @@
|
|||||||
##
|
|
||||||
## test_x.py
|
|
||||||
## Login : David Rousselie <david.rousselie@happycoders.org>
|
|
||||||
## Started on Fri May 20 10:46:58 2005
|
|
||||||
## $Id: test_x.py,v 1.1 2005/07/11 20:39:31 dax Exp $
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import unittest
|
|
||||||
from jmc.jabber.x import *
|
|
||||||
|
|
||||||
class X_TestCase(unittest.TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.mail_component = MailComponent()
|
|
||||||
|
|
||||||
def test_get_form(self):
|
|
||||||
self.reg_form.add_field(type = "text-single", \
|
|
||||||
label = "Connection name", \
|
|
||||||
var = "name")
|
|
||||||
|
|
||||||
self.reg_form.add_field(type = "text-single", \
|
|
||||||
label = "Login", \
|
|
||||||
var = "login")
|
|
||||||
|
|
||||||
self.reg_form.add_field(type = "text-private", \
|
|
||||||
label = "password", \
|
|
||||||
var = "password")
|
|
||||||
|
|
||||||
self.reg_form.add_field(type = "text-single", \
|
|
||||||
label = "Host", \
|
|
||||||
var = "host")
|
|
||||||
|
|
||||||
self.reg_form.add_field(type = "text-single", \
|
|
||||||
label = "Port", \
|
|
||||||
var = "port")
|
|
||||||
|
|
||||||
field = self.reg_form.add_field(type = "list-single", \
|
|
||||||
label = "Mailbox type", \
|
|
||||||
var = "type")
|
|
||||||
field.add_option(label = "POP3", \
|
|
||||||
value = "pop3")
|
|
||||||
field.add_option(label = "POP3S", \
|
|
||||||
value = "pop3s")
|
|
||||||
field.add_option(label = "IMAP", \
|
|
||||||
value = "imap")
|
|
||||||
field.add_option(label = "IMAPS", \
|
|
||||||
value = "imaps")
|
|
||||||
|
|
||||||
self.reg_form.add_field(type = "text-single", \
|
|
||||||
label = "Mailbox (IMAP)", \
|
|
||||||
var = "mailbox", \
|
|
||||||
value = "INBOX")
|
|
||||||
|
|
||||||
self.reg_form.add_field(type = "boolean", \
|
|
||||||
label = "Retrieve mail", \
|
|
||||||
var = "retrieve")
|
|
||||||
pass
|
|
||||||
|
|
||||||
@@ -1,95 +0,0 @@
|
|||||||
##
|
|
||||||
## utils.py
|
|
||||||
## Login : David Rousselie <dax@happycoders.org>
|
|
||||||
## Started on Mon Oct 24 21:44:43 2005 dax
|
|
||||||
## $Id$
|
|
||||||
##
|
|
||||||
## Copyright (C) 2005 dax
|
|
||||||
## 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 2 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, write to the Free Software
|
|
||||||
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
||||||
##
|
|
||||||
|
|
||||||
import xml.dom.minidom
|
|
||||||
import re
|
|
||||||
|
|
||||||
def xmldiff(node1, node2):
|
|
||||||
if node1.nodeType == node1.TEXT_NODE:
|
|
||||||
if not node2.nodeType == node2.TEXT_NODE \
|
|
||||||
or re.compile(node2.data + "$").match(node1.data) is None:
|
|
||||||
raise Exception("data in text node " + node1.data + " does not match " + node2.data)
|
|
||||||
elif node1.nodeType == node1.DOCUMENT_NODE:
|
|
||||||
if not node2.nodeType == node2.DOCUMENT_NODE:
|
|
||||||
raise Exception("node1 is Document but not node2 (" + node2.nodeType + ")")
|
|
||||||
elif node1.tagName != node2.tagName:
|
|
||||||
raise Exception("Different tag name : " + node1.tagName + " != " + node2.tagName)
|
|
||||||
else:
|
|
||||||
for attr in node1._get_attributes().keys():
|
|
||||||
if not node2.hasAttribute(attr) \
|
|
||||||
or node1.getAttribute(attr) != node2.getAttribute(attr):
|
|
||||||
raise Exception("(" + node1.tagName + ") Different attributes : " + node1.getAttribute(attr) + " != " + node2.getAttribute(attr))
|
|
||||||
if len(node1.childNodes) != len(node2.childNodes):
|
|
||||||
raise Exception("(" + node1.tagName + ") Different children number : " + str(len(node1.childNodes)) + " != " + str(len(node2.childNodes)))
|
|
||||||
for i in range(len(node1.childNodes)):
|
|
||||||
xmldiff(node1.childNodes[i], node2.childNodes[i])
|
|
||||||
|
|
||||||
# def xmldiff(events1, events2):
|
|
||||||
# for (event1, node1) in events1:
|
|
||||||
# (event2, node2) = events2.next()
|
|
||||||
# print event1 + " " + str(node1)
|
|
||||||
# if not (event1 == event2) or not xml_diff_nodes(node1, node2):
|
|
||||||
# return False
|
|
||||||
# return True
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
document1 = """\
|
|
||||||
<slideshow attr='value'>
|
|
||||||
<title>Demo slideshow</title>
|
|
||||||
<slide><title>Slide title</title>
|
|
||||||
<point>This is a demo</point>
|
|
||||||
<point>Of a program for processing slides</point>
|
|
||||||
</slide>
|
|
||||||
|
|
||||||
<slide><title>Another demo slide</title>
|
|
||||||
<point>It is important</point>
|
|
||||||
<point>To have more than</point>
|
|
||||||
<point>one slide</point>
|
|
||||||
</slide>
|
|
||||||
</slideshow>
|
|
||||||
"""
|
|
||||||
|
|
||||||
document2 = """\
|
|
||||||
<slideshow attr='value'>
|
|
||||||
<title>Demo slideshow</title>
|
|
||||||
<slide><title>Slide title</title>
|
|
||||||
<point>This is a demo</point>
|
|
||||||
<point>Of a program for processing slides</point>
|
|
||||||
</slide>
|
|
||||||
|
|
||||||
<slide><title>Another demo slide</title>
|
|
||||||
<point>It is important</point>
|
|
||||||
<point>To have more than</point>
|
|
||||||
<point>one slide</point>
|
|
||||||
</slide>
|
|
||||||
</slideshow>
|
|
||||||
"""
|
|
||||||
|
|
||||||
dom1 = xml.dom.minidom.parseString(document1)
|
|
||||||
dom2 = xml.dom.minidom.parseString(document2)
|
|
||||||
|
|
||||||
try:
|
|
||||||
xmldiff(dom1, dom2)
|
|
||||||
except Exception, msg:
|
|
||||||
print msg
|
|
||||||
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user