megapy/tests/unit-tests.py

"""
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.
"""
from mega import Mega
import unittest
import random
import os

email = 'your@email.com'
password = 'password'

mega = Mega()
m = mega.login(email, password)

FIND_RESP = None
TEST_CONTACT = 'test@mega.co.nz'
TEST_PUBLIC_URL = 'https://mega.co.nz/#!EYI2VagT!Ic1yblki8oM4v6XHquCe4gu84kxc4glFchj8OvcT5lw'
TEST_FILE = os.path.basename(__file__)
TEST_FOLDER = 'mega.py_testfolder_{0}'.format(random.random())


def pre_test():
    global FIND_RESP
    try:
        m.upload(TEST_FILE)
        return True
    except:
        raise ValueError('Pre-test functions failed!')


class TestMega(unittest.TestCase):

    def test_mega(self):
        self.assertTrue(isinstance(mega, Mega))

    def test_login(self):
        self.assertTrue(isinstance(mega, Mega))

    def test_get_user(self):
        resp = m.get_user()
        self.assertTrue(isinstance(resp, dict))

    def test_get_quota(self):
        resp = m.get_quota()
        self.assertTrue(isinstance(resp, long))

    def test_get_storage_space(self):
        resp = m.get_storage_space(mega=True)
        self.assertTrue(isinstance(resp, dict))

    def test_get_files(self):
        files = m.get_files()
        self.assertTrue(isinstance(files, dict))

    def test_get_link(self):
        file = m.find(TEST_FILE)
        if file:
            link = m.get_link(file)
            self.assertTrue(isinstance(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.assertTrue(isinstance(resp, int))

    def test_create_folder(self):
        resp = m.create_folder(TEST_FOLDER)
        self.assertTrue(isinstance(resp, dict))

    def test_rename(self):
        file = m.find(TEST_FOLDER)
        if file:
            resp = m.rename(file, TEST_FOLDER)
            self.assertTrue(isinstance(resp, int))

    def test_delete_folder(self):
        folder_node = m.find(TEST_FOLDER)[0]
        resp = m.delete(folder_node)
        self.assertTrue(isinstance(resp, int))

    def test_delete(self):
        file = m.find(TEST_FILE)
        if file:
            resp = m.delete(file[0])
            self.assertTrue(isinstance(resp, int))
        else:
            raise ValueError('file not found')

    def test_destroy(self):
        file = m.find(TEST_FILE)
        if file:
            resp = m.destroy(file[0])
            self.assertTrue(isinstance(resp, int))
        else:
            raise ValueError('file not found')

    def test_empty_trash(self):
        #resp None if already empty, else int
        resp = m.empty_trash()
        if resp is not None:
            self.assertTrue(isinstance(resp, int))

    def test_add_contact(self):
        resp = m.add_contact(TEST_CONTACT)
        self.assertTrue(isinstance(resp, int))

    def test_remove_contact(self):
        resp = m.remove_contact(TEST_CONTACT)
        self.assertTrue(isinstance(resp, int))


if __name__ == '__main__':
    if pre_test():
        unittest.main()