## $Id: test_browser.py 17962 2024-11-21 03:53:25Z henrik $
##
## Copyright (C) 2011 Uli Fouquet & Henrik Bettermann
## 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 shutil
import tempfile
import datetime
import pytz
import os
import grok
from zope.event import notify
from zope.component.hooks import setSite, clearSite
from zope.component import createObject
from zope.testbrowser.testing import Browser
from hurry.workflow.interfaces import IWorkflowInfo, IWorkflowState
from waeup.kofa.app import University
from waeup.kofa.university.faculty import Faculty
from waeup.kofa.university.department import Department
from waeup.kofa.testing import FunctionalTestCase
from waeup.kofa.interfaces import (
    IExtFileStore, IFileStoreNameChooser, IUserAccount, IJobManager)
from waeup.kofa.applicants.container import ApplicantsContainer
from waeup.fceokene.testing import FunctionalLayer

session = datetime.datetime.now().year - 2
SAMPLE_IMAGE = os.path.join(os.path.dirname(__file__), 'test_image.jpg')

class ApplicantUITest(FunctionalTestCase):
    """Perform some browser tests.
    """
    layer = FunctionalLayer

    def setUp(self):
        super(ApplicantUITest, self).setUp()
        # Setup a sample site for each test
        app = University()
        self.dc_root = tempfile.mkdtemp()
        app['datacenter'].setStoragePath(self.dc_root)

        # Prepopulate the ZODB...
        self.getRootFolder()['app'] = app
        # we add the site immediately after creation to the
        # ZODB. Catalogs and other local utilities are not setup
        # before that step.
        self.app = self.getRootFolder()['app']
        # Set site here. Some of the following setup code might need
        # to access grok.getSite() and should get our new app then
        setSite(app)

        self.login_path = 'http://localhost/app/login'

        # Add bec applicants container
        self.beccontainer = ApplicantsContainer()
        self.beccontainer.mode = 'create'
        self.beccontainer.code = u'bec%s' % session
        self.beccontainer.prefix = u'bec'
        self.beccontainer.application_category = u'bec'
        self.beccontainer.year = session
        self.beccontainer.application_fee = 300.0
        self.beccontainer.title = u'This is the bec%s container' % session
        self.app['applicants'][self.beccontainer.code] = self.beccontainer
        delta = datetime.timedelta(days=10)
        self.beccontainer.startdate = datetime.datetime.now(pytz.utc) - delta
        self.beccontainer.enddate = datetime.datetime.now(pytz.utc) + delta

        # Add ug applicants container
        self.putmecontainer = ApplicantsContainer()
        self.putmecontainer.mode = 'create'
        self.putmecontainer.code = u'putme%s' % session
        self.putmecontainer.prefix = u'putme'
        self.putmecontainer.application_category = u'bec' # doesn't matter
        self.putmecontainer.year = session
        self.putmecontainer.application_fee = 300.0
        self.putmecontainer.title = u'This is the putme%s container' % session
        self.app['applicants'][self.putmecontainer.code] = self.putmecontainer
        delta = datetime.timedelta(days=10)
        self.putmecontainer.startdate = datetime.datetime.now(pytz.utc) - delta
        self.putmecontainer.enddate = datetime.datetime.now(pytz.utc) + delta

        # Add tpu applicants container
        self.tpucontainer = ApplicantsContainer()
        self.tpucontainer.mode = 'create'
        self.tpucontainer.code = u'tpu%s' % session
        self.tpucontainer.prefix = u'tpu'
        self.tpucontainer.application_category = u'basic'
        self.tpucontainer.year = session
        self.tpucontainer.application_fee = 300.0
        self.tpucontainer.title = u'This is the tpu%s container' % session
        self.app['applicants'][self.tpucontainer.code] = self.tpucontainer
        delta = datetime.timedelta(days=10)
        self.tpucontainer.startdate = datetime.datetime.now(pytz.utc) - delta
        self.tpucontainer.enddate = datetime.datetime.now(pytz.utc) + delta

        # Add utp applicants container
        self.utpcontainer = ApplicantsContainer()
        self.utpcontainer.mode = 'create'
        self.utpcontainer.code = u'utp%s' % session
        self.utpcontainer.prefix = u'utp'
        self.utpcontainer.application_category = u'basic'
        self.utpcontainer.year = session
        self.utpcontainer.application_fee = 300.0
        self.utpcontainer.title = u'This is the utp%s container' % session
        self.app['applicants'][self.utpcontainer.code] = self.utpcontainer
        delta = datetime.timedelta(days=10)
        self.utpcontainer.startdate = datetime.datetime.now(pytz.utc) - delta
        self.utpcontainer.enddate = datetime.datetime.now(pytz.utc) + delta

        # Populate university
        self.certificate = createObject('waeup.Certificate')
        self.certificate.code = 'CERT1'
        self.certificate.application_category = 'bec'
        self.certificate.start_level = 100
        self.certificate.end_level = 500
        self.certificate.study_mode = u'nce_ft'
        self.app['faculties']['fac1'] = Faculty()
        self.app['faculties']['fac1']['dep1'] = Department()
        self.app['faculties']['fac1']['dep1'].certificates.addCertificate(
            self.certificate)

        # Add (customized) applicants
        becapplicant = createObject(u'waeup.Applicant')
        becapplicant.firstname = u'Anna'
        becapplicant.lastname = u'Post'
        self.app['applicants'][self.beccontainer.code].addApplicant(becapplicant)
        self.becapplication_number = becapplicant.application_number
        self.becapplicant = self.app['applicants'][self.beccontainer.code][
            self.becapplication_number]
        self.becapplicant_path = ('http://localhost/app/applicants/bec%s/%s'
            % (session, self.becapplication_number))

        putmeapplicant = createObject(u'waeup.Applicant')
        putmeapplicant.firstname = u'Anna'
        putmeapplicant.lastname = u'Post'
        self.app['applicants'][self.putmecontainer.code].addApplicant(putmeapplicant)
        self.putmeapplication_number = putmeapplicant.application_number
        self.putmeapplicant = self.app['applicants'][self.putmecontainer.code][
            self.putmeapplication_number]
        self.putmeapplicant_path = ('http://localhost/app/applicants/putme%s/%s'
            % (session, self.putmeapplication_number))

        tpuapplicant = createObject(u'waeup.Applicant')
        tpuapplicant.firstname = u'Michelle'
        tpuapplicant.lastname = u'Obama'
        self.app['applicants'][self.tpucontainer.code].addApplicant(tpuapplicant)
        self.tpuapplication_number = tpuapplicant.application_number
        self.tpuapplicant = self.app['applicants'][self.tpucontainer.code][
            self.tpuapplication_number]
        self.tpuapplicant_path = ('http://localhost/app/applicants/tpu%s/%s'
            % (session, self.tpuapplication_number))
        IUserAccount(
            self.app['applicants'][self.tpucontainer.code][
            self.tpuapplicant.application_number]).setPassword('apwd')

        utpapplicant = createObject(u'waeup.Applicant')
        utpapplicant.firstname = u'Michelle'
        utpapplicant.lastname = u'Obama'
        self.app['applicants'][self.utpcontainer.code].addApplicant(utpapplicant)
        self.utpapplication_number = utpapplicant.application_number
        self.utpapplicant = self.app['applicants'][self.utpcontainer.code][
            self.utpapplication_number]
        self.utpapplicant_path = ('http://localhost/app/applicants/utp%s/%s'
            % (session, self.utpapplication_number))
        IUserAccount(
            self.app['applicants'][self.utpcontainer.code][
            self.utpapplicant.application_number]).setPassword('apwd')

        self.browser = Browser()
        self.browser.handleErrors = False

    def tearDown(self):
        super(ApplicantUITest, self).tearDown()
        shutil.rmtree(self.dc_root)
        clearSite()
        return

    def test_manage_and_view_bec_applicant(self):
        # Managers can manage bec applicants.
        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
        self.browser.open(self.becapplicant_path)
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        self.assertTrue("'O' Level" in self.browser.contents)
        self.browser.open(self.becapplicant_path + '/manage')
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        self.assertTrue("'O' Level" in self.browser.contents)
        self.browser.open(self.becapplicant_path + '/edit')
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        self.assertTrue("'O' Level" in self.browser.contents)
        self.browser.open(self.becapplicant_path + '/application_slip.pdf')
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        return

    def test_manage_and_view_putme_applicant(self):
        # Managers can manage bec applicants.
        self.browser.addHeader('Authorization', 'Basic mgr:mgrpw')
        self.browser.open(self.putmeapplicant_path)
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        self.assertTrue("'O' Level" in self.browser.contents)
        self.browser.open(self.putmeapplicant_path + '/manage')
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        self.assertTrue("'O' Level" in self.browser.contents)
        self.browser.open(self.putmeapplicant_path + '/edit')
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        self.assertTrue("'O' Level" in self.browser.contents)
        self.browser.open(self.putmeapplicant_path + '/application_slip.pdf')
        self.assertEqual(self.browser.headers['Status'], '200 Ok')
        
    def test_final_submit_tpu(self):
        # Create NCE certificate
        certificate1 = createObject('waeup.Certificate')
        certificate1.application_category = 'basic'
        certificate1.code = 'NCECERT1'
        certificate1.title = 'NCECert1'
        certificate1.start_level = 100
        certificate1.end_level = 500
        certificate1.study_mode = u'nce_ft'
        self.app['faculties']['fac1']['dep1'].certificates.addCertificate(
            certificate1)
        IWorkflowState(self.tpuapplicant).setState('paid')
        self.browser.open(self.login_path)
        self.browser.getControl(
            name="form.login").value = self.tpuapplicant.applicant_id
        self.browser.getControl(name="form.password").value = 'apwd'
        self.browser.getControl("Login").click()
        self.browser.open(self.tpuapplicant_path + '/edit')
        self.browser.getControl(name="form.email").value = 'xx@yy.zz'
        self.browser.getControl(name="form.school_tpu").value = ['tpu0007']
        self.browser.getControl(name="form.subj_comb").value = ['NCECERT1']
        image = open(SAMPLE_IMAGE, 'rb')
        ctrl = self.browser.getControl(name='form.passport')
        file_ctrl = ctrl.mech_control
        file_ctrl.add_file(image, filename='myphoto.jpg')
        self.browser.getControl(name="confirm_passport").value = True
        self.browser.getControl("Save").click()
        self.browser.getControl("Finally Submit").click()
        self.assertTrue(
            'Application submitted' in self.browser.contents)
        # Create 10 applicants who already selected tpu0010
        for i in range(1,11):
            dummy = createObject(u'waeup.Applicant')
            dummy.firstname = u'John'
            dummy.lastname = u'Doe %s' %i
            dummy.school_tpu = 'tpu0007'
            self.app['applicants'][
                self.tpucontainer.code].addApplicant(dummy)
            IWorkflowState(dummy).setState('submitted')
        IWorkflowState(self.tpuapplicant).setState('paid')
        self.tpuapplicant.locked = False
        self.browser.open(self.tpuapplicant_path + '/edit')
        self.browser.getControl(name="form.school_tpu").value = ['tpu0007']
        self.browser.getControl("Finally Submit").click()
        self.assertTrue("Maximum number of applications per school exceeded."
            in self.browser.contents)
        self.assertEqual(self.tpuapplicant.state, 'paid')
        # Create second certificate
        certificate2 = createObject('waeup.Certificate')
        certificate2.code = 'NCECERT2'
        certificate2.title = 'NCECert2'
        certificate2.application_category = 'basic'
        certificate2.start_level = 100
        certificate2.end_level = 500
        certificate2.study_mode = u'nce_ft'
        self.app['faculties']['fac1']['dep1'].certificates.addCertificate(
            certificate2)
        # Create one applicant with same subject and same school
        dummy = createObject(u'waeup.Applicant')
        dummy.firstname = u'John'
        dummy.lastname = u'Doe %s' %i
        dummy.school_tpu = 'tpu0012'
        dummy.subj_comb = certificate1
        self.app['applicants'][
            self.tpucontainer.code].addApplicant(dummy)
        IWorkflowState(dummy).setState('submitted')
        self.browser.open(self.tpuapplicant_path + '/edit')
        self.browser.getControl(name="form.school_tpu").value = ['tpu0012']
        self.browser.getControl(name="form.subj_comb").value = ['NCECERT1']
        self.browser.getControl("Finally Submit").click()
        self.assertTrue("Maximum number of applications per school exceeded."
            in self.browser.contents)
        # Same school with another subj_comb can be submitted
        self.assertEqual(self.tpuapplicant.state, 'paid')
        self.browser.open(self.tpuapplicant_path + '/edit')
        self.browser.getControl(name="form.school_tpu").value = ['tpu0012']
        self.browser.getControl(name="form.subj_comb").value = ['NCECERT2']
        self.browser.getControl("Finally Submit").click()
        self.assertEqual(self.tpuapplicant.state, 'submitted')
        return

    def test_final_submit_utp(self):
        # Create BED certificate
        certificate1 = createObject('waeup.Certificate')
        certificate1.application_category = 'basic'
        certificate1.code = 'BEDCERT1'
        certificate1.title = 'BEDCert1'
        certificate1.start_level = 100
        certificate1.end_level = 500
        certificate1.study_mode = u'nce_ft'
        self.app['faculties']['fac1']['dep1'].certificates.addCertificate(
            certificate1)
        IWorkflowState(self.utpapplicant).setState('paid')
        self.browser.open(self.login_path)
        self.browser.getControl(
            name="form.login").value = self.utpapplicant.applicant_id
        self.browser.getControl(name="form.password").value = 'apwd'
        self.browser.getControl("Login").click()
        self.browser.open(self.utpapplicant_path + '/edit')
        self.browser.getControl(name="form.email").value = 'xx@yy.zz'
        self.browser.getControl(name="form.school_utp").value = ['utp0011']
        self.browser.getControl(name="form.subj_comb").value = ['BEDCERT1']
        image = open(SAMPLE_IMAGE, 'rb')
        ctrl = self.browser.getControl(name='form.passport')
        file_ctrl = ctrl.mech_control
        file_ctrl.add_file(image, filename='myphoto.jpg')
        self.browser.getControl(name="confirm_passport").value = True
        self.browser.getControl("Save").click()
        self.browser.getControl("Finally Submit").click()
        self.assertTrue(
            'Application submitted' in self.browser.contents)
        # Create 10 applicants who already selected utp0011
        for i in range(1,11):
            dummy = createObject(u'waeup.Applicant')
            dummy.firstname = u'John'
            dummy.lastname = u'Doe %s' %i
            dummy.school_utp = 'utp0011'
            self.app['applicants'][
                self.utpcontainer.code].addApplicant(dummy)
            IWorkflowState(dummy).setState('submitted')
        IWorkflowState(self.utpapplicant).setState('paid')
        self.utpapplicant.locked = False
        self.browser.open(self.utpapplicant_path + '/edit')
        self.browser.getControl(name="form.school_utp").value = ['utp0011']
        self.browser.getControl("Finally Submit").click()
        self.assertTrue("Maximum number of applications per school exceeded."
            in self.browser.contents)
        self.assertEqual(self.utpapplicant.state, 'paid')
        # Create second certificate
        certificate2 = createObject('waeup.Certificate')
        certificate2.code = 'BEDCERT2'
        certificate2.title = 'BEDCert2'
        certificate2.application_category = 'basic'
        certificate2.start_level = 100
        certificate2.end_level = 500
        certificate2.study_mode = u'nce_ft'
        self.app['faculties']['fac1']['dep1'].certificates.addCertificate(
            certificate2)
        # Create one applicant with same subject and same school
        dummy = createObject(u'waeup.Applicant')
        dummy.firstname = u'John'
        dummy.lastname = u'Doe %s' %i
        dummy.school_utp = 'utp0012'
        dummy.subj_comb = certificate1
        self.app['applicants'][
            self.utpcontainer.code].addApplicant(dummy)
        IWorkflowState(dummy).setState('submitted')
        self.browser.open(self.utpapplicant_path + '/edit')
        self.browser.getControl(name="form.school_utp").value = ['utp0012']
        self.browser.getControl(name="form.subj_comb").value = ['BEDCERT1']
        self.browser.getControl("Finally Submit").click()
        # UTP applications allow two, one is not enough
        self.assertEqual(self.utpapplicant.state, 'submitted')
        return