You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

178 lines
5.1 KiB

# Copyright 2014 Google Inc. All Rights Reserved.
#
# Licensed 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.
"""Unit tests for googleapiclient.schema."""
from __future__ import absolute_import
__author__ = "jcgregorio@google.com (Joe Gregorio)"
import json
import os
import unittest2 as unittest
from googleapiclient.schema import Schemas
DATA_DIR = os.path.join(os.path.dirname(__file__), "data")
def datafile(filename):
return os.path.join(DATA_DIR, filename)
LOAD_FEED = """{
"items": [
{
"longVal": 42,
"kind": "zoo#loadValue",
"enumVal": "A String",
"anyVal": "", # Anything will do.
"nullVal": None,
"stringVal": "A String",
"doubleVal": 3.14,
"booleanVal": True or False, # True or False.
},
],
"kind": "zoo#loadFeed",
}"""
class SchemasTest(unittest.TestCase):
def setUp(self):
f = open(datafile("zoo.json"))
discovery = f.read()
f.close()
discovery = json.loads(discovery)
self.sc = Schemas(discovery)
def test_basic_formatting(self):
self.assertEqual(
sorted(LOAD_FEED.splitlines()),
sorted(self.sc.prettyPrintByName("LoadFeed").splitlines()),
)
def test_empty_edge_case(self):
self.assertTrue("Unknown type" in self.sc.prettyPrintSchema({}))
def test_simple_object(self):
self.assertEqual({}, eval(self.sc.prettyPrintSchema({"type": "object"})))
def test_string(self):
self.assertEqual(
type(""), type(eval(self.sc.prettyPrintSchema({"type": "string"})))
)
def test_integer(self):
self.assertEqual(
type(20), type(eval(self.sc.prettyPrintSchema({"type": "integer"})))
)
def test_number(self):
self.assertEqual(
type(1.2), type(eval(self.sc.prettyPrintSchema({"type": "number"})))
)
def test_boolean(self):
self.assertEqual(
type(True), type(eval(self.sc.prettyPrintSchema({"type": "boolean"})))
)
def test_string_default(self):
self.assertEqual(
"foo", eval(self.sc.prettyPrintSchema({"type": "string", "default": "foo"}))
)
def test_integer_default(self):
self.assertEqual(
20, eval(self.sc.prettyPrintSchema({"type": "integer", "default": 20}))
)
def test_number_default(self):
self.assertEqual(
1.2, eval(self.sc.prettyPrintSchema({"type": "number", "default": 1.2}))
)
def test_boolean_default(self):
self.assertEqual(
False,
eval(self.sc.prettyPrintSchema({"type": "boolean", "default": False})),
)
def test_null(self):
self.assertEqual(None, eval(self.sc.prettyPrintSchema({"type": "null"})))
def test_any(self):
self.assertEqual("", eval(self.sc.prettyPrintSchema({"type": "any"})))
def test_array(self):
self.assertEqual(
[{}],
eval(
self.sc.prettyPrintSchema(
{"type": "array", "items": {"type": "object"}}
)
),
)
def test_nested_references(self):
feed = {
"items": [
{
"photo": {
"hash": "A String",
"hashAlgorithm": "A String",
"filename": "A String",
"type": "A String",
"size": 42,
},
"kind": "zoo#animal",
"etag": "A String",
"name": "A String",
}
],
"kind": "zoo#animalFeed",
"etag": "A String",
}
self.assertEqual(feed, eval(self.sc.prettyPrintByName("AnimalFeed")))
def test_additional_properties(self):
items = {
"animals": {
"a_key": {
"photo": {
"hash": "A String",
"hashAlgorithm": "A String",
"filename": "A String",
"type": "A String",
"size": 42,
},
"kind": "zoo#animal",
"etag": "A String",
"name": "A String",
}
},
"kind": "zoo#animalMap",
"etag": "A String",
}
self.assertEqual(items, eval(self.sc.prettyPrintByName("AnimalMap")))
def test_unknown_name(self):
self.assertRaises(KeyError, self.sc.prettyPrintByName, "UknownSchemaThing")
if __name__ == "__main__":
unittest.main()