Notes-taking-application-API-using-flask-and-sqlalchemy/test_api_endpoint.py at main · rsiva0400/Notes-taking-application-API-using-flask-and-sqlalchemy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import pytest
import requests

@pytest.fixture
def api_url():
    return "http://127.0.0.1:5000"

@pytest.mark.parametrize("username, password, expected_status_code", [
    ("valid_username", "valid_password", 200),
    ("valid_username", "invalid_password", 400),
    ("", "", 400),
])
def test_signup(api_url, username, password, expected_status_code):
    payload = {"username": username, "password": password}
    response = requests.post(api_url + "/signup", json=payload)

    assert response.status_code == expected_status_code

@pytest.mark.parametrize("username, password, expected_status_code", [
    ("valid_username", "valid_password", 200),
    ("valid_username", "invalid_password", 400),
    ("", "", 400),
])
def test_login(api_url, username, password, expected_status_code):
    payload = {"username": username, "password": password}
    response = requests.post(api_url + "/login", json=payload)

    assert response.status_code == expected_status_code

def test_create_note(api_url):
    login_payload = {"username": 'valid_username', "password": 'valid_password'}
    login_response = requests.post(api_url + "/login", json=login_payload).json()

    access_token = login_response['access_token']
    create_note_payload = {"access_token": access_token, "message": 'Hi, How are you?', "access_list" : "newUser1,newUser2"}
    create_note_response = requests.post(api_url + "/notes/create", json=create_note_payload)

    assert create_note_response.status_code == 200

@pytest.mark.parametrize("note_id, expected_status_code", [
    (1, 200),
    (60, 400),
])

def test_fetch_note(api_url, note_id, expected_status_code):
    login_payload = {"username": 'valid_username', "password": 'valid_password'}
    login_response = requests.post(api_url + "/login", json=login_payload).json()

    access_token = login_response['access_token']
    fetch_note_payload = {"access_token": access_token}

    fetch_note_response = requests.get(api_url + f'/notes/{note_id}', json=fetch_note_payload)

    assert fetch_note_response.status_code == expected_status_code
    print(fetch_note_response.json())

@pytest.mark.parametrize("note_id, updated_text,expected_status_code", [
    (2, 'i am fine',400),
    (1, 'a'*10, 200),
    (6, 'abcd',400),
])

def test_update_note(api_url, note_id, updated_text, expected_status_code):
    login_payload = {"username": 'valid_username', "password": 'valid_password'}
    login_response = requests.post(api_url + "/login", json=login_payload).json()

    access_token = login_response['access_token']
    update_note_payload = {"access_token": access_token, 'updated_text': updated_text}

    update_note_response = requests.put(api_url + f'/notes/{note_id}', json=update_note_payload)
    print(update_note_response.json())
    assert update_note_response.status_code == expected_status_code


if __name__ == "__main__":
    pytest.main()