Exemplo n.º 1
0
    def test_remove_friend(self):
        uid_1 = "uid_user_1"
        uid_2 = "uid_user_2"

        usecret_1 = "usecret_1"
        usecret_2 = "usecret_2"

        resp, code = self.test_login(uid=uid_1, usecret=usecret_1, result=True)
        assert code == 200, resp

        resp_dict = json.loads(resp)

        headers = dict()
        headers['token'] = resp_dict['token']

        resp, code = api.on_add_friend(headers=headers, friend_uid=None)
        assert code == 400, str(code) + resp

        resp, code = api.on_add_friend(headers=headers, friend_uid="--")
        assert code == 400, str(code) + resp

        resp, code = self.test_login(uid=uid_2, usecret=usecret_2, result=True)
        assert code == 200, resp

        resp, code = api.on_remove_friend(headers=headers, friend_uid=uid_2)
        assert code == 200, str(code) + resp

        resp, code = api.on_remove_friend(headers=headers, friend_uid=uid_2)
        assert code == 400, str(code) + resp

        print "test[test_remove_friend] finished"
Exemplo n.º 2
0
def route_user_add_friend(friend_uid):
    if request.method == 'GET':
        # headers
        headers = request.headers

        return api.on_add_friend(headers=headers, friend_uid=friend_uid)
    elif request.method == 'DELETE':
        # headers
        headers = request.headers

        return api.on_remove_friend(headers=headers, friend_uid=friend_uid)