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