109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
"""
|
|
These unit tests will upload a test file,a test folder and a test contact,
|
|
Perform api operations on them,
|
|
And them remove them from your account.
|
|
"""
|
|
import unittest
|
|
import random
|
|
import os
|
|
|
|
from mega import Mega
|
|
|
|
email = os.environ['EMAIL']
|
|
password = os.environ['PASS']
|
|
|
|
mega = Mega()
|
|
# anonymous login
|
|
m = mega.login()
|
|
# normal login
|
|
# m = mega.login(email, password)
|
|
|
|
FIND_RESP = None
|
|
TEST_CONTACT = 'test@mega.co.nz'
|
|
TEST_PUBLIC_URL = (
|
|
'https://mega.nz/#!hYVmXKqL!r0d0-WRnFwulR_shhuEDwrY1Vo103-am1MyUy8oV6Ps'
|
|
)
|
|
TEST_FILE = os.path.basename(__file__)
|
|
TEST_FOLDER = 'mega.py_testfolder_{0}'.format(random.random())
|
|
|
|
|
|
class TestMega(unittest.TestCase):
|
|
|
|
def test_mega(self):
|
|
self.assertIsInstance(mega, Mega)
|
|
|
|
def test_login(self):
|
|
self.assertIsInstance(mega, Mega)
|
|
|
|
def test_get_user(self):
|
|
resp = m.get_user()
|
|
self.assertIsInstance(resp, dict)
|
|
|
|
def test_get_quota(self):
|
|
resp = m.get_quota()
|
|
self.assertIsInstance(int(resp), int)
|
|
|
|
def test_get_storage_space(self):
|
|
resp = m.get_storage_space(mega=True)
|
|
self.assertIsInstance(resp, dict)
|
|
|
|
def test_get_files(self):
|
|
files = m.get_files()
|
|
self.assertIsInstance(files, dict)
|
|
|
|
def test_get_link(self):
|
|
file = m.find(TEST_FILE)
|
|
if file:
|
|
link = m.get_link(file)
|
|
self.assertIsInstance(link, str)
|
|
|
|
def test_import_public_url(self):
|
|
resp = m.import_public_url(TEST_PUBLIC_URL)
|
|
file_handle = m.get_id_from_obj(resp)
|
|
resp = m.destroy(file_handle)
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_create_folder(self):
|
|
resp = m.create_folder(TEST_FOLDER)
|
|
self.assertIsInstance(resp, dict)
|
|
|
|
def test_rename(self):
|
|
file = m.find(TEST_FOLDER)
|
|
if file:
|
|
resp = m.rename(file, TEST_FOLDER)
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_delete_folder(self):
|
|
folder_node = m.find(TEST_FOLDER)[0]
|
|
resp = m.delete(folder_node)
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_delete(self):
|
|
file = m.find(TEST_FILE)
|
|
if file:
|
|
resp = m.delete(file[0])
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_destroy(self):
|
|
file = m.find(TEST_FILE)
|
|
if file:
|
|
resp = m.destroy(file[0])
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_empty_trash(self):
|
|
# resp None if already empty, else int
|
|
resp = m.empty_trash()
|
|
if resp is not None:
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_add_contact(self):
|
|
resp = m.add_contact(TEST_CONTACT)
|
|
self.assertIsInstance(resp, int)
|
|
|
|
def test_remove_contact(self):
|
|
resp = m.remove_contact(TEST_CONTACT)
|
|
self.assertIsInstance(resp, int)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|