class TestSiteStatistics(unittest.TestCase):
    
    def setUp(self):
        self.obj = SiteStatistics()
         
    def tearDown(self):
        del self.obj

    def test_key(self):
        domain_name = 'test_name'
        correct_timestamp = '20110101'
        self.obj.composite_key(domain_name, correct_timestamp)
        key = self.obj.get_key()
        assert key[0] == domain_name
        assert key[1] == correct_timestamp

    def test_os(self):
        key_1 = 'Windows MS PS 7.0.0.0.1.1.2'
        value_1 = 44
        key_2 = 'Windows MS XP SP 3'
        value_2 = 11
        
        doc = self.obj.get_os()
        doc[key_1] = value_1
        doc[key_2] = value_2
        self.obj.set_os(doc)
        
        doc = self.obj.get_os()
        assert doc[key_1] == value_1
        assert doc[key_2] == value_2

    def test_browsers(self):
        key_1 = 'FireFox 2.3.4.5.6'
        value_1 = 22
        key_2 = 'MS IE 9.9.9.9.9'
        value_2 = 11
        
        doc = self.obj.get_browsers()
        doc[key_1] = value_1
        doc[key_2] = value_2
        self.obj.set_browsers(doc)
        
        doc = self.obj.get_browsers()
        assert doc[key_1] == value_1
        assert doc[key_2] == value_2

    def test_number_of_visits(self):
        value = 100
        self.obj.set_number_of_visits(value)
        t = self.obj.get_number_of_visits()
        assert t == value

    def test_total_duration(self):
        value = 100
        self.obj.set_total_duration(value)
        t = self.obj.get_total_duration()
        assert t == value

    def test_screen_res(self):
        key_1 = (1024, 768)
        value_1 = 2000
        key_2 = (360, 240)
        value_2 = 98765
        
        doc = self.obj.get_screen_res()
        doc[key_1] = value_1
        doc[key_2] = value_2
        self.obj.set_screen_res(doc)
        
        doc = self.obj.get_screen_res()
        assert doc[key_1] == value_1
        assert doc[key_2] == value_2

    def test_languages(self):
        key_1 = 'en_ca'
        value_1 = 87878
        key_2 = 'ua_uk'
        value_2 = 98765
        
        doc = self.obj.get_languages()
        doc[key_1] = value_1
        doc[key_2] = value_2
        self.obj.set_languages(doc)
        
        doc = self.obj.get_languages()
        assert doc[key_1] == value_1
        assert doc[key_2] == value_2

    def test_countries(self):
        key_1 = 'ca'
        value_1 = 87878
        key_2 = 'uk'
        value_2 = 98765
        
        doc = self.obj.get_countries()
        doc[key_1] = value_1
        doc[key_2] = value_2
        self.obj.set_countries(doc)
        
        doc = self.obj.get_countries()
        assert doc[key_1] == value_1
        assert doc[key_2] == value_2
 def _init_target_object(self, composite_key):
     """ abstract method to instantiate new object that will be holding aggregated data """
     obj = SiteStatistics()
     obj.composite_key(composite_key[0], composite_key[1])
     obj.set_number_of_visits(0)
     return obj
 def setUp(self):
     self.obj = SiteStatistics()