import unittest
from sqlalchemy import create_engine, pool
from tempfile import mkdtemp
import shutil

from datetime import datetime

from iro.model import POOL_SIZE as DB_POOL_SIZE
from iro.model.schema import User, Base, Offer, Userright, Job, Message
from decimal import Decimal

from ngdatabase.mysql import Server, createConfig, Database
from .dbtestcase import DBTestCase

class SampleDatabase(Database):
    def createPassword(self):
        self.password="test"
        return self.password

class ModuleData:
    def __init__(self):
        self.tdir = mkdtemp(prefix='iro-mysql-')
        self.server = Server('%s/my.cnf'%self.tdir)
        self.db = SampleDatabase("test","test",'%s/my.cnf'%self.tdir)
        self.engine = create_engine('mysql://test:test@localhost/test?unix_socket=%s/socket'%self.tdir,
                poolclass = pool.SingletonThreadPool,  pool_size=DB_POOL_SIZE, )#echo=True)

    def setUp(self):
        with open('%s/my.cnf'%self.tdir,'w') as cnf:
            cnf.write(createConfig(self.tdir))
        self.server.create()
        self.server.start()
        self.db.create()
        Base.metadata.create_all(self.engine)
    
    def tearDown(self):
        self.server.stop()
        shutil.rmtree(self.tdir)
 

md=ModuleData()

def setUpModule():
    md.setUp()

def tearDownModule():
    md.tearDown()

class DBTests(DBTestCase):
    """tests for the db model"""
    def setUp(self):
        if not self.engine:
            self.engine = md.engine
    
    def testRoutes(self):
        '''test routes'''
        with self.session() as session:
            u=User(name='test',apikey='abcdef123456789')
            o=Offer(name="sipgate_basic", provider="sipgate", route="basic", typ="sms")
            u.rights.append(Userright(o)) 
            session.add(u)
        
        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.routes('sms').all(),[('sipgate_basic',),])


    def testTyps(self):
        with self.session() as session:
            o=Offer(name="sipgate_basic", provider="sipgate", route="basic", typ="sms")
            session.add(o)
        
        with self.session() as session:
            self.failUnlessEqual(session.typs.all(),[('sms',),])

        with self.session() as session:
            o=Offer(name="s2", provider="sipgate", route="basic", typ="sms")
            o=Offer(name="s3", provider="sipgate", route="basic", typ="sms2")
            session.add(o)
        
        with self.session() as session:
            self.failUnlessEqual(session.typs.order_by(Offer.typ).all(),[('sms',),('sms2',)])


class BillTest(DBTestCase):
    """test the bill function"""
    def setUp(self):
        if not self.engine:
            self.engine = md.engine

    def testBill(self):
        '''test bill function'''
        apikey='abcdef123456789'
        with self.session() as session:
            u=User(name='test',apikey=apikey)
            session.add(u)
            o = Offer(name='sipgate_basic',provider="sipgate",route="basic",typ="sms")
            j = Job(hash='a1',info='i',status='sended')
            m = Message(recipient='0123456789', isBilled=False, date=datetime.now() , price=0.30, offer=o, job=j)
            u.rights.append(Userright(o))
            u.jobs.append(j)
            session.add(m)

        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.rights[0].bill.all(),[(1L,Decimal('0.3000'),'i')])

    def testBillAgreget(self):
        apikey='abcdef123456789'
        with self.session() as session:
            u=User(name='test',apikey=apikey)
            session.add(u)
            o = Offer(name='sipgate_basic',provider="sipgate",route="basic",typ="sms")
            j = Job(hash='a1',info='i',status='sended')
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now() , price=0.30, offer=o))
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now() , price=0.4, offer=o))
            u.rights.append(Userright(o))
            u.jobs.append(j)
 
        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.rights[0].bill.all(),[(2L,Decimal('0.7000'),'i')])
        
    def testBillIsBilled(self):
        apikey='abcdef123456789'
        with self.session() as session:
            u=User(name='test',apikey=apikey)
            session.add(u)
            o = Offer(name='sipgate_basic',provider="sipgate",route="basic",typ="sms")
            j = Job(hash='a1',info='i',status='sended')
            j.messages.append(Message(recipient='0123456789', isBilled=True, date=datetime.now() , price=0.30, offer=o))
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now() , price=0.4, offer=o))
            u.rights.append(Userright(o))
            u.jobs.append(j)

        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.rights[0].bill.all(),[(1L,Decimal('0.4000'),'i')])

    def testBillMultipleJobs(self):
        apikey='abcdef123456789'
        with self.session() as session:
            u=User(name='test',apikey=apikey)
            session.add(u)
            o = Offer(name='sipgate_basic',provider="sipgate",route="basic",typ="sms")
            u.rights.append(Userright(o))
            
            j = Job(hash='a1',info='i',status='sended')
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now() , price=0.4, offer=o))
            u.jobs.append(j)

            j = Job(hash='a2',info='a',status='sended')
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now(), price=0.4, offer=o))
            u.jobs.append(j) 
        
        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.rights[0].bill.order_by(Job.info).all(),
                    [(1L,Decimal('0.4000'),'a'),
                    (1L,Decimal('0.4000'),'i')
                    ])

    def testBillMultipleOffers(self):
        apikey='abcdef123456789'
        with self.session() as session:
            u=User(name='test',apikey=apikey)
            session.add(u)
            o = Offer(name='sipgate_basic',provider="sipgate",route="basic",typ="sms")
            u.rights.append(Userright(o))
            
            j = Job(hash='a2',info='a',status='sended')
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now(), price=0.4, offer=o))
            u.jobs.append(j) 
 
            o = Offer(name='sipgate_gold',provider="sipgate",route="gold",typ="sms")
            u.rights.append(Userright(offer=o))
            
            j = Job(hash='a3',info='a',status='sended')
            j.messages.append(Message(recipient='0123456789', isBilled=False, date=datetime.now(), price=0.5, offer=o))
            u.jobs.append(j) 
       
        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.rights[0].bill.all(),
                    [(1L,Decimal('0.4000'),'a')])
        
            self.failUnlessEqual(u.rights[1].bill.all(),[(1L,Decimal('0.5000'),'a')])

if __name__ == '__main__':
        unittest.main()