示例#1
0
class CidadeTest(unittest.TestCase):
    cidade_sp = {
        'sigla_uf_nome_cidade': u'SP_SAO PAULO',
        'area_km2': '1099',
    }

    def setUp(self):
        self.db = MongoDb()
        self.db.insert_or_update_cidade(self.cidade_sp)

    def tearDown(self):
        self.db._db.ufs.remove()

    def test_get(self):
        result = self.db.get_one_cidade(u'SP', u'São Paulo')
        for key, expected in self.cidade_sp.items():
            self.assertEqual(expected, result[key])

    def test_get_alt(self):
        result = self.db.get_one_cidade(u'SP', u'Outro lugar (São Paulo)')
        self.assertEqual('1099', result['area_km2'])

    def test_update(self):
        self.db.insert_or_update_cidade({
            'sigla_uf_nome_cidade': u'SP_SAO PAULO',
            'area_km2': '2000',
        })

        result = self.db.get_one_cidade(u'SP', u'São Paulo')
        self.assertEqual('2000', result['area_km2'])
示例#2
0
class CidadeTest(unittest.TestCase):
    cidade_sp = {
        'sigla_uf_nome_cidade': u'SP_SAO PAULO',
        'area_km2': '1099',
    }

    def setUp(self):
        self.db = MongoDb()
        self.db.insert_or_update_cidade(self.cidade_sp)

    def tearDown(self):
        self.db._db.ufs.remove()

    def test_get(self):
        result = self.db.get_one_cidade(u'SP', u'São Paulo')
        for key, expected in self.cidade_sp.items():
            self.assertEqual(expected, result[key])

    def test_get_alt(self):
        result = self.db.get_one_cidade(u'SP', u'Outro lugar (São Paulo)')
        self.assertEqual('1099', result['area_km2'])

    def test_update(self):
        self.db.insert_or_update_cidade({
            'sigla_uf_nome_cidade': u'SP_SAO PAULO',
            'area_km2': '2000',
        })

        result = self.db.get_one_cidade(u'SP', u'São Paulo')
        self.assertEqual('2000', result['area_km2'])