import unittest
from sqlalchemy.orm.exc import MultipleResultsFound
from datetime import datetime

from iro.model.schema import User, Offer, Userright, Job, Message
from decimal import Decimal

from .dbtestcase import DBTestCase

class DBTests(DBTestCase):
    """tests for the db model"""
    
    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 testProviders(self):
        '''test providers'''
        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.providers('sms').all(),[('sipgate',),])

    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 Has_RightTests(DBTestCase):
    '''test User.has_right'''
    def testSimple(self):
        '''test a very simple case'''
        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.has_right("sms"),"sipgate_basic")
            self.failUnlessEqual(u.has_right("sms", offer_name="sipgate_basic"),"sipgate_basic")
            self.failUnlessEqual(u.has_right("sms", route="basic"),"sipgate_basic")
            self.failUnlessEqual(u.has_right("sms", provider="sipgate"),"sipgate_basic")

    def testDouble(self):
        '''now we have different routes'''
        with self.session() as session:
            u=User(name='test',apikey='abcdef123456789')
            o=Offer(name="b", provider="sipgate", route="basic", typ="sms")
            u.rights.append(Userright(o)) 
            o=Offer(name="c", provider="sipgate", route="c", typ="sms")
            u.rights.append(Userright(o)) 
            session.add(u)
        
        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.has_right("sms", offer_name="b"),"b")
            self.failUnlessEqual(u.has_right("sms", route="basic"),"b")
            self.failUnlessEqual(u.has_right("sms", offer_name="c"),"c")
            self.failUnlessEqual(u.has_right("sms", route="c"),"c")
            self.assertRaises(MultipleResultsFound, u.has_right,"sms")
            self.assertRaises(MultipleResultsFound, u.has_right,"sms", provider="sipgate")
    
    def testUnknown(self):
        ''' a unknown typ'''
        with self.session() as session:
            u=User(name='test',apikey='abcdef123456789')
            o=Offer(name="b", provider="sipgate", route="basic", typ="sms")
            u.rights.append(Userright(o)) 
            o=Offer(name="c", provider="sipgate", route="c", typ="sms")
            u.rights.append(Userright(o)) 
            session.add(u)
        
        with self.session() as session:
            u=session.merge(u)
            self.failUnlessEqual(u.has_right("fax", offer_name="b"), None)
            self.failUnlessEqual(u.has_right("fax"), None)



class BillTest(DBTestCase):
    """test the bill function"""
    
    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(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(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(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(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(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(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(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()