basic unit testing

Author: richard

tests.py renamed to examples.py

@@ -35,10 +35,10 @@ def test():
         print(files[file])
 
     # upload file
-    print(m.upload('tests.py'))
+    print(m.upload('examples.py'))
 
     # search for a file in account
-    file = m.find('tests.py')
+    file = m.find('examples.py')
 
     if file:
         # get public link

mega/mega.py

@@ -84,7 +84,7 @@ def api_request(self, data):
             data = [data]
 
         req = requests.post(
-            '{0}://g.api.{1}/cs'.format(self.schema, self.domain),
+            '{0}://eu.api.{1}/cs'.format(self.schema, self.domain),
             params=params,
             data=json.dumps(data),
             timeout=self.timeout)

setup.py

@@ -30,7 +30,7 @@ def get_package_data(package):
 
 setup(
     name='mega.py',
-    version='0.9.14',
+    version='0.9.15',
     packages=get_packages('mega'),
     package_data=get_package_data('mega'),
     description='Python lib for the Mega.co.nz API',

tests/unit-tests.py

@@ -0,0 +1,123 @@
+"""
+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 = '[email protected]'
+password = 'password'
+
+mega = Mega()
+m = mega.login(email, password)
+
+FIND_RESP = None
+TEST_CONTACT = '[email protected]'
+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():
+    m.upload(TEST_FILE)
+    # cached response to lower API requests for testing
+    FIND_RESP = m.find(TEST_FILE)
+    if FIND_RESP:
+        return True
+    else:
+        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_find(self):
+        file = FIND_RESP
+        if file:
+            self.assertTrue(isinstance(file, tuple), msg='this is {0} end'.format(file))
+
+    def test_get_link(self):
+        file = FIND_RESP
+        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()