source: main/waeup.uniben/trunk/src/waeup/uniben/students/tests/test_export.py

import datetime
from zope.component import queryUtility
from zope.interface.verify import verifyObject, verifyClass
from waeup.kofa.interfaces import ICSVExporter
from waeup.kofa.schoolgrades import ResultEntry
from waeup.kofa.students.tests.test_batching import (
    StudentImportExportSetup, curr_year)
from waeup.uniben.utils.utils import CustomKofaUtils
from waeup.uniben.students.export import (
    CustomStudentExporter, CustomStudentPaymentExporter,
    MedicalHistoryExporter, NYSCExporter)
from waeup.uniben.testing import FunctionalLayer


class CustomStudentExporterTest(StudentImportExportSetup):

    layer = FunctionalLayer

    def setUp(self):
        super(CustomStudentExporterTest, self).setUp()
        self.setup_for_export()
        result_entry = ResultEntry(
            CustomKofaUtils.EXAM_SUBJECTS_DICT.keys()[0],
            CustomKofaUtils.EXAM_GRADES[0][0]
            )
        self.student.alr_results = [
            result_entry]
        self.student.fst_sit_results = [
            result_entry]
        self.student.scd_sit_results = [
            result_entry]
        return

    def test_ifaces(self):
        # make sure we fullfill interface contracts
        obj = CustomStudentExporter()
        verifyObject(ICSVExporter, obj)
        verifyClass(ICSVExporter, CustomStudentExporter)
        return


    def disabled_test_export_all(self):
        # we can really export students
        # set values we can expect in export file
        self.setup_student(self.student)
        exporter = CustomStudentExporter()
        exporter.export_all(self.app, self.outfile)
        result = open(self.outfile, 'rb').read()
        self.assertMatches(
            'adm_code,alr_date,alr_fname,alr_no,alr_results,'
            'clr_code,date_of_birth,def_adm,disabled,email,email2,emp2_end,'
            'emp2_position,emp2_reason,emp2_start,emp_end,emp_position,'
            'emp_reason,emp_start,employer,employer2,'
            'financial_clearance_date,financially_cleared_by,'
            'firstname,flash_notice,former_matric,'
            'fst_sit_date,fst_sit_fname,fst_sit_no,fst_sit_results,'
            'fst_sit_type,hq2_degree,hq2_disc,hq2_matric_no,hq2_school,'
            'hq2_session,hq2_type,hq_degree,hq_disc,hq_fname,hq_matric_no,'
            'hq_school,hq_session,hq_type,is_staff,lastname,lga,library,'
            'marit_stat,matric_number,middlename,nationality,'
            'next_kin_address,next_kin_name,next_kin_phone,next_kin_relation,'
            'nysc_lga,nysc_location,nysc_year,officer_comment,parent_email,parents_email,'
            'perm_address,personal_updated,phone,'
            'physical_clearance_date,provisionally_cleared,reg_number,'
            'religion,scd_sit_date,scd_sit_fname,scd_sit_no,'
            'scd_sit_results,scd_sit_type,sex,student_id,'
            'suspended,suspended_comment,password,state,history,certcode,is_postgrad,'
            'current_level,current_session\r\nmy adm code,,,,'
            '"[(\'printing_craft_practice\', \'A1\')]",my clr code,1981-02-04#,,,'
            'anna@sample.com,,,,,,,,,,,,,,Anna,,,,,,"[(\'printing_craft_practice\', \'A1\')]"'
            ',,,,,,,,,,,,,,,,Tester,,,,234,M.,NG,,,,,,,,,,,'
            '"Studentroad 21\nLagos 123456\n",,+234-123-12345#,,,123,,,,,'
            '"[(\'printing_craft_practice\', \'A1\')]",,f,A111111,0,,,created,'
            '[u\'2012-11-06 13:16:41 WAT - Record created by system\'],'
            'CERT1,0,200,2012\r\n',
            result
            )
        return

