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()
