import time
import pykolab

from pykolab import wap_client
from pykolab.auth import Auth
from pykolab.imap import IMAP
from twisted.trial import unittest

import tests.functional.resource_func as funcs

conf = pykolab.getConf()


class TestResourceAdd(unittest.TestCase):

    @classmethod
    def setUp(self):
        from tests.functional.purge_users import purge_users
        purge_users()

        self.john = {
            'local': 'john.doe',
            'domain': 'example.org'
        }

        from tests.functional.user_add import user_add
        user_add("John", "Doe")

        funcs.purge_resources()
        self.audi = funcs.resource_add("car", "Audi A4")
        self.passat = funcs.resource_add("car", "VW Passat")
        self.boxter = funcs.resource_add("car", "Porsche Boxter S")
        self.cars = funcs.resource_add("collection", "Company Cars", [self.audi['dn'], self.passat['dn'], self.boxter['dn']])

        from tests.functional.synchronize import synchronize_once
        synchronize_once()

    def test_001_resource_created(self):
        auth = Auth()
        auth.connect()
        resource = auth.find_resource(self.audi['mail'])
        self.assertEqual(resource, self.audi['dn'])

        collection = auth.find_resource(self.cars['mail'])
        self.assertEqual(collection, self.cars['dn'])

    def test_002_resource_collection(self):
        auth = Auth()
        auth.connect()
        attrs = auth.get_entry_attributes(None, self.cars['dn'], ['*'])
        self.assertIn('groupofuniquenames', attrs['objectclass'])
        self.assertEqual(len(attrs['uniquemember']), 3)