def test_twitter_profile(baseurl, apikey, apisecret, twitter_handle): headers = conftest.auth_headers(apikey, apisecret) response = import_twitter_user( conftest.twitter_import_user_api_url(baseurl), headers, twitter_handle) expect(response.status_code).to(equal(200)) status_check_url = "{}{}".format(baseurl, response.json()["_links"]["self"]["href"]) latest_response = response.json() idx = 1 while latest_response["status"] not in ["Finished", "Failed", "Error" ] and idx < 50: time.sleep(5) response = get(status_check_url, headers=headers) expect(response.status_code).to(equal(200)) latest_response = response.json() print("Retry {}: Status - {}. Last updated = {}".format( idx, latest_response["status"], latest_response["updated"])) idx += 1 expect(latest_response["search_key"]).to(equal(twitter_handle)) expect(latest_response["status"]).to(equal('Finished')) people_url = "{}{}".format(baseurl, response.json()["_links"]["people"]["href"]) response = get(people_url, headers=headers) expect(response.status_code).to(equal(200))
def test_create_person_only(baseurl, apikey, apisecret): person_data = factories.get_person_data() person_api_url = conftest.person_api_url(baseurl) auth_headers = conftest.auth_headers(apikey, apisecret) response = post(person_api_url, json=person_data, headers=auth_headers) response_json = json.loads(response.content) expect(response.status_code).to(equal(200)) expect(response_json["name"]).to(equal(person_data["name"]))
def test_create_person_with_content(baseurl, apikey, apisecret): content_data = factories.get_content_data() person_data = factories.get_person_data(content_data) person_api_url = conftest.person_api_url(baseurl) auth_headers = conftest.auth_headers(apikey, apisecret) response = post(person_api_url, json=person_data, headers=auth_headers) response_json = json.loads(response.content) expect(response.status_code).to(equal(200)) expect(response_json["name"]).to(equal(person_data["name"])) expect(response_json["contents"][0]).to(have_key("receptiviti_scores")) expect(response_json["contents"][0]).to(have_key("liwc_scores"))
def test_merge_personality(baseurl, apikey, apisecret): auth_headers = conftest.auth_headers(apikey, apisecret) person1_id = create_one_person(auth_headers, baseurl) person2_id = create_one_person(auth_headers, baseurl) person3_id = create_one_person(auth_headers, baseurl) query_params = "&".join(["person_ids={}".format(person1_id) for person_id in [person1_id, person2_id, person3_id]]) merge_personality_api_url = "{}?{}".format(conftest.merge_personality_api_url(baseurl), query_params) response = get(merge_personality_api_url, headers=auth_headers) expect(response.status_code).to(equal(200)) response_json = json.loads(response.content) expect(response_json).to(have_key("receptiviti_scores")) expect(response_json).to(have_key("liwc_scores"))
def test_upload_csv_with_email_notification_enabled(baseurl, apikey, apisecret): csv_file_to_upload = get_sample_csv_file("CSV_Upload_samples.csv") headers = conftest.auth_headers(apikey, apisecret) send_email = False response = upload_csv_data(conftest.upload_api_url(baseurl), headers, csv_file_to_upload, send_email) expect(response.status_code).to(equal(200)) status_check_url = "{}{}".format(baseurl, response.json()["_links"]["self"]["href"]) latest_response = response.json() idx = 1 while latest_response["status"] not in ["completed", "failed", "errored"] and idx < 50: time.sleep(5) response = get(status_check_url, headers=headers) expect(response.status_code).to(equal(200)) latest_response = response.json() print("Retry {}: Status - {}. Last updated = {}".format(idx, latest_response["status"], latest_response["updated"])) idx += 1 expect(latest_response["result"]["success"]).to(equal(5))
def test_add_content_for_an_existing_person(baseurl, apikey, apisecret): auth_headers = conftest.auth_headers(apikey, apisecret) person = get_one_person(auth_headers, baseurl) sample_data = factories.get_content_data() create_sample_url = conftest.person_content_api_url(baseurl, person["_id"]) print("*" * 20) print("create_sample_url={}".format(create_sample_url)) print("*" * 20) response = post(create_sample_url, json=sample_data, headers=auth_headers) expect(response.status_code).to(equal(200)) response_json = json.loads(response.content) expect(response_json["content_handle"]).to(equal(sample_data["content_handle"])) expect(response_json).to(have_key("receptiviti_scores")) expect(response_json).to(have_key("liwc_scores")) _print_interesting_content_information(response_json)
def test_get_people_in_the_system(baseurl, apikey, apisecret): auth_headers = conftest.auth_headers(apikey, apisecret) people = get_all_people_in_the_system(auth_headers, baseurl) expect(len(people)).to(be_above(0)) expect(people[0]).to(have_key("_id"))