Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
M
MetObsAPI
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package registry
Container registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
MetObs
MetObsAPI
Commits
913e60d0
Verified
Commit
913e60d0
authored
2 years ago
by
David Hoese
Browse files
Options
Downloads
Patches
Plain Diff
Switch data api tests to pytest
parent
c5789b77
Branches
Branches containing commit
No related tags found
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
metobsapi/tests/test_data_api.py
+123
-118
123 additions, 118 deletions
metobsapi/tests/test_data_api.py
with
123 additions
and
118 deletions
metobsapi/tests/test_data_api.py
+
123
−
118
View file @
913e60d0
import
json
import
json
import
unittest
from
unittest
import
mock
from
unittest
import
mock
import
metobsapi
import
pytest
def
fake_data
(
interval
,
symbols
,
num_vals
,
single_result
=
False
):
def
fake_data
(
interval
,
symbols
,
num_vals
,
single_result
=
False
):
...
@@ -53,121 +52,129 @@ def fake_data(interval, symbols, num_vals, single_result=False):
...
@@ -53,121 +52,129 @@ def fake_data(interval, symbols, num_vals, single_result=False):
return
series
return
series
class
TestDataAPI
(
unittest
.
TestCase
):
@pytest.fixture
def
setUp
(
self
):
def
app
():
metobsapi
.
app
.
config
[
"
TESTING
"
]
=
True
from
metobsapi
import
app
as
real_app
metobsapi
.
app
.
config
[
"
DEBUG
"
]
=
True
self
.
app
=
metobsapi
.
app
.
test_client
()
def
test_doc
(
self
):
real_app
.
config
.
update
({
"
TESTING
"
:
True
,
"
DEBUG
"
:
True
})
res
=
self
.
app
.
get
(
"
/api/data
"
)
yield
real_app
@pytest.fixture
def
client
(
app
):
return
app
.
test_client
()
class
TestDataAPI
:
def
test_doc
(
self
,
client
):
res
=
client
.
get
(
"
/api/data
"
)
assert
b
"
Data Request Application
"
in
res
.
data
assert
b
"
Data Request Application
"
in
res
.
data
def
test_bad_format
(
self
):
def
test_bad_format
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.fake
"
)
res
=
client
.
get
(
"
/api/data.fake
"
)
self
.
assert
In
(
b
"
No data file format
"
,
res
.
data
)
assert
b
"
No data file format
"
in
res
.
data
def
test_bad_begin_json
(
self
):
def
test_bad_begin_json
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?symbols=air_temp&begin=blah
"
)
res
=
client
.
get
(
"
/api/data.json?symbols=air_temp&begin=blah
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
400
)
assert
res
[
"
code
"
]
==
400
self
.
assert
Equal
(
res
[
"
status
"
]
,
"
error
"
)
assert
res
[
"
status
"
]
==
"
error
"
self
.
assert
In
(
"
timestamp
"
,
res
[
"
message
"
]
)
assert
"
timestamp
"
in
res
[
"
message
"
]
def
test_bad_order
(
self
):
def
test_bad_order
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?order=blah&symbols=air_temp
"
)
res
=
client
.
get
(
"
/api/data.json?order=blah&symbols=air_temp
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
In
(
"
column
"
,
res
[
"
message
"
]
)
assert
"
column
"
in
res
[
"
message
"
]
self
.
assert
In
(
"
row
"
,
res
[
"
message
"
]
)
assert
"
row
"
in
res
[
"
message
"
]
def
test_bad_epoch
(
self
):
def
test_bad_epoch
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?epoch=blah&symbols=air_temp
"
)
res
=
client
.
get
(
"
/api/data.json?epoch=blah&symbols=air_temp
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
In
(
"'
h
'"
,
res
[
"
message
"
]
)
assert
"'
h
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
m
'"
,
res
[
"
message
"
]
)
assert
"'
m
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
s
'"
,
res
[
"
message
"
]
)
assert
"'
s
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
u
'"
,
res
[
"
message
"
]
)
assert
"'
u
'"
in
res
[
"
message
"
]
def
test_bad_interval
(
self
):
def
test_bad_interval
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?interval=blah&symbols=air_temp
"
)
res
=
client
.
get
(
"
/api/data.json?interval=blah&symbols=air_temp
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
In
(
"'
1m
'"
,
res
[
"
message
"
]
)
assert
"'
1m
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
5m
'"
,
res
[
"
message
"
]
)
assert
"'
5m
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
1h
'"
,
res
[
"
message
"
]
)
assert
"'
1h
'"
in
res
[
"
message
"
]
def
test_missing_inst
(
self
):
def
test_missing_inst
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?site=X&symbols=air_temp&begin=-05:00:00
"
)
res
=
client
.
get
(
"
/api/data.json?site=X&symbols=air_temp&begin=-05:00:00
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
400
)
assert
res
[
"
code
"
]
==
400
self
.
assert
Equal
(
res
[
"
status
"
]
,
"
error
"
)
assert
res
[
"
status
"
]
==
"
error
"
self
.
assert
In
(
"'
site
'"
,
res
[
"
message
"
]
)
assert
"'
site
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
inst
'"
,
res
[
"
message
"
]
)
assert
"'
inst
'"
in
res
[
"
message
"
]
def
test_missing_site
(
self
):
def
test_missing_site
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?inst=X&symbols=air_temp&begin=-05:00:00
"
)
res
=
client
.
get
(
"
/api/data.json?inst=X&symbols=air_temp&begin=-05:00:00
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
400
)
assert
res
[
"
code
"
]
==
400
self
.
assert
Equal
(
res
[
"
status
"
]
,
"
error
"
)
assert
res
[
"
status
"
]
==
"
error
"
self
.
assert
In
(
"'
site
'"
,
res
[
"
message
"
]
)
assert
"'
site
'"
in
res
[
"
message
"
]
self
.
assert
In
(
"'
inst
'"
,
res
[
"
message
"
]
)
assert
"'
inst
'"
in
res
[
"
message
"
]
def
test_missing_symbols
(
self
):
def
test_missing_symbols
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?begin=-05:00:00
"
)
res
=
client
.
get
(
"
/api/data.json?begin=-05:00:00
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
400
)
assert
res
[
"
code
"
]
==
400
self
.
assert
Equal
(
res
[
"
status
"
]
,
"
error
"
)
assert
res
[
"
status
"
]
==
"
error
"
self
.
assert
In
(
"'
symbols
'"
,
res
[
"
message
"
]
)
assert
"'
symbols
'"
in
res
[
"
message
"
]
def
test_too_many_points
(
self
):
def
test_too_many_points
(
self
,
client
):
res
=
self
.
app
.
get
(
"
/api/data.json?symbols=aoss.tower.air_temp&begin=1970-01-01T00:00:00
"
)
res
=
client
.
get
(
"
/api/data.json?symbols=aoss.tower.air_temp&begin=1970-01-01T00:00:00
"
)
self
.
assert
Equal
(
res
.
status_code
,
413
)
assert
res
.
status_code
==
413
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
In
(
"
too many values
"
,
res
[
"
message
"
]
)
assert
"
too many values
"
in
res
[
"
message
"
]
self
.
assert
Equal
(
res
[
"
code
"
]
,
413
)
assert
res
[
"
code
"
]
==
413
self
.
assert
Equal
(
res
[
"
status
"
]
,
"
fail
"
)
assert
res
[
"
status
"
]
==
"
fail
"
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_shorthand_one_symbol_json_row
(
self
,
query_func
):
def
test_shorthand_one_symbol_json_row
(
self
,
query_func
,
client
):
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
# row should be the default
# row should be the default
res
=
self
.
app
.
get
(
"
/api/data.json?site=aoss&inst=tower&symbols=air_temp&begin=-00:10:00
"
)
res
=
client
.
get
(
"
/api/data.json?site=aoss&inst=tower&symbols=air_temp&begin=-00:10:00
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
200
)
assert
res
[
"
code
"
]
==
200
self
.
assert
Equal
(
res
[
"
num_results
"
]
,
9
)
assert
res
[
"
num_results
"
]
==
9
self
.
assert
ListEqual
(
res
[
"
results
"
][
"
symbols
"
]
,
[
"
air_temp
"
]
)
assert
res
[
"
results
"
][
"
symbols
"
]
==
[
"
air_temp
"
]
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
timestamps
"
])
,
9
)
assert
len
(
res
[
"
results
"
][
"
timestamps
"
])
==
9
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
data
"
])
,
9
)
assert
len
(
res
[
"
results
"
][
"
data
"
])
==
9
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
data
"
][
0
])
,
1
)
assert
len
(
res
[
"
results
"
][
"
data
"
][
0
])
==
1
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_shorthand_one_symbol_json_column
(
self
,
query_func
):
def
test_shorthand_one_symbol_json_column
(
self
,
query_func
,
client
):
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
res
=
self
.
app
.
get
(
"
/api/data.json?site=aoss&inst=tower&symbols=air_temp&begin=-00:10:00&order=column
"
)
res
=
client
.
get
(
"
/api/data.json?site=aoss&inst=tower&symbols=air_temp&begin=-00:10:00&order=column
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
200
)
assert
res
[
"
code
"
]
==
200
self
.
assert
Equal
(
res
[
"
num_results
"
]
,
9
)
assert
res
[
"
num_results
"
]
==
9
self
.
assert
In
(
"
air_temp
"
,
res
[
"
results
"
][
"
data
"
]
)
assert
"
air_temp
"
in
res
[
"
results
"
][
"
data
"
]
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
data
"
][
"
air_temp
"
])
,
9
)
assert
len
(
res
[
"
results
"
][
"
data
"
][
"
air_temp
"
])
==
9
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
timestamps
"
])
,
9
)
assert
len
(
res
[
"
results
"
][
"
timestamps
"
])
==
9
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_wind_speed_direction_json
(
self
,
query_func
):
def
test_wind_speed_direction_json
(
self
,
query_func
,
client
):
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
wind_speed
"
,
"
wind_direction
"
,
"
wind_east
"
,
"
wind_north
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
wind_speed
"
,
"
wind_direction
"
,
"
wind_east
"
,
"
wind_north
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
res
=
self
.
app
.
get
(
res
=
client
.
get
(
"
/api/data.json?symbols=aoss.tower.wind_speed:aoss.tower.wind_direction&begin=-00:10:00&order=column
"
"
/api/data.json?symbols=aoss.tower.wind_speed:aoss.tower.wind_direction&begin=-00:10:00&order=column
"
)
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
200
)
assert
res
[
"
code
"
]
==
200
self
.
assert
Equal
(
res
[
"
num_results
"
]
,
9
)
assert
res
[
"
num_results
"
]
==
9
self
.
assert
In
(
"
aoss.tower.wind_direction
"
,
res
[
"
results
"
][
"
data
"
]
)
assert
"
aoss.tower.wind_direction
"
in
res
[
"
results
"
][
"
data
"
]
self
.
assert
In
(
"
aoss.tower.wind_speed
"
,
res
[
"
results
"
][
"
data
"
]
)
assert
"
aoss.tower.wind_speed
"
in
res
[
"
results
"
][
"
data
"
]
self
.
assert
Equal
(
len
(
list
(
res
[
"
results
"
][
"
data
"
].
keys
()))
,
2
)
assert
len
(
list
(
res
[
"
results
"
][
"
data
"
].
keys
()))
==
2
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_one_symbol_two_insts_json_row
(
self
,
query_func
):
def
test_one_symbol_two_insts_json_row
(
self
,
query_func
,
client
):
r
=
fake_data
(
r
=
fake_data
(
"
1m
"
,
"
1m
"
,
{
{
...
@@ -178,17 +185,17 @@ class TestDataAPI(unittest.TestCase):
...
@@ -178,17 +185,17 @@ class TestDataAPI(unittest.TestCase):
)
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
# row should be the default
# row should be the default
res
=
self
.
app
.
get
(
"
/api/data.json?symbols=aoss.tower.air_temp:mendota.buoy.air_temp&begin=-00:10:00
"
)
res
=
client
.
get
(
"
/api/data.json?symbols=aoss.tower.air_temp:mendota.buoy.air_temp&begin=-00:10:00
"
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assert
Equal
(
res
[
"
code
"
]
,
200
)
assert
res
[
"
code
"
]
==
200
self
.
assert
Equal
(
res
[
"
num_results
"
]
,
9
)
assert
res
[
"
num_results
"
]
==
9
self
.
assert
ListEqual
(
res
[
"
results
"
][
"
symbols
"
]
,
[
"
aoss.tower.air_temp
"
,
"
mendota.buoy.air_temp
"
]
)
assert
res
[
"
results
"
][
"
symbols
"
]
==
[
"
aoss.tower.air_temp
"
,
"
mendota.buoy.air_temp
"
]
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
timestamps
"
])
,
9
)
assert
len
(
res
[
"
results
"
][
"
timestamps
"
])
==
9
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
data
"
])
,
9
)
assert
len
(
res
[
"
results
"
][
"
data
"
])
==
9
self
.
assert
Equal
(
len
(
res
[
"
results
"
][
"
data
"
][
0
])
,
2
)
assert
len
(
res
[
"
results
"
][
"
data
"
][
0
])
==
2
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_one_symbol_three_insts_json_row
(
self
,
query_func
):
def
test_one_symbol_three_insts_json_row
(
self
,
query_func
,
client
):
r
=
fake_data
(
r
=
fake_data
(
"
1m
"
,
"
1m
"
,
{
{
...
@@ -207,89 +214,87 @@ class TestDataAPI(unittest.TestCase):
...
@@ -207,89 +214,87 @@ class TestDataAPI(unittest.TestCase):
st
[(
"
site2
"
,
"
inst2
"
)]
=
st
[(
"
aoss
"
,
"
tower
"
)]
st
[(
"
site2
"
,
"
inst2
"
)]
=
st
[(
"
aoss
"
,
"
tower
"
)]
st
[(
"
site3
"
,
"
inst3
"
)]
=
st
[(
"
aoss
"
,
"
tower
"
)]
st
[(
"
site3
"
,
"
inst3
"
)]
=
st
[(
"
aoss
"
,
"
tower
"
)]
with
mock
.
patch
(
"
metobsapi.util.data_responses.SYMBOL_TRANSLATIONS
"
,
st
):
with
mock
.
patch
(
"
metobsapi.util.data_responses.SYMBOL_TRANSLATIONS
"
,
st
):
res
=
self
.
app
.
get
(
res
=
client
.
get
(
"
/api/data.json?symbols=site1.inst1.air_temp:site2.inst2.air_temp:site3.inst3.air_temp&begin=-00:10:00
"
"
/api/data.json?symbols=site1.inst1.air_temp:site2.inst2.air_temp:site3.inst3.air_temp&begin=-00:10:00
"
)
)
res
=
json
.
loads
(
res
.
data
.
decode
())
res
=
json
.
loads
(
res
.
data
.
decode
())
self
.
assertEqual
(
res
[
"
code
"
],
200
)
assert
res
[
"
code
"
]
==
200
self
.
assertEqual
(
res
[
"
num_results
"
],
9
)
assert
res
[
"
num_results
"
]
==
9
self
.
assertListEqual
(
assert
res
[
"
results
"
][
"
symbols
"
]
==
[
"
site1.inst1.air_temp
"
,
"
site2.inst2.air_temp
"
,
"
site3.inst3.air_temp
"
]
res
[
"
results
"
][
"
symbols
"
],
[
"
site1.inst1.air_temp
"
,
"
site2.inst2.air_temp
"
,
"
site3.inst3.air_temp
"
]
assert
len
(
res
[
"
results
"
][
"
timestamps
"
])
==
9
)
assert
len
(
res
[
"
results
"
][
"
data
"
])
==
9
self
.
assertEqual
(
len
(
res
[
"
results
"
][
"
timestamps
"
]),
9
)
assert
len
(
res
[
"
results
"
][
"
data
"
][
0
])
==
3
self
.
assertEqual
(
len
(
res
[
"
results
"
][
"
data
"
]),
9
)
self
.
assertEqual
(
len
(
res
[
"
results
"
][
"
data
"
][
0
]),
3
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_one_symbol_csv
(
self
,
query_func
):
def
test_one_symbol_csv
(
self
,
query_func
,
client
):
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
# row should be the default
# row should be the default
res
=
self
.
app
.
get
(
"
/api/data.csv?symbols=aoss.tower.air_temp&begin=-00:10:00
"
)
res
=
client
.
get
(
"
/api/data.csv?symbols=aoss.tower.air_temp&begin=-00:10:00
"
)
res
=
res
.
data
.
decode
()
res
=
res
.
data
.
decode
()
# header, data, newline at end
# header, data, newline at end
lines
=
res
.
split
(
"
\n
"
)
lines
=
res
.
split
(
"
\n
"
)
self
.
assert
Equal
(
len
(
lines
)
,
5
+
9
+
1
)
assert
len
(
lines
)
==
5
+
9
+
1
# time + 1 channel
# time + 1 channel
self
.
assert
Equal
(
len
(
lines
[
5
].
split
(
"
,
"
))
,
2
)
assert
len
(
lines
[
5
].
split
(
"
,
"
))
==
2
self
.
assert
In
(
"
# code: 200
"
,
res
)
assert
"
# code: 200
"
in
res
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_one_symbol_xml
(
self
,
query_func
):
def
test_one_symbol_xml
(
self
,
query_func
,
client
):
from
xml.dom.minidom
import
parseString
from
xml.dom.minidom
import
parseString
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
# row should be the default
# row should be the default
res
=
self
.
app
.
get
(
"
/api/data.xml?symbols=aoss.tower.air_temp&begin=-00:10:00
"
)
res
=
client
.
get
(
"
/api/data.xml?symbols=aoss.tower.air_temp&begin=-00:10:00
"
)
res
=
parseString
(
res
.
data
.
decode
())
res
=
parseString
(
res
.
data
.
decode
())
# symbols: time and air_temp
# symbols: time and air_temp
self
.
assert
Equal
(
len
(
res
.
childNodes
[
0
].
childNodes
[
0
].
childNodes
)
,
2
)
assert
len
(
res
.
childNodes
[
0
].
childNodes
[
0
].
childNodes
)
==
2
# data rows
# data rows
self
.
assert
Equal
(
len
(
res
.
childNodes
[
0
].
childNodes
[
1
].
childNodes
)
,
9
)
assert
len
(
res
.
childNodes
[
0
].
childNodes
[
1
].
childNodes
)
==
9
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_three_symbol_csv
(
self
,
query_func
):
def
test_three_symbol_csv
(
self
,
query_func
,
client
):
"""
Test that multiple channels in a CSV file are structured properly.
"""
"""
Test that multiple channels in a CSV file are structured properly.
"""
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
,
"
rel_hum
"
,
"
wind_speed
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
,
"
rel_hum
"
,
"
wind_speed
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
# row should be the default
# row should be the default
res
=
self
.
app
.
get
(
res
=
client
.
get
(
"
/api/data.csv?symbols=aoss.tower.air_temp:
"
"
aoss.tower.rel_hum:aoss.tower.wind_speed&begin=-00:10:00
"
"
/api/data.csv?symbols=aoss.tower.air_temp:
"
"
aoss.tower.rel_hum:aoss.tower.wind_speed&begin=-00:10:00
"
)
)
res
=
res
.
data
.
decode
()
res
=
res
.
data
.
decode
()
# header, data, newline at end
# header, data, newline at end
lines
=
res
.
split
(
"
\n
"
)
lines
=
res
.
split
(
"
\n
"
)
self
.
assert
Equal
(
len
(
lines
)
,
5
+
9
+
1
)
assert
len
(
lines
)
==
5
+
9
+
1
# time + 3 channels
# time + 3 channels
self
.
assert
Equal
(
len
(
lines
[
5
].
split
(
"
,
"
))
,
4
)
assert
len
(
lines
[
5
].
split
(
"
,
"
))
==
4
self
.
assert
In
(
"
# code: 200
"
,
res
)
assert
"
# code: 200
"
in
res
@mock.patch
(
"
metobsapi.data_api.query
"
)
@mock.patch
(
"
metobsapi.data_api.query
"
)
def
test_three_symbol_csv_repeat
(
self
,
query_func
):
def
test_three_symbol_csv_repeat
(
self
,
query_func
,
client
):
"""
Test that multiple channels in a CSV file are structured properly.
"""
"""
Test that multiple channels in a CSV file are structured properly.
"""
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
,
"
rel_hum
"
,
"
wind_speed
"
]},
9
)
r
=
fake_data
(
"
1m
"
,
{(
"
aoss
"
,
"
tower
"
):
[
"
time
"
,
"
air_temp
"
,
"
rel_hum
"
,
"
wind_speed
"
]},
9
)
query_func
.
return_value
=
r
query_func
.
return_value
=
r
# row should be the default
# row should be the default
res
=
self
.
app
.
get
(
res
=
client
.
get
(
"
/api/data.csv?symbols=aoss.tower.air_temp:
"
"
aoss.tower.air_temp:aoss.tower.air_temp&begin=-00:10:00
"
"
/api/data.csv?symbols=aoss.tower.air_temp:
"
"
aoss.tower.air_temp:aoss.tower.air_temp&begin=-00:10:00
"
)
)
res
=
res
.
data
.
decode
()
res
=
res
.
data
.
decode
()
# header, data, newline at end
# header, data, newline at end
lines
=
res
.
split
(
"
\n
"
)
lines
=
res
.
split
(
"
\n
"
)
# header, data (one empty line), newline at end
# header, data (one empty line), newline at end
self
.
assert
Equal
(
len
(
lines
)
,
5
+
1
+
1
)
assert
len
(
lines
)
==
5
+
1
+
1
# time + 1 channel
# time + 1 channel
self
.
assert
Equal
(
len
(
lines
[
5
].
split
(
"
,
"
))
,
1
)
assert
len
(
lines
[
5
].
split
(
"
,
"
))
==
1
self
.
assert
In
(
"
# code: 400
"
,
res
)
assert
"
# code: 400
"
in
res
# @mock.patch('metobsapi.data_api.query')
# @mock.patch('metobsapi.data_api.query')
# def test_jsonp_bad_symbol_400(self, query_func):
# def test_jsonp_bad_symbol_400(self, query_func
, client
):
# XXX: Not currently possible with flask-json
# XXX: Not currently possible with flask-json
# r = fake_data('1m', {('aoss', 'tower'): ['time', 'air_temp']}, 9)
# r = fake_data('1m', {('aoss', 'tower'): ['time', 'air_temp']}, 9)
# query_func.return_value = r
# query_func.return_value = r
# # row should be the default
# # row should be the default
# res =
self.app
.get('/api/data.json?site=aoss&inst=tower&symbols=bad&begin=-00:10:00&callback=test')
# res =
client
.get('/api/data.json?site=aoss&inst=tower&symbols=bad&begin=-00:10:00&callback=test')
#
self.
assert
Equal(
res.status_code
,
400
)
# assert
res.status_code
==
400
# res = res.data.decode()
# res = res.data.decode()
#
self.
assert
Equal(
res['code']
,
400
)
# assert
res['code']
==
400
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment