# -*- coding: utf-8 -*-
import unittest

import pygeoip
from pygeoip import MEMORY_CACHE, MMAP_CACHE, STANDARD
from tests.config import COUNTRY_DB_PATH

class TestGeoIPCacheMethods(unittest.TestCase):
    def setUp(self):
        self.gi_mmap = pygeoip.GeoIP(COUNTRY_DB_PATH, MMAP_CACHE)
        self.gi_memory = pygeoip.GeoIP(COUNTRY_DB_PATH, MEMORY_CACHE)
        self.gi_standard = pygeoip.GeoIP(COUNTRY_DB_PATH, STANDARD)

    def testCountryCodeByAddr(self):
        code_mmap = self.gi_mmap.country_code_by_addr('64.17.254.216')
        code_memory = self.gi_memory.country_code_by_addr('64.17.254.216')
        code_standard = self.gi_standard.country_code_by_addr('64.17.254.216')

        self.assertEqual(code_mmap, 'US')
        self.assertEqual(code_memory, 'US')
        self.assertEqual(code_standard, 'US')