def test_delete(request, populate):
conf = request.config
conn = TessMongoConnection(conf.getoption('db_host'),
conf.getoption('db_port'),
conf.getoption('db_user'),
password=conf.getoption('db_passwd',
default=None),
db=conf.getoption('db_name', default=None))
print(conn.connection.name)
for key, val in populate.items():
entity = entity_map[key]
ents = [entity(**entry) for entry in val]
# Test deleting one entry
result = conn.delete([ents[0]])
assert result.deleted_count == 1
assert conn.connection[key].count_documents({'_id': ents[0].id}) == 0
found = conn.connection[key].find()
print(list(found))
assert conn.connection[key].count_documents({}) == len(val) - 1
# Test deleting the collection's entries
result = conn.delete(ents)
assert result.deleted_count == len(val) - 1
# Ensure that the collection is empty
found = conn.connection[key].find()
assert conn.connection[key].count_documents({}) == 0
def test_insert(self, request, populate, entity_map):
conf = request.config
conn = TessMongoConnection(conf.getoption('db_host'),
conf.getoption('db_port'),
conf.getoption('db_user'),
password=conf.getoption('db_passwd',
default=None),
db=conf.getoption('db_name', default=None))
for entity_type in populate:
if len(populate[entity_type]) == 0:
continue
end = \
len(populate[entity_type]) if len(populate[entity_type]) < 100 \
else 100
entities = [
entity_map[entity_type].json_decode(e)
for e in populate[entity_type][:end]
]
conn.delete(entities)
for entity in entities:
res = conn.insert(entity)
assert len(res.inserted_ids) == 1
def test_delete(self, request, populate, entity_map):
conf = request.config
conn = TessMongoConnection(conf.getoption('db_host'),
conf.getoption('db_port'),
conf.getoption('db_user'),
password=conf.getoption('db_passwd',
default=None),
db=conf.getoption('db_name', default=None))
for entity_type in populate:
if len(populate[entity_type]) == 0:
continue
end = \
len(populate[entity_type]) if len(populate[entity_type]) < 100 \
else 100
for query_ent in populate[entity_type][:end]:
entity = entity_map[entity_type].json_decode(query_ent)
res = conn.delete(entity)
found = conn.find(entity_type,
**entity.json_encode(exclude=['_id']))
assert res.deleted_count == 1
assert len(found) == 0
conn.insert(entity)