# -*- coding: utf-8 -*-

import unittest
import server
import threading, xmlrpclib
from multiprocessing import Queue
from multiprocessing.managers import BaseManager

class StoppableXMLRPCServer(server.SecureUserDBXMLRPCServer, threading.Thread):
    running=True
    def __init__(self, *args, **kwargs):
        server.SecureUserDBXMLRPCServer.__init__(self, *args, **kwargs)
        threading.Thread.__init__(self)
   
   
    def run(self):
        # *serve_forever* muss in einem eigenen Thread laufen, damit man es
        # unterbrechen kann!
        while (self.running):
            self.handle_request()
   
   
    def stop(self):
        self.running=False
        self.server_close()


def init_server():
    userlist=[{"name":"test","password":"test",  "class":server.User},
              {"name":"test2","password":"test2", "class": server.User},
	       {"name":"admin","password":"admin", "class": server.Admin}]


    
    class MyManager(BaseManager):
        pass
    
    MyManager.register('MessageJob', server.MessageJob) 
    manager = MyManager()
    manager.start()
    
    
    #anbieter erzeugen und konfigurieren
    import anbieter
    sip=anbieter.sipgate()
    sip.read_basic_config("iro.conf")
    localhost=""

    #Benutzerdatenbank erstellen
    queue = Queue()
    provider={"sms":sip, "fax":sip, "mail":localhost}
    jobqueue=server.Jobs(manager,  queue, provider)
    userdb=server.MyUserDB(userlist,jobqueue)


    #Server starten
    serv = StoppableXMLRPCServer(addr=("localhost", 8000), 
                                      userdb=userdb,
                                      certificate="./certs/test.cert.pem",privatekey="./certs/test.key.pem")
    serv.relam="xmlrpc"
    return serv
 
    
class TestServer(unittest.TestCase):
    
    def setUp(self):
      self.serv = init_server()
      self.serv.start()

    def tearDown(self):
      self.serv.stop()
      xmlrpclib.Server("https://test:test@localhost:8000").status()			#letzte nachricht abrufen, damit richt geschlossen wird

    def testLogin(self):
        self.assertEqual(xmlrpclib.Server("https://test:test@localhost:8000").status(), {})
        self.assertEqual(xmlrpclib.Server("https://test2:test2@localhost:8000").status(), {})
        self.assertRaises(xmlrpclib.ProtocolError, xmlrpclib.Server("https://test2:test@localhost:8000").status)
        self.assertRaises(xmlrpclib.ProtocolError,xmlrpclib.Server ("https://test:test2@localhost:8000").status)
    
    def testsendSMS(self):
	servstr="https://test:test@localhost:8000"
	client=xmlrpclib.Server(servstr)
	id=client.startSMS("test",["01234"] )
        self.assertEqual(client.status(id),{id: {'status': ['init',{}], 'name':  {'content': 'test'}}} )
        
    def testTwoUser(self):
 	u1="https://test:test@localhost:8000"
	u2="https://test2:test2@localhost:8000"
	admin="https://admin:admin@localhost:8000"
	client1=xmlrpclib.Server(u1)
	client2=xmlrpclib.Server(u2)
	admin=xmlrpclib.Server(admin)
	id1=client1.startSMS("test",["01234"] )
	self.assertEqual(client2.status(),{} ) 
	self.assertEqual(admin.status(id1),{id1: {'status': ['init',{}], 'name':  {'content': 'test'}}} )
	id2=client2.startSMS("test2",["01234"] )
	self.assertNotEqual(id1, id2)
	self.assertEqual(client1.status(),{id1: {'status': ['init',{}], 'name':  {'content': 'test'}}})
	self.assertEqual(client2.status(),{id2: {'status': ['init',{}], 'name':   {'content': 'test2'}}})
	self.assertEqual(admin.status(),{id1: {'status': ['init',{}], 'name':  {'content': 'test'}},
					id2: {'status': ['init',{}], 'name':   {'content': 'test2'}}} )
	
	self.assertRaises(xmlrpclib.Fault, client2.status, id1)
        self.assertRaises(xmlrpclib.Fault, client1.status, id2)
    
if __name__ == "__main__":
    unittest.main()