def add_friends_list(token, profile_id): headers = {'token': token} data = {'profile_id': profile_id} r = requests.post(friends_url, headers=headers, data=data) assert json.loads(r.content)['status'] == 'success' def delete_friends_list(token, profile_id): headers = {'token': token} data = {'profile_id': profile_id} r = requests.delete(friends_url, headers=headers, data=data) assert json.loads(r.content)['status'] == 'success' if __name__ == '__main__': login = login() token = login['token'] result = search_profile(token) profile_id = result[0]['profile_id'] add_friends_list(token, profile_id) prev = get_friends_list(token) delete_friends_list(token, profile_id) curr = get_friends_list(token) assert len(prev) == len(curr) + 1
import requests import json from login_test import login profile_url = 'http://localhost:5000/profile' profile_data = { 'username': '******', 'school': 'UIUC', 'intro': 'test intro', 'profile_icon': None } def edit_profile(token): headers = {'token': token} r = requests.post(profile_url, headers=headers, data = profile_data) result = json.loads(r.content) assert result == profile_data def load_profile(token): headers = {'token': token} r = requests.get(profile_url, headers=headers) result = json.loads(r.content) assert result == profile_data if __name__ == '__main__': login = login() token = login['token'] edit_profile(token) load_profile(token)
import requests import json from login_test import login password_url = 'http://localhost:5000/change_password' new_password_data = {'old_password': '******', 'new_password': '******'} old_password_data = {'old_password': '******', 'new_password': '******'} def change_password(token, data): headers = {'token': token} r = requests.post(password_url, headers=headers, data=data) return json.loads(r.content) assert json.loads(r.content)['status'] == 'success' if __name__ == '__main__': login_result = login() token = login_result['token'] result = change_password(token, new_password_data) assert result['status'] == 'success' result = change_password(token, new_password_data) assert result['status'] == 'error' result = change_password(token, old_password_data) assert result['status'] == 'success'
import check_mkt_open import sell_open import login_test import sys if (check_mkt_open.check_today_open() == 0): sys.exit() else: app = login_test.login() sell_open.sell_available_amount() login_test.logout() sys.exit()