test_cache.py 822 Bytes
Newer Older
William Tisäter committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# -*- 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):
15 16 17
        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')
William Tisäter committed
18

19 20 21
        self.assertEqual(code_mmap, 'US')
        self.assertEqual(code_memory, 'US')
        self.assertEqual(code_standard, 'US')