class MongoDbTest(unittest.TestCase):
def setUp(self):
self.db = MongoDb()
self.db.insert_or_update({
'cep': 'UNIQUE_KEY',
'logradouro': 'A',
'bairro': 'A',
'cidade': 'A',
'estado': 'A'
})
def test_remove_empty_fields(self):
'''
Quando um registro é atualizado no banco de dados,
as chaves inexistentes devem ser removidas.
'''
self.db.insert_or_update({'cep': 'UNIQUE_KEY', 'estado': 'B'})
result = self.db.get_one('UNIQUE_KEY')
self.assertEqual(result['estado'], 'B')
self.assertNotIn('logradouro', result)
self.assertNotIn('bairro', result)
self.assertNotIn('cidade', result)
def tearDown(self):
self.db.remove('UNIQUE_KEY')
class MongoDbTest(unittest.TestCase):
def setUp(self):
self.db = MongoDb()
self.db.insert_or_update({
'cep': 'UNIQUE_KEY',
'logradouro': 'A',
'bairro': 'A',
'cidade': 'A',
'estado': 'A'
})
def test_remove_empty_fields(self):
'''
Quando um registro é atualizado no banco de dados,
as chaves inexistentes devem ser removidas.
'''
self.db.insert_or_update({
'cep': 'UNIQUE_KEY',
'estado': 'B'
})
result = self.db.get_one('UNIQUE_KEY')
self.assertEqual(result['estado'], 'B')
self.assertNotIn('logradouro', result)
self.assertNotIn('bairro', result)
self.assertNotIn('cidade', result)
def tearDown(self):
self.db.remove('UNIQUE_KEY')
def test_insert_notfound(self):
db = MongoDb()
cep = u'11111111'
db.remove(cep)
db.insert_or_update({
u'cep': cep,
u'_meta': {
u'v_date': 'v_date',
CepTracker._notfound_key: True,
}
})
expected = {
u'cep': cep,
u'estado': u'SP',
u'_meta': {
u'v_date': 'v_date',
}
}
db.insert_or_update(expected)
result = db.get_one(cep, fields={'_id': False})
self.assertEqual(expected, result)
def test_insert_notfound(self):
db = MongoDb()
cep = u'11111111'
db.remove(cep)
db.insert_or_update({
u'cep': cep,
u'_meta': {
u'v_date': 'v_date',
CepTracker._notfound_key: True,
}
})
expected = {
u'cep': cep,
u'estado': u'SP',
u'_meta': {
u'v_date': 'v_date',
}
}
db.insert_or_update(expected)
result = db.get_one(cep, fields={'_id': False})
self.assertEqual(expected, result)
