class FileStorageTest(unittest.TestCase):
def setUp(self):
self.subject = FileStorage('test_storage.json')
def tearDown(self):
if os.path.exists(self.subject.file_name):
os.remove(self.subject.file_name)
def test_exist_if_not_exist(self):
self.subject.exist()
self.assertEqual(self.subject.exist(), False)
def test_exist_if_exist(self):
open(self.subject.file_name, "w").write("")
self.subject.exist()
self.assertEqual(self.subject.exist(), True)
os.remove(self.subject.file_name)
def test_load(self):
data = [MediaInfo(), MediaInfo()]
open(self.subject.file_name, "w").write(json.dumps(data))
self.subject.load()
self.assertEqual(len(self.subject.items()), 2)
os.remove(self.subject.file_name)
def test_save(self):
self.subject.add(MediaInfo())
self.subject.add(MediaInfo())
self.subject.save()
self.assertEqual(len(self.subject.items()), 2)
