def setUpClass(cls): cls.app = app cls.client = app.test_client() cls.db = db.get_db(alias=DB_ALIAS) if cls.db.list_collection_names(): logger.error( f"Database has data: {cls.db.list_collection_names()}") logger.error("Please drop data from it before applying tests") # try to upload data into database for fixture in cls.fixtures: logger.debug(f"Search for {fixture}") data_file = next(pathlib.Path('tests').glob(f"**/{fixture}.json")) logger.debug(f"Found {data_file}") with open(data_file) as handle: data = json.load(handle) data = sanitize_data(data) collection = cls.db[fixture] try: collection.insert_many(data) except BulkWriteError as e: logger.error(f"Cannot insert data: {e}")
def setUp(self) -> None: app = create_app(MONGODB_SETTINGS={"DB": "kanban-testing"}, CELERY_ALWAYS_EAGER=True) app.testing = True self.app = app.test_client() initialize_apps(app) self.db = db.get_db()
def import_authors(csv_file): ''' Delete db collection named 'authors' and create a new with the data of `csv_file`''' database = db.get_db() collection_name = 'author' db_authors = database[collection_name] # Delete `db` collection with the `collection_name`''' database.drop_collection(collection_name) data = read_data(csv_file) data_json = json.loads(data.to_json(orient='records')) db_authors.insert_many(data_json)
class Test_Fetch_Friends(unittest.TestCase): test_client = app.test_client() get_db = db.get_db() def test_fetch_friends(self): response = self.test_client.get('fetch_friends?person_1=1&person_2=5') self.assertEqual(200, response.status_code) def test_fetch_friends_400_URL(self): response = self.test_client.get('fetch_friends') self.assertEqual(400, response.status_code) if __name__ == "__main__": unittest.main()
class Test_Fetch_Employees(unittest.TestCase): test_client = app.test_client() get_db = db.get_db() def test_fetch_employees(self): response = self.test_client.get('fetch_employees?company_id=1') self.assertEqual(200, response.status_code) def test_fetch_employees_400_URL(self): response = self.test_client.get('fetch_employees') self.assertEqual(400, response.status_code) def test_fetch_employees_400_TYPE(self): response = self.test_client.get('fetch_employees?company_id=type') print(response) self.assertEqual(400, response.status_code) if __name__ == "__main__": unittest.main()
class Test_Fetch_Fruits(unittest.TestCase): test_client = app.test_client() get_db = db.get_db() def test_fetch_people(self): response = self.test_client.get('fetch_person?person_id=6') self.assertEqual(200, response.status_code) def test_fetch_people_400_URL(self): response = self.test_client.get('fetch_person') self.assertEqual(200, response.status_code) def test_fetch_people_400_TYPE(self): response = self.test_client.get('fetch_person?person_id=type') print(response) self.assertEqual(400, response.status_code) if __name__ == "__main__": unittest.main()
def setUp(self): self.app = app.test_client() self.db = db.get_db()
def setUp(self): self.app = app.test_client() if 'mongoengine' not in app.extensions: initialize_db(app) initialize_routes(api) self.db = db.get_db()
def setUp(self): self.file = 'authors.csv' self.data = read_data(self.file) self.app = app.test_client() self.db = db.get_db()