| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| import os |
| from json import loads |
| import unittest |
| from webtest import TestApp |
| |
| from ..run_webservices import app |
| |
| test_app = TestApp(app) |
| |
| class TestLatLonExtraction(unittest.TestCase): |
| def test_successful_latlon_extract(self): |
| expected_return = { |
| "success": True, |
| "lat_name": "lat", |
| "lon_name": "lon", |
| "lat_min": -45.759998321533203, |
| "lat_max": -45.759998321533203, |
| "lon_min": -24.639999389648438, |
| "lon_max": 60.279998779296875 |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time.nc') |
| |
| response = test_app.get('/lfme/list_latlon/' + file_location) |
| |
| self.assertDictEqual(expected_return, response.json) |
| |
| def test_successful_latlon_extract_jsonp(self): |
| expected_return = { |
| "success": True, |
| "lat_name": "lat", |
| "lon_name": "lon", |
| "lat_min": -45.759998321533203, |
| "lat_max": -45.759998321533203, |
| "lon_min": -24.639999389648438, |
| "lon_max": 60.279998779296875 |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time.nc') |
| |
| response = test_app.get('/lfme/list_latlon/' + file_location + '?callback=test_callback') |
| |
| json = response.text |
| |
| # Strip out the callback functino and the json string from the response |
| # and check for proper content. |
| callback = json[:json.index('(')] |
| json = json[json.index('(') + 1 : json.rindex(')')] |
| json = loads(json) |
| |
| self.assertDictEqual(expected_return, json) |
| self.assertEqual(callback, "test_callback") |
| |
| def test_failure_latlon_extract(self): |
| expected_return = { |
| "success": False, |
| "variables": ["invalid_lon", "invalid_time", "invalid_lat"] |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time_invalid.nc') |
| |
| response = test_app.get('/lfme/list_latlon/' + file_location) |
| |
| self.assertDictEqual(expected_return, response.json) |
| |
| def test_failure_latlon_extract_jsonp(self): |
| expected_return = { |
| "success": False, |
| "variables": ["invalid_lon", "invalid_time", "invalid_lat"] |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time_invalid.nc') |
| |
| response = test_app.get('/lfme/list_latlon/' + file_location + '?callback=test_callback') |
| json = response.text |
| |
| # Strip out the callback functino and the json string from the response |
| # and check for proper content. |
| callback = json[:json.index('(')] |
| json = json[json.index('(') + 1 : json.rindex(')')] |
| json = loads(json) |
| |
| self.assertDictEqual(expected_return, json) |
| self.assertEqual(callback, "test_callback") |
| |
| class TestTimeExtraction(unittest.TestCase): |
| def test_successful_time_extract(self): |
| expected_return = { |
| "success": True, |
| "time_name": "time", |
| "start_time": "1989-01-01 00:00:00", |
| "end_time": "2008-12-01 00:00:00" |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time.nc') |
| |
| response = test_app.get('/lfme/list_time/' + file_location) |
| |
| self.assertDictEqual(expected_return, response.json) |
| |
| def test_successful_time_extract_jsonp(self): |
| expected_return = { |
| "success": True, |
| "time_name": "time", |
| "start_time": "1989-01-01 00:00:00", |
| "end_time": "2008-12-01 00:00:00" |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time.nc') |
| |
| response = test_app.get('/lfme/list_time/' + file_location + '?callback=test_callback') |
| json = response.text |
| |
| # Strip out the callback functino and the json string from the response |
| # and check for proper content. |
| callback = json[:json.index('(')] |
| json = json[json.index('(') + 1 : json.rindex(')')] |
| json = loads(json) |
| |
| self.assertDictEqual(expected_return, json) |
| self.assertEqual(callback, "test_callback") |
| |
| def test_failure_time_extract(self): |
| expected_return = { |
| "success": False, |
| "variables": ["invalid_lon", "invalid_time", "invalid_lat"] |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time_invalid.nc') |
| |
| response = test_app.get('/lfme/list_time/' + file_location) |
| |
| self.assertDictEqual(expected_return, response.json) |
| |
| def test_failure_time_extract_jsonp(self): |
| expected_return = { |
| "success": False, |
| "variables": ["invalid_lon", "invalid_time", "invalid_lat"] |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time_invalid.nc') |
| |
| response = test_app.get('/lfme/list_time/' + file_location + '?callback=test_callback') |
| json = response.text |
| |
| # Strip out the callback functino and the json string from the response |
| # and check for proper content. |
| callback = json[:json.index('(')] |
| json = json[json.index('(') + 1 : json.rindex(')')] |
| json = loads(json) |
| |
| self.assertDictEqual(expected_return, json) |
| self.assertEqual(callback, "test_callback") |
| |
| class TestVariableExtraction(unittest.TestCase): |
| def test_successful_variable_extract(self): |
| expected_return = { |
| 'success': True, |
| 'variables': ['lat', 'lon', 'time'] |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time.nc') |
| |
| response = test_app.get('/lfme/list_vars/' + file_location) |
| |
| self.assertDictEqual(expected_return, response.json) |
| |
| def test_successful_variable_extract_jsonp(self): |
| expected_return = { |
| 'success': True, |
| 'variables': ['lat', 'lon', 'time'] |
| } |
| |
| file_location = os.path.abspath('tests/example_data/lat_lon_time.nc') |
| |
| response = test_app.get('/lfme/list_vars/' + file_location + '?callback=test_callback') |
| json = response.text |
| |
| # Strip out the callback functino and the json string from the response |
| # and check for proper content. |
| callback = json[:json.index('(')] |
| json = json[json.index('(') + 1 : json.rindex(')')] |
| json = loads(json) |
| |
| self.assertDictEqual(expected_return, json) |
| self.assertEqual(callback, "test_callback") |
| |
| def test_failure_variable_extract(self): |
| expected_return = {'success': False} |
| |
| response = test_app.get('/lfme/list_vars/' + 'fake_path') |
| |
| self.assertDictEqual(expected_return, response.json) |
| |
| def test_failure_variable_extract_jsonp(self): |
| expected_return = {'success': False} |
| |
| response = test_app.get('/lfme/list_vars//fakepath?callback=test_callback') |
| json = response.text |
| |
| # Strip out the callback functino and the json string from the response |
| # and check for proper content. |
| callback = json[:json.index('(')] |
| json = json[json.index('(') + 1 : json.rindex(')')] |
| json = loads(json) |
| |
| self.assertDictEqual(expected_return, json) |
| self.assertEqual(callback, "test_callback") |
| |
| if __name__ == '__main__': |
| unittest.main() |
