Пример #1
0
class URIMapper:
    def __init__(self):
        self.dict = Dictionary()
        self.authen = Authentication()
        self.course = Course()
        self.error = Error()
    
    def execute(self, environ):
        path = environ['PATH_INFO']
        
        #----- dictionary group -----
        if path.startswith('/lookup:'):
            return self.dict.lookup(environ)   
        
        #----- authentication group -----
        elif path.startswith('/register:'):
            return self.authen.register(environ)
        elif path.startswith('/login:'******'/logout:'):
            return self.authen.logout(environ)
        
        #----- learning group -----
        elif path.startswith('/levels:'):
            return self.course.levels(environ)
        elif path.startswith('/lessons:'):
            return self.course.lessons(environ)
        elif path.startswith('/listen:'):
            return self.course.listen(environ)    
        
        else:
            return self.error.bad_request()
Пример #2
0
class SampleUsers():
    def __init__(self):
        self.a = Authentication()
        
    def load_sample(self):
        self.a._register('tuan anh', 'pta', '841673450799')
        self.a._register('*****@*****.**', md5('3.').hexdigest(), '01673450799')
Пример #3
0
class CourseTest(unittest.TestCase):
    def setUp(self):
        self.c = Course()
        self.a = Authentication()
        self.root = 'http://localhost/'
        self.api_key = 'c74df88f0b1db2e4f48fdc4903851d41'
    
    def _levels(self, lang='en'):
        request = 'levels:lang="%s"&session_id="%s"&api_key="%s"' \
                % (quote(lang), self.a._login('Tuan Anh', 'pta'), self.api_key)
        uri = self.root + request
        output = urlopen(uri).read()
        return get_status_code(output)
    
    def _lessons(self, lang='en'):
        try:
            request = 'levels:lang="%s"&session_id="%s"&api_key="%s"' \
                    % (quote(lang), self.a._login('Tuan Anh', 'pta'), self.api_key)
            uri = self.root + request
            output = urlopen(uri).read()
            if get_status_code(output) is not None:
                first_level = get_property(output, 'name')
                
                request = 'lessons:lang="%s"&level="%s"&session_id="%s"&api_key="%s"' \
                        % (quote(lang), quote(first_level), self.a._login('Tuan Anh', 'pta'), self.api_key)
                uri = self.root + request
                output = urlopen(uri).read()
                return get_status_code(output)
            else:
                return None
        except:
            return -1
            
    
    def _listen(self, username, password):
        pass
    
    def test_levels(self):
        self.assertEqual(self._levels('en'), None)
        self.assertEqual(self._levels('vi'), None)
        self.assertEqual(self._levels('asdf'), 406)
        self.assertEqual(self._levels(''), 406)
        
    def test_lessons(self):
        self.assertEqual(self._lessons('en'), None)
        self.assertEqual(self._lessons('vi'), None)
        self.assertEqual(self._lessons('enz'), -1)
Пример #4
0
class DictionaryTest(unittest.TestCase):
    def setUp(self):
        self.d = Dictionary()
        self.a = Authentication()
        self.root = 'http://localhost/'
        self.api_key = 'c74df88f0b1db2e4f48fdc4903851d41'
    
    def _lookup(self, username, password, keyword):
        request = 'lookup:keyword="%s"&session_id="%s"&api_key="%s"' \
                % (quote(keyword), self.a._login(username, password), self.api_key)
        uri = self.root + request
        data = urlopen(uri).read()
        return get_status_code(data)
    
    def test_lookup(self):
        self.assertEqual(self._lookup("Tuan Anh", "pta", 'hello'), 200)
        self.assertEqual(self._lookup("Tuan Anh", "", 'hello'), 401)
        self.assertEqual(self._lookup("Tuan Anh", "pta", ''), 404)
        self.assertEqual(self._lookup("Tuan Anh", "pta", 'helloasdfz'), 404)
Пример #5
0
 def setUp(self):
     self.a = Authentication()
     self.root = 'http://localhost/'
     self.api_key = 'c74df88f0b1db2e4f48fdc4903851d41'
Пример #6
0
class AuthenTest(unittest.TestCase):
    def setUp(self):
        self.a = Authentication()
        self.root = 'http://localhost/'
        self.api_key = 'c74df88f0b1db2e4f48fdc4903851d41'
        
    def tearDown(self):
        self.a._remove('')
        self.a._remove('AloneRoad')
    
    def _register(self, username, password, phone_number):
        request = 'register:username="******"&password="******"&phone_number="%s"&api_key="%s"' \
                % (quote(username), quote(password), quote(phone_number), self.api_key)
        uri = self.root + request
        status_code = get_status_code(urlopen(uri).read())
        return status_code
    
    def _login(self, username, password):
        request = 'login:username="******"&password="******"&api_key="%s"' \
                % (quote(username), quote(password), self.api_key)
        uri = self.root + request
        status_code = get_status_code(urlopen(uri).read())
        return status_code
    
    def _get_session_id(self, username, password):
        request = 'login:username="******"&password="******"&api_key="%s"' \
                % (quote(username), quote(password), self.api_key)
        uri = self.root + request
        data = urlopen(uri).read()
        if get_status_code(data) == 202:
            return get_property(data, 'session_id')
        return None
    
    def _logout(self, session_id):
        request = 'logout:session_id="%s"&api_key="%s"' \
                % (session_id, self.api_key)
        uri = self.root + request
        status_code = get_status_code(urlopen(uri).read())
        return status_code
        
    def test_register(self):
        self.assertEqual(self._register("", "asd", "01673450799"), 406)
        self.assertEqual(self._register("Tuan Anh", "", "01673450799"), 406)
        self.assertEqual(self._register("Tuan Anh", "asd", "673450799"), 406)
        self.assertEqual(self._register("Tuan Anh", "asd", "a1673450799"), 406)
        self.assertEqual(self._register("AloneRoad", "asd", "01673450799"), 406)
        
        self.a._remove('AloneRoad')
        md5_password = md5("3.").hexdigest()
        self.assertEqual(self._register("AloneRoad", md5_password, "01673450799"), 201)

    def test_login(self):
        self.assertEqual(self._login('Tuan Anh', 'pta'), 202)
        self.assertEqual(self._login('tuan anh', 'pta'), 202)
        self.assertEqual(self._login('tuananh', 'pta'), 401)
        self.assertEqual(self._login('Tuan Anh', 'PTA'), 401)
        self.assertEqual(self._login('Tuan Anh', 'ta'), 401)
        self.assertEqual(self._login('Tuan Anh', ''), 401)
        self.assertEqual(self._login('', 'pta'), 401)
    
    def test_logout(self):
        session_id = self._get_session_id("*****@*****.**", md5('3.').hexdigest())
        self.assertEqual(self._logout(session_id), 200)
        self.assertEqual(self._logout('asdf'), 401)
Пример #7
0
 def __init__(self):
     self.dict = Dictionary()
     self.authen = Authentication()
     self.course = Course()
     self.error = Error()
Пример #8
0
 def new_func(*args, **kwargs):
     auth = Authentication(request)
     if not auth.authenticate():
         return jsonify(auth.error)
     return func(*args, **kwargs)
Пример #9
0
 def __init__(self):
     self.a = Authentication()