1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
import unittest
import os
from tusclient.storage import filestorage
class FileStorageTest(unittest.TestCase):
def setUp(self):
self.storage_path = 'storage.json'
self.storage = filestorage.FileStorage(self.storage_path)
def tearDown(self):
self.storage.close()
os.remove(self.storage_path)
def test_set_get_remove_item(self):
url = 'http://tusd.tusdemo.net/files/unique_file_id'
key = 'unique_key'
url_2 = 'http://tusd.tusdemo.net/files/unique_file_id_2'
key_2 = 'unique_key_2'
self.storage.set_item(key, url)
self.storage.set_item(key_2, url_2)
self.assertEqual(self.storage.get_item(key), url)
self.assertEqual(self.storage.get_item(key_2), url_2)
self.storage.remove_item(key)
self.assertIsNone(self.storage.get_item(key))
|
