from Queue import deque

from iro.controller.task import createJob, Task

from iro.controller.pool  import taskPool

from iro.model.job import exJobs, ExJob

from iro.model.schema import Job, User

from iro.validate import vInteger

from .dbtestcase import DBTestCase

class exJobsTest(DBTestCase):

    '''tests for exJobs'''

    def setUp(self):

        DBTestCase.setUp(self)

        self.pool = data.pool

        data.pool = DumpPool()

        exJobs.clear()

        data.pool = self.pool

        self.pool = None

        job = exJobs.create(u, [Telnumber('123456789')], SMS('test'), ['test'])

        self.assertIsInstance(job, ExJob)

        self.assertTrue(vInteger(job.dbjob, None, minv=0 ))

        self.assertEqual(job.message, SMS('test'))

        self.assertEqual(job.recipients,  [Telnumber('123456789')])

        self.assertEqual(job.offers,['test'])

        self.assertEqual(job.tasks,{})

            j = session.query(Job.id).all()

            self.assertEqual(j,[(job.dbjob,)])

        self.assertEqual(exJobs[job.dbjob],job)

        job = ExJob(None, [Telnumber('123456789')], SMS('test'), ['test'])

        exJobs[1] = job

        self.assertEqual(len(exJobs), 1)

        self.assertEqual(job, exJobs[1])

            job = Job( info="info", status="started")

            ejob= ExJob(job.id, [Telnumber('123456789')], SMS('test'), ['test'])

            exJobs[job.id]=ejob

            self.assertEqual(job.extend, ejob)

            self.assertEqual(u.jobs[0].extend,ejob)

    def testUnknownExJob(self):

        self.assertRaises(KeyError,exJobs.__getitem__,'a1234567890')

    def setUp(self):

        DBTestCase.setUp(self)

        self.pool = data.pool

        data.pool = DumpPool()

    def tearDown(self):

        exJobs.clear()

        data.pool = self.pool

        self.pool = None

        taskPool.pool.q.queue = deque()

        DBTestCase.tearDown(self)

        with self.session() as session:

            u = User(name='test',apikey='abcdef123456789')

            session.add(u)

        exjob = createJob(u,[Telnumber('0123325456')],SMS('sms'),[])

        self.assertEqual(taskPool.pool.q.qsize(),1)

        self.assertEqual(exjob.tasks.keys(),[Telnumber('0123325456')])

        self.assertIsInstance(exjob.tasks[Telnumber('0123325456')], Task)

        with self.session() as session:

            u = User(name='test',apikey='abcdef123456789')

            session.add(u)

        job = createJob(u,[Telnumber('0123325456')],Fax('header','fax',[]),[])

        self.assertEqual(taskPool.pool.q.qsize(),1)

        with self.session() as session:

            u = User(name='test',apikey='abcdef123456789')

            session.add(u)

        job = createJob(u,[],Mail('sub','body','t@t.de'),[])