import unittest
from pymongo import MongoClient


class MongoDbTest(unittest.TestCase):

    def setUp(self):
        self.mongo_client = MongoClient()
        self.db = self.mongo_client["common_code_test"]
        self.test_collection = self.db.test_data

    def tearDown(self):
        self.mongo_client.drop_database(self.db)
        self.mongo_client.close()

    def add_simple_test_data(self):
        for i in range(10):
            self.test_collection.insert_one({"test_int": i, "test_txt": "item {}".format(i)})
        self.test_collection.insert_one({"test_txt": "item 1"})

    def add_list_test_data(self):
        self.test_collection.insert_one({"test_list": ["a", "b", "c"]})
        self.test_collection.insert_one({"test_list": ["c", "d"]})