class TestSum(unittest.TestCase):
"""Automatic tests"""
def setUp(self):
self.table_mng = TableManager()
self.table_mng.table_df.from_dict(ROWS)
def test_write_json(self):
"""
Test if the file is created properly.When the path is
passed as parameter to pands.to_json, the return should be None
"""
result = self.table_mng.write_table_json()
self.assertIsNone(result, "Should be a None")
self.assertEqual(True,
os.path.exists(r"../app/integrations/data.json"))
def test_dataframe_columns(self):
"""Test integrity of columns"""
self.table_mng.create_table(["Oceania", "Americas"])
self.assertIn(
"country",
self.table_mng.table_df.columns,
"Country should be a column in df",
)
self.assertEqual(self.table_mng.table_df.shape[1], 4,
"Should be 4 columns")
self.assertEqual(
self.table_mng.table_df.columns.values.tolist(),
["region", "country", "language", "time"],
)
def main():
mongodb = Mongo()
table_mng = TableManager()
auth = Auth()
regions = get_regions(os.getenv("REGIONS_API"))
table_mng.create_table(regions)
auth.get_authorization()
if not auth.is_authenticated:
logging.info("User is not authenticated so database can't be shown")
return
mongodb.insert_documents(table_mng.table_df)
mongodb.list_documents()