def test_get_regions_route_ok(app, mocker, client):
expected_status = 200
expected_length = 8
expected_limit = 100
expected_page = 1
expected_total = 8
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
# mock app key authorization db query
query_mock.return_value \
.filter.return_value \
.one.return_value = AppKey()
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.count.return_value = expected_total
response = client.get("/regions/US?app_key=123")
assert response.status_code == expected_status
assert len(response.json['regions']) == expected_length
assert response.json['limit'] == expected_limit
assert response.json['page'] == expected_page
assert response.json['total'] == expected_total
def test_get_regions_limit_10_page_2_of_3(app, mocker):
expected_status = 200
expected_length = 3
expected_properties = ['code_2', 'id', 'name']
expected_limit = 10
expected_page = 2
expected_total = 25
expected_previous_uri = 'http://localhost/regions/US/1/10'
expected_next_uri = 'http://localhost/regions/US/3/10'
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.count.return_value = expected_total
result = get_regions('US', expected_page, expected_limit)
assert result[1] == expected_status
assert len(result[0].json['regions']) == expected_length
assert result[0].json['regions'][0] == {
x: None for x in expected_properties}
assert result[0].json['previous_uri'] == expected_previous_uri
assert result[0].json['next_uri'] == expected_next_uri
assert result[0].json['limit'] == expected_limit
assert result[0].json['page'] == expected_page
assert result[0].json['total'] == expected_total
def test_get_regions(app, mocker):
expected_status = 200
expected_length = 3
expected_properties = ['code_2', 'id', 'name']
expected_limit = 100
expected_page = 1
expected_total = 3
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.count.return_value = expected_total
result = get_regions('US')
assert result[1] == expected_status
assert len(result[0].json['regions']) == expected_length
assert result[0].json['regions'][0] == {
x: None for x in expected_properties}
assert result[0].json['limit'] == expected_limit
assert result[0].json['page'] == expected_page
assert result[0].json['total'] == expected_total
def test_get_regions_filter(app, mocker):
expected_status = 200
expected_length = 2
expected_json = {
'code_2': None,
'country': None,
'created_at': None,
'id': None,
'name': None,
'status': None,
'status_changed_at': None,
'updated_at': None,
}
expected_limit = 10
expected_page = 1
expected_total = 2
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.count.return_value = expected_total
request_mock = mocker.patch('modules.locations.routes_admin.request')
request_mock.args = {'country_id': 1}
result = get_regions()
assert result[1] == expected_status
assert len(result[0].json['regions']) == expected_length
assert result[0].json['regions'][0] == expected_json
assert result[0].json['limit'] == expected_limit
assert result[0].json['page'] == expected_page
assert result[0].json['total'] == expected_total
def test_get_regions_limit_10_page_2_of_3(app, mocker):
expected_status = 200
expected_length = 10
expected_json = {
'code_2': None,
'country': None,
'created_at': None,
'id': None,
'name': None,
'status': None,
'status_changed_at': None,
'updated_at': None,
}
expected_limit = 10
expected_page = 2
expected_total = 25
expected_previous_uri = 'http://localhost/regions/1/10'
expected_next_uri = 'http://localhost/regions/3/10'
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.count.return_value = expected_total
result = get_regions(expected_page, expected_limit)
assert result[1] == expected_status
assert len(result[0].json['regions']) == expected_length
assert result[0].json['regions'][0] == expected_json
assert result[0].json['previous_uri'] == expected_previous_uri
assert result[0].json['next_uri'] == expected_next_uri
assert result[0].json['limit'] == expected_limit
assert result[0].json['page'] == expected_page
assert result[0].json['total'] == expected_total
def test_get_regions_limit_5_page_2_of_3_route(app, mocker, client):
expected_status = 200
expected_length = 5
expected_limit = 5
expected_page = 2
expected_total = 12
expected_next_uri = 'http://localhost/regions/US/3/5'
expected_previous_uri = 'http://localhost/regions/US/1/5'
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
# mock app key authorization db query
query_mock.return_value \
.filter.return_value \
.one.return_value = AppKey()
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.count.return_value = expected_total
response = client.get(
"/regions/US/{}/{}?app_key=123".format(expected_page,
expected_limit))
assert response.status_code == expected_status
assert len(response.json['regions']) == expected_length
assert response.json['limit'] == expected_limit
assert response.json['page'] == expected_page
assert response.json['total'] == expected_total
assert response.json['next_uri'] == expected_next_uri
assert response.json['previous_uri'] == expected_previous_uri
def test_get_regions_filter_route(app, mocker, client):
expected_status = 200
expected_length = 10
expected_limit = 10
expected_page = 1
expected_total = 15
expected_next_uri = 'http://localhost/regions/2/10?country_id=1'
query_mock = mocker.patch('flask_sqlalchemy._QueryProperty.__get__')
# mock app key authorization db query
query_mock.return_value \
.filter.return_value \
.one.return_value = AppKey()
# mock user login db query
role2 = Role()
role2.id = 2
role2.name = 'SUPER_ADMIN'
role2.password_reset_days = 365
admin1 = Administrator()
admin1.id = 1
admin1.password = '******'
admin1.roles = [role2]
query_mock.return_value \
.filter.return_value \
.first.return_value = admin1
auth_db_mock = mocker.patch('modules.administrators.authentication.db')
auth_db_mock.add.return_value = None
auth_db_mock.commit.return_value = None
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.limit.return_value \
.offset.return_value \
.__iter__.return_value = [Region()] * expected_length
query_mock.return_value \
.filter.return_value \
.order_by.return_value \
.filter.return_value \
.count.return_value = expected_total
# mock user login
auth_mock = mocker.patch(
'modules.administrators.Authentication.is_account_locked')
auth_mock.return_value = False
credentials = base64.b64encode(
'admin1:admin1pass'.encode('ascii')).decode('utf-8')
response = client.get("/regions?country_id=1&app_key=123",
headers={"Authorization": f"Basic {credentials}"})
assert response.status_code == expected_status
assert len(response.json['regions']) == expected_length
assert response.json['limit'] == expected_limit
assert response.json['page'] == expected_page
assert response.json['total'] == expected_total
assert response.json['next_uri'] == expected_next_uri