spectacle/tests/testbase.py

#!/usr/bin/python -tt
# vim: ai ts=4 sts=4 et sw=4

#    Copyright (c) 2009 Intel Corporation
#
#    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; version 2 of the License
#
#    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.

SCRIPTS = """
specify -N -n -o output.spec testpkg.yaml 1>output.1.o 2>output.2.o
mv output.spec output.orig.spec
patch -s < input.p
specify -N -n -o output.spec testpkg.yaml 1>ouput.1 2>output.2
if [ ! -f output.no -a ! -f output.spec ]; then
  mv output.2 output.error
  exit 1
fi
if [ -f output.p ]; then
  diff -upN output.orig.spec output.spec > newoutput.p
  mv output.spec output.new.spec
  patch < output.p >/dev/null
fi
if [ -f output.1p ]; then
  diff -upN output.1.o output.1 > newoutput.1p
  mv output.1 output.1.n
  patch < output.1p >/dev/null
fi
if [ -f output.2p ]; then
  diff -upN output.2.o output.2 > newoutput.2p
  mv output.2 output.2.n
  patch < output.2p >/dev/null
fi
"""

import os,sys
import glob
import shutil
import filecmp

BLUE='\033[%dm' % 34
RED='\033[%dm' % 31
RESET = '\033[0m'

def prep_working_env(cases_dir, case, dst_dir):
    shutil.copy(os.path.join(cases_dir, 'base', 'testpkg.yaml'), dst_dir)
    for out in glob.glob(os.path.join(cases_dir, 'test-'+case, '*')):
        try:
            if not os.path.isdir(out):
                shutil.copy(out, dst_dir)
        except:
            pass # ignore if file missing

def cleanup(work_dir):
    shutil.rmtree(work_dir)

def run_and_check(work_dir):
    result = True
    cwd = os.getcwd()
    os.chdir(work_dir)
    os.system(SCRIPTS)
    if os.path.exists('output.error'):
        # something wrong with tested tools
        print >> sys.stderr, file(os.path.join(work_dir, 'output.error')).read()
        result = False

    os.chdir(cwd)
    return result

def compare_outfile(work_dir):
    all_equ = True
    #print glob.glob(os.path.join(work_dir, '*'))
    desc = {'output.p': '*.spec',
            'output.1p': 'STDOUT',
            'output.2p': 'STDERR'}
    orig = {'output.p': 'output.orig.spec',
            'output.1p': 'output.1.o',
            'output.2p': 'output.2.o'}
    new  = {'output.p': 'output.new.spec',
            'output.1p': 'output.1.n',
            'output.2p': 'output.2.n'}
    for out in ('output.p', 'output.1p', 'output.2p'):
        fp = os.path.join(work_dir, out)
        if os.path.exists(fp):
            if not filecmp.cmp(os.path.join(work_dir, orig[out]),\
                               os.path.join(work_dir, new[out])):
                all_equ = False

                exp_output_diff = file(fp).read().strip()
                output_diff = file(os.path.join(work_dir, 'new'+out)).read().strip()

                if not output_diff:
                    output_diff = '<EMPTY>'
                if not exp_output_diff:
                    exp_output_diff = '<EMPTY>'

                print "%sExpected changes of %s:" % (BLUE,desc[out])
                print '----------------------------------------------------------------------%s' % RESET
                print exp_output_diff
                print '%s----------------------------------------------------------------------%s' % (BLUE,RESET)
                print "%sActual:" % RED
                print '----------------------------------------------------------------------%s' % RESET
                print output_diff
                print '%s----------------------------------------------------------------------%s' % (RED,RESET)

    return all_equ