source: products/qPloneComments/tags/3.2.2/tests/helperNotify.py @ 1591

#########################################################
##      Helper methods for testing Mail sending        ##
#########################################################

import sys
import os, os.path
from Products.SecureMailHost.SecureMailHost import SecureMailBase

PREFIX = os.path.abspath(os.path.dirname(__file__))

ALL_PROPS = ['enable_approve_user_notification', 'enable_reply_user_notification',
             'enable_rejected_user_notification','enable_moderation',
             'require_email', 'enable_anonymous_commenting',
             'enable_published_notification', 'enable_approve_notification']

def sample_file_path(file):
    return os.path.join(PREFIX, 'sample', file)

def output_file_path(file):
    return os.path.join(PREFIX, 'output', file)

def getFileContent(f_path):
    result_f = open(f_path,"r")
    result = result_f.read()
    result_f.close()
    return result

def writeToFile(f_path, text):
    result_f = open(f_path,'w')
    result_f.write(text.as_string()+'\n')
    result_f.close()

def clearFile(f_path):
    result_f = open(f_path,"w")
    result_f.write('')
    result_f.close()

def _send_MH( self, mfrom, mto, messageText ):
    files = [f for f in os.listdir(output_file_path("")) if f.startswith('mail')]
    files.sort()
    fn = files and (files[-1]+ '1') or 'mail'
    writeToFile(output_file_path(fn), messageText)

def _send_SMH(self, mfrom, mto, messageText, debug=False):
    files = [f for f in os.listdir(output_file_path("")) if f.startswith('mail')]
    files.sort()
    fn = files and (files[-1]+ '1') or 'mail'
    writeToFile(output_file_path(fn), messageText)

def send_SMH(self, message, mto=None, mfrom=None, subject=None, encode=None):
    files = [f for f in os.listdir(output_file_path("")) if f.startswith('mail')]
    files.sort()
    fn = files and (files[-1]+ '1') or 'mail'
    writeToFile(output_file_path(fn), message)

def prepareMailSendTest():
    # patch SecureMailHost
    SecureMailBase.send = send_SMH
    SecureMailBase._send = _send_SMH

def setProperties(prop_sheet, *props):
    for p in ALL_PROPS:
        prop_sheet._updateProperty(p, p in props)

def testMailExistance():
    for f in os.listdir(output_file_path("")):
        if f.startswith('mail'):
            return True
    return False

def getMails():
    return [file(output_file_path(f),'r').read()
            for f in os.listdir(output_file_path(""))
            if f.startswith('mail')]

def cleanOutputDir():
    for f in os.listdir(output_file_path("")):
        if f.startswith('mail'):
            os.remove(output_file_path(f))