class CustomStudentPaymentExporterTest(StudentImportExportSetup):

    layer = FunctionalLayer

    def setUp(self):
        super(CustomStudentPaymentExporterTest, self).setUp()
        self.setup_for_export()
        return

    def test_ifaces(self):
        # make sure we fullfill interface contracts
        obj = CustomStudentPaymentExporter()
        verifyObject(ICSVExporter, obj)
        verifyClass(ICSVExporter, CustomStudentPaymentExporter)
        return

    def test_export_all(self):
        # we can really export student payments
        # set values we can expect in export file
        self.setup_student(self.student)
        self.student['payments']['my-payment'].r_company = 'interswatch'
        self.student['payments']['my-payment'].r_card_num = '789'
        exporter = CustomStudentPaymentExporter()
        exporter.export_all(self.app, self.outfile)
        result = open(self.outfile, 'rb').read()
        self.assertMatches(
            'ac,amount_auth,creation_date,gateway_amt,net_amt,p_category,p_combi,p_currency,p_current,'
            'p_id,p_item,p_level,p_option,p_session,p_state,payment_date,provider_amt,'
            'r_amount_approved,r_card_num,r_code,r_company,r_desc,'
            'r_pay_reference,r_payment_link,rebate_amount,staff_rebate,thirdparty_amt,student_id,state,'
            'current_session\r\n'
            '666,12.12,%s-04-01 13:12:01#,,,schoolfee,[],,1,my-id,p-item,'
            '100,,%s,paid,%s-04-01 14:12:01#,,12.12,'
            '789,r-code,interswatch,,,,,,,A111111,created,2012\r\n'
            % (curr_year-6, curr_year-6, curr_year-6), result
            )
        return

class MedicalHistoryExporterTest(StudentImportExportSetup):

    layer = FunctionalLayer

    def setUp(self):
        super(MedicalHistoryExporterTest, self).setUp()
        self.setup_for_export()
        return

    def test_ifaces(self):
        # make sure we fullfill interface contracts
        obj = MedicalHistoryExporter()
        verifyObject(ICSVExporter, obj)
        verifyClass(ICSVExporter, MedicalHistoryExporter)
        return

    def test_export_all(self):
        # we can really export students
        # set values we can expect in export file
        self.setup_student(self.student)
        self.student.sore_throat = True
        exporter = MedicalHistoryExporter()
        exporter.export_all(self.app, self.outfile)
        result = open(self.outfile, 'rb').read()
        self.assertMatches(
            'asthma,body_pains,breathing,catarrh,company_confirmed,'
            'company_suspected,cough,diabetes,fever,headaches,hypertension,'
            'lagos_abuja,medical_updated,medicines,negative,obesity,'
            'others,outside,positive,smell,sneezing,sore_throat,taste,'
            'vaccination,weakness,student_id,display_fullname,'
            'matric_number,faccode,depcode,state,current_session,'
            'current_level,genotype,bloodgroup\r\n'
            ',,,,,,,,,,,,,,,,,,,,,1,,,,A111111,Anna M. Tester,234,'
            'NA,NA,created,2012,200,,\r\n' , result
            )
        return

curr_year = datetime.datetime.now().year
year_range = range(curr_year - 11, curr_year + 1)
year_range_str = ','.join([str(i) for i in year_range])

class NYSCExporterTest(StudentImportExportSetup):

    layer = FunctionalLayer

    def setUp(self):
        super(NYSCExporterTest, self).setUp()
        self.setup_for_export()
        return

    def test_ifaces(self):
        # make sure we fullfill interface contracts
        obj = NYSCExporter()
        verifyObject(ICSVExporter, obj)
        verifyClass(ICSVExporter, NYSCExporter)
        return

    def test_export_all(self):
        # we can really export students
        # set values we can expect in export file
        self.setup_student(self.student)
        self.student.nysc = True
        self.student.nysc_verdict = 'B'
        self.student.nysc_processed = False
        self.student.lga = u'cross_river_abc'
        exporter = NYSCExporter()
        exporter.export_all(self.app, self.outfile)
        result = open(self.outfile, 'rb').read()
        self.assertMatches(
            'student_id,matric_number,reg_number,firstname,middlename,lastname,'
            'sex,email,email2,phone,nationality,date_of_birth,state,lg_state,lg_area,current_mode,'
            'certificate,faccode,depcode,current_level,current_session,'
            'current_verdict,entry_session,nysc_verdict,nysc_senate_info,'
            'nysc_date_of_graduation,nysc_updated,nysc_processed,%s\r\n'
            'A111111,234,123,Anna,M.,Tester,f,anna@sample.com,,+234-123-12345,NG,1981-02-04#,created,'
            'cross-river,abc,ug_ft,Unnamed Certificate,NA,NA,200,2012,0,2010,'
            'Student with carryover courses,,,,0,,,,,,'
            '12.12,,,,,,\r\n' % year_range_str,
            result
            )
        return