def test_auto_sync(self):
# test documents with ids already in place
with EssentialDB(filepath=SYNC_DB_FILE).get_collection() as db:
docs = self.docs
db.insert_many(docs)
with EssentialDB(filepath=SYNC_DB_FILE).get_collection() as db2:
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
def test_auto_sync(self):
# test documents with ids already in place
db = EssentialDB( filepath=SYNC_DB_FILE, autosync=True )
docs = _gen_docs(10)
db.insert_many(docs)
del(db)
db2 = EssentialDB( filepath=SYNC_DB_FILE)
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
def test_sync_load(self):
docs = self.docs
db = EssentialDB(filepath=SYNC_DB_FILE).get_collection()
db.insert_many(docs)
db.sync()
del (db)
db2 = EssentialDB(filepath=SYNC_DB_FILE).get_collection()
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
def _init_db(self):
#first, see if already exists
if not os.path.isfile(self.db_path):
print("creating db...")
from quotes import initial_quotes
with EssentialDB(filepath=self.db_path) as db:
db.set("quotes", initial_quotes)
def setUp(self):
self.collection = EssentialDB().get_collection()
generator = DocumentGenerator()
def make_nested():
nested_gen = DocumentGenerator()
nested_gen.set_template({"n0": 'word', "n1": 'word', "n2": 10})
return nested_gen.gen_doc()
template = {
"field 0": 'word',
"field 1": 'word',
"field 2": 'word',
"number": 10,
"nested": make_nested
}
generator.set_template(template)
self.docs = generator.gen_docs(10)
async def weather(self, request):
with EssentialDB(filepath=self.db_path) as db:
weather = db.get("weather")
do_update = False
try:
delta = datetime.datetime.now() - weather["updated"]
if delta > datetime.timedelta(
minutes=mm_config["weather"]["update_minutes"]):
do_update = True
except:
# probably not an error - just no weather update yet.
do_update = True
pass
if do_update:
new_weather = weather_connector.get_weather()
db.set("weather", new_weather)
weather = new_weather
return {'weather': weather}
class EssentialPrompt(Cmd):
# def setup(self):
# self.db = None
def do_use(self, args):
self.db = EssentialDB(filepath=args.strip())
print(self.db)
def do_collection(self, args):
self.collection = self.db.get_collection(args)
def do_find(self, args):
pprint.pprint(self.collection.find())
def do_show(self, args):
if args == "collections":
print("Collections:")
print([c for c in self.db.collections])
def do_quit(self, args):
"""Quits the program."""
print("Quitting.")
raise SystemExit
async def quote(self, request):
with EssentialDB(filepath=self.db_path) as db:
quote = random.choice(db.get("quotes"))
return {'quote': quote}
def __init__(self, filepath):
self.db = EssentialDB(filepath=filepath)
self.db.sync()
class EssentialTropicsStorage:
def __init__(self, filepath):
self.db = EssentialDB(filepath=filepath)
self.db.sync()
def profile(self, id=None, login=None):
if id:
return self._profiles().get(id)
if login:
return self._profiles().find_one({'login': login})
def profiles(self):
return self._profiles().find()
def store_profile(self, profile):
with self._profiles() as profiles:
return profiles.insert_one(profile)
def store_profiles(self, profiles):
with self._profiles() as profile_store:
return profile_store.insert_many(profiles)
def remove_profile(self, profile_id):
with self._profiles() as profiles:
return profiles.remove({'_id': profile_id})
def credential(self, id=None, login=None):
if id:
return self._credentials().get(id)
elif login:
return self._credentials().find_one({'login': login})
return None
def store_credential(self, credential):
with self._credentials() as credential_collection:
credential_collection.insert_one(credential)
def remove_credential(self, id=None, login=None):
with self._credentials() as credential_collection:
if id:
return credential_collection.remove({'_id': id})
elif login:
return credential_collection.remove({'login': login})
return None
def session(self, token=None, login=None, profile_id=None):
if token:
return self._sessions().get(token)
elif profile_id:
return self._sessions().find_one({'profile_id': profile_id})
else:
return self._sessions().find_one({'login': login})
def store_session(self, session):
with self._sessions() as credential_collection:
return credential_collection.insert_one(session)
def remove_session(self, token):
with self._sessions() as session_collection:
return session_collection.remove({'_id': token})
def _profiles(self):
return self.db.get_collection("profiles")
def _credentials(self):
return self.db.get_collection("credentials")
def _sessions(self):
return self.db.get_collection("sessions")
def _reset_all(self, seriously):
if seriously:
with self._profiles() as profiles:
profiles.remove()
with self._credentials() as credentials:
credentials.remove()
return True
return False
class TestEssentialDBActions(unittest.TestCase):
def setUp(self):
self.collection = EssentialDB().get_collection()
generator = DocumentGenerator()
def make_nested():
nested_gen = DocumentGenerator()
nested_gen.set_template({"n0": 'word', "n1": 'word', "n2": 10})
return nested_gen.gen_doc()
template = {
"field 0": 'word',
"field 1": 'word',
"field 2": 'word',
"number": 10,
"nested": make_nested
}
generator.set_template(template)
self.docs = generator.gen_docs(10)
def test_set_get(self):
self.collection.set("my-test-number", 1)
self.collection.set("my-test-string", "Hello")
num = self.collection.get("my-test-number")
text = self.collection.get("my-test-string")
self.assertEqual(num, 1)
self.assertEqual(text, "Hello")
# now, try get before set
not_set = self.collection.get("doesn't exist")
self.assertIsNone(not_set)
def test_insert_and_find_one(self):
self.collection.insert_one(self.docs[0])
response = self.collection.find_one()
self.assertEqual(response["field 1"], self.docs[0]["field 1"])
self.assertEqual(self.collection.count(), 1)
def test_insert_many(self):
self.collection.insert_many(self.docs)
self.assertEqual(self.collection.count(), 10)
def test_find_one_of_many_by_id(self):
self.collection.insert_many(self.docs)
response = self.collection.find_one({"_id": self.docs[0]["_id"]})
self.assertEqual(response["_id"], self.docs[0]["_id"])
response = self.collection.find_one(self.docs[0]["_id"])
self.assertEqual(response["_id"], self.docs[0]["_id"])
def test_find_one_complex_query(self):
self.collection.insert_many(self.docs)
response = self.collection.find_one({
"field 0": self.docs[5]["field 0"],
"field 1": self.docs[5]["field 1"]
})
self.assertEqual(response["_id"], self.docs[5]["_id"])
def test_find_one_complex_dot_query(self):
self.collection.insert_many(self.docs)
response = self.collection.find_one({
"nested.n0":
self.docs[5]["nested"]["n0"],
"nested.n1":
self.docs[5]["nested"]["n1"]
})
self.assertEqual(response["_id"], self.docs[5]["_id"])
def test_find_one_filter(self):
def filter(doc):
if doc["field 0"] == "Hello":
return True
return False
self.docs[5]["field 0"] = "Hello"
self.collection.insert_many(self.docs)
response = self.collection.find(filter=filter)
self.assertEqual(response[0]["_id"], self.docs[5]["_id"])
def test_find_or(self):
self.collection.insert_many(self.docs)
q = {
"$or": [{
"field 0": self.docs[5]["field 0"]
}, {
"field 1": self.docs[6]["field 1"]
}]
}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_find_nor(self):
self.collection.insert_many(self.docs)
q = {
"$nor": [{
"field 1": self.docs[0]["field 1"]
}, {
"field 1": self.docs[1]["field 1"]
}]
}
response = self.collection.find(q)
self.assertEqual(len(response), 8)
def test_find_and(self):
self.collection.insert_many(self.docs)
q = {
"field 1": self.docs[0]["field 1"],
"field 2": self.docs[0]["field 2"]
}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_find_or_complex(self):
self.docs[0]["number"] = 0
self.docs[1]["number"] = 5
self.docs[2]["number"] = 20
self.docs[3]["number"] = 25
self.collection.insert_many(self.docs)
q = {"$or": [{"number": {"$eq": 5}}, {"number": {"$eq": 20}}]}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_find_nor_complex(self):
self.docs[0]["number"] = 0
self.docs[1]["number"] = 5
self.collection.insert_many(self.docs)
q = {"$nor": [{"number": {"$eq": 10}}, {"number": {"$eq": 0}}]}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_find_and_complex(self):
self.docs[0]["number"] = 0
self.docs[0]["field 0"] = "Hello"
self.docs[5]["number"] = 2
self.docs[5]["field 0"] = "Hello"
self.docs[7]["number"] = 4
self.docs[7]["field 0"] = "Bye"
self.collection.insert_many(self.docs)
q = {"$and": [{"number": {"$lt": 2}}, {"field 0": {"$eq": "Hello"}}]}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_find_not(self):
self.docs[0]["field 0"] = "Hello"
self.docs[0]["field 1"] = "Goodbye"
self.collection.insert_many(self.docs)
q = {"$not": [{'field 0': "Hello"}]}
response = self.collection.find(q)
self.assertEqual(len(response), 9)
q = {"$not": [{'field 0': "Hello", "field 1": "Goodbye"}]}
response = self.collection.find(q)
self.assertEqual(len(response), 9)
q = {"$not": [{'field 0': "Hello", "field 1": "Doesnt Exist"}]}
response = self.collection.find(q)
self.assertEqual(len(response), 10)
def test_update_one(self):
self.collection.insert_many(self.docs)
self.collection.update(
{
"field 0": self.docs[5]["field 0"],
"field 1": self.docs[5]["field 1"]
}, {"field 2": "Hello"})
find = self.collection.find_one({"field 2": "Hello"})
self.assertEqual(find["_id"], self.docs[5]["_id"])
def test_remove_one(self):
self.collection.insert_many(self.docs)
self.collection.remove({'fields 0': self.docs[5]["field 0"]})
results = self.collection.find_one(
{'fields 0': self.docs[5]["field 0"]})
self.assertEqual(results, None)
def test_remove_many(self):
self.docs[6]["number"] = 1
self.docs[7]["number"] = 1
self.collection.insert_many(self.docs)
q = {"number": {"$ne": 1}}
count = self.collection.remove(q)
self.assertEqual(count, 8)
def test_remove_all(self):
self.collection.insert_many(self.docs)
self.collection.remove()
results = self.collection.count()
self.assertEqual(results, 0)
def test_eq(self):
self.collection.insert_many(self.docs)
q = {"field 1": {"$eq": self.docs[3]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_gt(self):
self.docs[6]["number"] = 11
self.collection.insert_many(self.docs)
q = {"number": {"$gt": 10}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_gte(self):
self.docs[6]["number"] = 11
self.collection.insert_many(self.docs)
q = {"number": {"$gte": 11}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_lt(self):
self.docs[6]["number"] = 9
self.collection.insert_many(self.docs)
q = {"number": {"$lt": 10}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_lte(self):
self.docs[6]["number"] = 9
self.collection.insert_many(self.docs)
q = {"number": {"$lte": 9}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_ne(self):
self.docs[6]["number"] = 9
self.collection.insert_many(self.docs)
q = {"number": {"$ne": 10}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_in(self):
self.docs[6]["number"] = 6
self.docs[7]["number"] = 7
self.collection.insert_many(self.docs)
q = {"number": {"$in": [6, 7]}}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_nin(self):
self.docs[6]["number"] = 6
self.docs[7]["number"] = 7
self.collection.insert_many(self.docs)
q = {"number": {"$nin": [6, 7]}}
response = self.collection.find(q)
self.assertEqual(len(response), len(self.docs) - 2)
def test_create_index(self):
self.collection.insert_many(self.docs)
self.collection.createIndex({"field 1": "hashed"})
q = {"field 1": {"$eq": self.docs[5]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_updating_index(self):
self.collection.insert_many(self.docs)
self.collection.createIndex({"field 1": "hashed"})
q = {"field 1": {"$eq": self.docs[5]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
self.collection.insert_one({"field 1": "myvalue"})
q = {"field 1": {"$eq": "myvalue"}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_remove_document_from_index(self):
self.collection.insert_many(self.docs)
self.collection.createIndex({"field 1": "hashed"})
q = {"field 1": {"$eq": self.docs[5]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
self.collection.remove({"field 1": self.docs[5]["field 1"]})
response = self.collection.find(q)
self.assertEqual(len(response), 0)
def test_drop_indexes(self):
self.collection.insert_many(self.docs)
self.collection.createIndex({"field 1": "hashed"})
q = {"field 1": {"$eq": self.docs[5]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
self.collection.dropIndexes()
q = {"field 1": {"$eq": self.docs[6]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_missing_fields(self):
self.docs[6]["new field"] = 1
self.docs[7]["new field"] = 1
self.collection.insert_many(self.docs)
q = {"new field": {"$eq": 1}}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
q = {"new field": 1}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_bad_id(self):
self.collection.insert_many(self.docs)
q = {"_id": "doesn't exist"}
response = self.collection.find(q)
self.assertEqual(len(response), 0)
def test_sync_load(self):
docs = self.docs
db = EssentialDB(filepath=SYNC_DB_FILE).get_collection()
db.insert_many(docs)
db.sync()
del (db)
db2 = EssentialDB(filepath=SYNC_DB_FILE).get_collection()
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
def test_auto_sync(self):
# test documents with ids already in place
with EssentialDB(filepath=SYNC_DB_FILE).get_collection() as db:
docs = self.docs
db.insert_many(docs)
with EssentialDB(filepath=SYNC_DB_FILE).get_collection() as db2:
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
@classmethod
def tearDownClass(cls):
import os
try:
os.remove(SYNC_DB_FILE)
except:
pass
def test_with(self):
with EssentialDB().adhoc as col:
self.assertIsInstance(col, Collection)
def test_adhoc_collection(self):
db = EssentialDB()
col = db.adhoc
self.assertIsInstance(col, Collection)
def do_use(self, args):
self.db = EssentialDB(filepath=args.strip())
print(self.db)
class TestEssentialDB(unittest.TestCase):
def setUp(self):
self.collection = EssentialDB()
self.docs = _gen_docs(10)
def test_set_get(self):
self.collection.set("my-test-number", 1)
self.collection.set("my-test-string", "Hello")
num = self.collection.get("my-test-number")
text = self.collection.get("my-test-string")
self.assertEqual(num, 1)
self.assertEqual(text, "Hello")
#now, try get before set
not_set = self.collection.get("doesn't exist")
self.assertIsNone(not_set)
def test_insert_and_find_one(self):
self.collection.insert_one(self.docs[0])
response = self.collection.find_one()
self.assertEqual(response["field 1"], self.docs[0]["field 1"])
self.assertEqual(self.collection.count(), 1)
def test_insert_many(self):
self.collection.insert_many(self.docs)
self.assertEqual(self.collection.count(), 10)
def test_find_one_of_many_by_id(self):
self.collection.insert_many(self.docs)
response = self.collection.find_one({"_id": self.docs[0]["_id"]})
self.assertEqual(response["_id"], self.docs[0]["_id"])
def test_find_one_complex_query(self):
self.collection.insert_many(self.docs)
response = self.collection.find_one({"field 0": self.docs[5]["field 0"], "field 1": self.docs[5]["field 1"]})
self.assertEqual(response["_id"], self.docs[5]["_id"])
def test_find_one_complex_dot_query(self):
self.collection.insert_many(self.docs)
response = self.collection.find_one({"nested.n0": self.docs[5]["nested"]["n0"], "nested.n1": self.docs[5]["nested"]["n1"]})
self.assertEqual(response["_id"], self.docs[5]["_id"])
def test_find_one_filter(self):
def filter(doc):
if doc["field 0"] == "Hello":
return True
return False
self.docs[5]["field 0"] = "Hello"
self.collection.insert_many(self.docs)
response = self.collection.find(filter=filter)
self.assertEqual(response[0]["_id"], self.docs[5]["_id"])
def test_find_or(self):
self.collection.insert_many(self.docs)
q = {"$or": [{"field 0": self.docs[5]["field 0"]}, {"field 1": self.docs[6]["field 1"]}]}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_find_nor(self):
self.collection.insert_many(self.docs)
q = {"$nor": [{"field 1": self.docs[0]["field 1"]}, {"field 1": self.docs[1]["field 1"]}]}
response = self.collection.find(q)
self.assertEqual(len(response), 8)
def test_find_and(self):
self.collection.insert_many(self.docs)
q = {"field 1": self.docs[0]["field 1"], "field 2": self.docs[0]["field 2"]}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_find_or_complex(self):
self.docs[0]["number"] = 0
self.docs[1]["number"] = 5
self.docs[2]["number"] = 20
self.docs[3]["number"] = 25
self.collection.insert_many(self.docs)
q = {"$or": [{"number": {"$eq": 5}}, {"number": {"$eq": 20}}]}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_find_nor_complex(self):
self.docs[0]["number"] = 0
self.docs[1]["number"] = 5
self.collection.insert_many(self.docs)
q = {"$nor": [{"number": {"$eq": 10}}, {"number": {"$eq": 0}}]}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_find_and_complex(self):
self.docs[0]["number"] = 0
self.docs[0]["field 0"] = "Hello"
self.docs[5]["number"] = 2
self.docs[5]["field 0"] = "Hello"
self.docs[7]["number"] = 4
self.docs[7]["field 0"] = "Bye"
self.collection.insert_many(self.docs)
q = {"$and": [{"number": {"$lt": 2}}, {"field 0": {"$eq": "Hello"}}]}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_find_not(self):
self.docs[0]["field 0"] = "Hello"
self.docs[0]["field 1"] = "Goodbye"
self.collection.insert_many(self.docs)
q = {"$not": [{'field 0': "Hello"}]}
response = self.collection.find(q)
self.assertEqual(len(response), 9)
q = {"$not": [{'field 0': "Hello", "field 1": "Goodbye"}]}
response = self.collection.find(q)
self.assertEqual(len(response), 9)
q = {"$not": [{'field 0': "Hello", "field 1": "Doesnt Exist"}]}
response = self.collection.find(q)
self.assertEqual(len(response), 10)
def test_update_one(self):
self.collection.insert_many(self.docs)
self.collection.update({"field 0": self.docs[5]["field 0"], "field 1": self.docs[5]["field 1"]}, {"field 2": "Hello"})
find = self.collection.find_one({"field 2": "Hello"})
self.assertEqual(find["_id"], self.docs[5]["_id"])
def test_remove_one(self):
self.collection.insert_many(self.docs)
self.collection.remove({'fields 0': self.docs[5]["field 0"]})
results = self.collection.find_one({'fields 0': self.docs[5]["field 0"]})
self.assertEqual(results, [])
def test_remove_many(self):
self.docs[6]["number"] = 1
self.docs[7]["number"] = 1
self.collection.insert_many(self.docs)
q = {"number": {"$ne": 1}}
count = self.collection.remove(q)
self.assertEqual(count, 8)
def test_remove_all(self):
self.collection.insert_many(self.docs)
self.collection.remove()
results = self.collection.count()
self.assertEqual(results, 0)
def test_eq(self):
self.collection.insert_many(self.docs)
q = {"field 1": {"$eq": self.docs[3]["field 1"]}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_gt(self):
self.docs[6]["number"] = 11
self.collection.insert_many(self.docs)
q = {"number": {"$gt": 10}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_gte(self):
self.docs[6]["number"] = 11
self.collection.insert_many(self.docs)
q = {"number": {"$gte": 11}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_lt(self):
self.docs[6]["number"] = 9
self.collection.insert_many(self.docs)
q = {"number": {"$lt": 10}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_lte(self):
self.docs[6]["number"] = 9
self.collection.insert_many(self.docs)
q = {"number": {"$lte": 9}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_ne(self):
self.docs[6]["number"] = 9
self.collection.insert_many(self.docs)
q = {"number": {"$ne": 10}}
response = self.collection.find(q)
self.assertEqual(len(response), 1)
def test_in(self):
self.docs[6]["number"] = 6
self.docs[7]["number"] = 7
self.collection.insert_many(self.docs)
q = {"number": {"$in": [6, 7]}}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_nin(self):
self.docs[6]["number"] = 6
self.docs[7]["number"] = 7
self.collection.insert_many(self.docs)
q = {"number": {"$nin": [6, 7]}}
response = self.collection.find(q)
self.assertEqual(len(response), len(self.docs)-2)
def test_nin(self):
self.docs[6]["number"] = 6
self.docs[7]["number"] = 7
self.collection.insert_many(self.docs)
q = {"number": {"$nin": [6, 7]}}
response = self.collection.find(q)
self.assertEqual(len(response), len(self.docs)-2)
def test_create_index(self):
self.collection.createIndex({"field 1": 1})
q = {"field 1": {"$eq": self.docs[5]["field 1"]}}
response = self.collection.find(q)
#self.assertEqual(len(response), 1)
def test_missing_fields(self):
self.docs[6]["new field"] = 1
self.docs[7]["new field"] = 1
self.collection.insert_many(self.docs)
q = {"new field": {"$eq": 1}}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
q = {"new field": 1}
response = self.collection.find(q)
self.assertEqual(len(response), 2)
def test_bad_id(self):
self.collection.insert_many(self.docs)
q = {"_id": "doesn't exist"}
response = self.collection.find(q)
self.assertEqual(len(response), 0)
def test_sync_load(self):
with EssentialDB( filepath=SYNC_DB_FILE) as db:
docs = _gen_docs(10)
db.insert_many(docs)
with EssentialDB( filepath=SYNC_DB_FILE) as db2:
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
def test_auto_sync(self):
# test documents with ids already in place
db = EssentialDB( filepath=SYNC_DB_FILE, autosync=True )
docs = _gen_docs(10)
db.insert_many(docs)
del(db)
db2 = EssentialDB( filepath=SYNC_DB_FILE)
find = db2.find_one({"_id": docs[5]["_id"]})
self.assertEqual(find["_id"], docs[5]["_id"])
@classmethod
def tearDownClass(cls):
import os
os.remove(SYNC_DB_FILE)
def setUp(self):
self.collection = EssentialDB()
self.docs = _gen_docs(10)