#!/usr/local/bin/python
#
# File:		objargtest
# Copyright:	(c) 2001 The Regents of the University of California
# Release:	$Name: release-0-8-8 $
# Revision:	$Revision: 1.5 $
# Date:		$Date: 2003/06/06 18:49:00 $
# Description:	Excercise objects in hopes of finding bugs
#

import objarg.Employee
import objarg.EmployeeArray
import Numeric

def toFloat(d):
  tmp = Numeric.zeros((1,), Numeric.Float32)
  try:
    tmp.savespace(1)
    tmp[0] = d
  except AttributeError:
    tmp[0] = d
  return tmp[0]

class TestCounter:
  partno = 0
  numpass = 0
  numfail = 0
  numxfail = 0
  numxpass = 0

  def __init__(self, numparts = -1):
    self.numparts = numparts
    print "NPARTS", numparts

  def describeTest(self, description):
    self.partno = self.partno + 1
    print "PART", self.partno
    print "COMMENT:", description

  def evalTest(self, result, expected):
    if (result):
      if (expected):
        self.numpass = self.numpass + 1
        print "RESULT", self.partno, "PASS"
      else:
        self.numxpass = self.numxpass + 1
        print "RESULT", self.partno, "XPASS"
    else:
      if (expected):
        self.numfail = self.numfail + 1
        print "RESULT", self.partno, "FAIL"
      else:
        self.numxfail = self.numxfail + 1
        print "RESULT", self.partno, "XFAIL"
        
  def finish(self):
    if (self.numparts < 0):
      self.numparts = self.partno
    if (self.numpass == self.numparts):
      print "TEST_RESULT PASS"
    else:
      if (self.numpass + self.numxfail == self.numparts):
        print "TEST_RESULT XFAIL"
      else:
        print "TEST_RESULT FAIL"
      

class Entry:
  def __init__(self, name, age, salary, status):
    self.name = name
    self.age = age
    self.salary = salary
    self.status = status

TestData = [
  Entry( "John Smith", 35, 75.7e3, "c"),
  Entry( "Jane Doe", 40, 85.5e3, "m"),
  Entry( "Ella Vader", 64, 144.2e3, "r" ),
  Entry( "Marge Inovera", 32, 483.2e3, "s" ),
  Entry( "Hughy Louis Dewey", 45, 182.9e3, "m" ),
  Entry( "Heywood Yubuzof", 12, 20.8e3, "x" ),
  Entry( "Picov Andropov", 90, 120.6e3, "r" )
]

if __name__ == '__main__':
  counter = TestCounter()
  
  testlen = len(TestData)
  employeeArray = objarg.EmployeeArray.EmployeeArray()
  index = 0
  for i in TestData:
    counter.describeTest("objarg.Employee.Employee() != None")
    e = objarg.Employee.Employee()
    counter.evalTest(e != None, 1)
    
    counter.describeTest("e.init(i.name, i.age, i.salary, i.status)")
    counter.evalTest(e.init(i.name, i.age, i.salary, i.status), 1)

    counter.describeTest("employeeArray.appendEmployee(e)")
    counter.evalTest(employeeArray.appendEmployee(e), 1)
    index = index + 1

    counter.describeTest("i.salary == e.getSalary()")
    counter.evalTest(i.salary == e.getSalary(), 1)

    counter.describeTest("i.salary == e.getSalary()")
    counter.evalTest(i.salary == e.getSalary(), 1)

    counter.describeTest("i.age == e.getAge()")
    counter.evalTest(i.age == e.getAge(), 1)

    counter.describeTest("i.status == e.getStatus()")
    counter.evalTest(i.status == e.getStatus(), 1)

    counter.describeTest("index == employeeArray.getLength()")
    counter.evalTest(index == employeeArray.getLength(), 1)

    counter.describeTest("e.isSame(employeeArray.at(index))")
    counter.evalTest(e.isSame(employeeArray.at(index)), 1)

  index = 0
  for i in TestData:
    index = index + 1
    (empInd, emp) = employeeArray.findByName(i.name)
    counter.describeTest("employeeArray.findByName(i.name)")
    counter.evalTest(empInd == index, 1)
    if (empInd):
      counter.describeTest("emp.isSame(employeeArray.at(empInd))")
      counter.evalTest(emp.isSame(employeeArray.at(empInd)), 1)
  f = objarg.Employee.Employee()
  f.init("Hire High", 21, 0, "s")
  counter.describeTest("employeeArray.promoteToMaxSalary(f)")
  counter.evalTest(employeeArray.promoteToMaxSalary(f), 1)

  counter.describeTest("f.getSalary == 483.2e3")
  counter.evalTest(f.getSalary() == toFloat(483.2e3), 1)

  counter.describeTest("employeeArray.appendEmployee(f)")
  counter.evalTest(employeeArray.appendEmployee(f), 1)

  f = objarg.Employee.Employee()
  f.init("Amadeo Avogadro, conte di Quaregna", 225, 6.022045e23, "d")

  counter.describeTest("not employeeArray.promoteToMaxSalary(f)")
  (ret, f) = employeeArray.promoteToMaxSalary(f)
  counter.evalTest(not ret, 1)

  counter.describeTest("not employeeArray.appendEmployee(None)");
  counter.evalTest(not employeeArray.appendEmployee(None), 1)

  counter.describeTest('not employeeArray.findByName("New Riders")')
  (ret, f) = employeeArray.findByName("New Riders")
  counter.evalTest(not (ret or f), 1)

  counter.describeTest('not employeeArray.promoteToMaxSalary(None)')
  (ret, f) = employeeArray.promoteToMaxSalary(None)
  counter.evalTest(not (ret or f), 1)
  
  counter.finish()
0
