tests.py

# -*- coding: utf-8 -*-
import os
import sys
import unittest
import tempfile

from mega import Mega
from mega.exceptions import MegaIncorrectPasswordExcetion

class TestMega(unittest.TestCase): 
 def setUp(self):
 unittest.TestCase.setUp(self)
 self._email = os.environ.get('MEGAEMAIL') or sys.argv[0]
 self._password = os.environ.get('MEGAPASSWORD') or sys.argv[1]

 def _check_file_exists(self, file_name, files):
 uploaded = False
 for f in files['f']:
 if isinstance(f['a'], dict):
 if f['a'].get('n') == file_name:
 uploaded = True
 return uploaded

 def _test_upload_file(self, api):
 api.get_files()

 # Create temp file
 uFile, uFilePath = tempfile.mkstemp()
 os.write(uFile, "Does it work?")
 os.close(uFile)
 api.uploadfile(uFilePath) # inception
 files = api.get_files()
 
 uFileName = os.path.basename(uFilePath)
 uploaded = self._check_file_exists(uFileName, files)
 self.assertEqual(uploaded, True)
 
 def test_login_fail(self):
 with self.assertRaises(MegaIncorrectPasswordExcetion):
 Mega.from_credentials("valid@email.com", "test");
 
 def test_login_valid(self):
 Mega.from_credentials(self._email, self._password)
 
 def test_upload_file_logged(self):
 self._test_upload_file(Mega.from_credentials(self._email, self._password))
 
 def test_upload_file_ephemeral(self):
 self._test_upload_file(Mega.from_ephemeral())

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