def test_four(self): # Test for save data and load data with tempfile.TemporaryDirectory() as tempdir: db = fileStoreDB(tempdir) db.create("test_data", test_data) db.saveData("testfile.pk") self.assertTrue(os.path.isfile("testfile.pk")) db2 = fileStoreDB(tempdir) db2.loadData("testfile.pk") self.assertEqual(db2.read("test_data"), test_data)
def test_three(self): # Test for save data with tempfile.TemporaryDirectory() as tempdir: db = fileStoreDB(tempdir) db.create("test_data", test_data) db.saveData("testfile.pk") self.assertTrue(os.path.isfile("testfile.pk"))
def test_one(self): # Test for create, read with tempfile.TemporaryDirectory() as tempdir: db = fileStoreDB(tempdir) db.create("test_data", test_data) self.assertEqual(db.read("test_data"), test_data) with self.assertRaises(KeyError): db.read("test_data2")
def test_six(self): # Using a weather api to store data with tempfile.TemporaryDirectory() as tempdir: db = fileStoreDB(tempdir) r = requests.get('https://www.metaweather.com/api/location/44418/') if (r.ok): wdata = r.json() db.create("weather_data", wdata) self.assertEqual(db.read("weather_data"), wdata) db.create("weather_data2", wdata, 10) time.sleep(8) self.assertEqual(db.read("weather_data2"), wdata) time.sleep(2) with self.assertRaises(KeyError): db.read("weather_data2")
def test_five(self): # Test for time to live parameter with tempfile.TemporaryDirectory() as tempdir: db = fileStoreDB(tempdir) db.create("test_data", test_data, 10) db.create("test_data2", test_data, 20) self.assertEqual(db.read("test_data"), test_data) self.assertEqual(db.read("test_data2"), test_data) time.sleep(10) with self.assertRaises(KeyError): db.read("test_data") self.assertEqual(db.read("test_data2"), test_data) time.sleep(10) with self.assertRaises(KeyError): db.read("test_data2")