| Index: third_party/google-endpoints/apitools/base/protorpclite/message_types_test.py
|
| diff --git a/third_party/google-endpoints/apitools/base/protorpclite/message_types_test.py b/third_party/google-endpoints/apitools/base/protorpclite/message_types_test.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..8a5afdb7f3cab9b41b60a91a9a7897f448e01cc0
|
| --- /dev/null
|
| +++ b/third_party/google-endpoints/apitools/base/protorpclite/message_types_test.py
|
| @@ -0,0 +1,86 @@
|
| +#!/usr/bin/env python
|
| +#
|
| +# Copyright 2013 Google Inc.
|
| +#
|
| +# 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.
|
| +#
|
| +
|
| +"""Tests for apitools.base.protorpclite.message_types."""
|
| +import datetime
|
| +import unittest
|
| +
|
| +from apitools.base.protorpclite import message_types
|
| +from apitools.base.protorpclite import messages
|
| +from apitools.base.protorpclite import test_util
|
| +from apitools.base.protorpclite import util
|
| +
|
| +
|
| +class ModuleInterfaceTest(test_util.ModuleInterfaceTest,
|
| + test_util.TestCase):
|
| +
|
| + MODULE = message_types
|
| +
|
| +
|
| +class DateTimeFieldTest(test_util.TestCase):
|
| +
|
| + def testValueToMessage(self):
|
| + field = message_types.DateTimeField(1)
|
| + message = field.value_to_message(
|
| + datetime.datetime(2033, 2, 4, 11, 22, 10))
|
| + self.assertEqual(
|
| + message_types.DateTimeMessage(milliseconds=1991128930000), message)
|
| +
|
| + def testValueToMessageBadValue(self):
|
| + field = message_types.DateTimeField(1)
|
| + self.assertRaisesWithRegexpMatch(
|
| + messages.EncodeError,
|
| + 'Expected type datetime, got int: 20',
|
| + field.value_to_message, 20)
|
| +
|
| + def testValueToMessageWithTimeZone(self):
|
| + time_zone = util.TimeZoneOffset(60 * 10)
|
| + field = message_types.DateTimeField(1)
|
| + message = field.value_to_message(
|
| + datetime.datetime(2033, 2, 4, 11, 22, 10, tzinfo=time_zone))
|
| + self.assertEqual(
|
| + message_types.DateTimeMessage(milliseconds=1991128930000,
|
| + time_zone_offset=600),
|
| + message)
|
| +
|
| + def testValueFromMessage(self):
|
| + message = message_types.DateTimeMessage(milliseconds=1991128000000)
|
| + field = message_types.DateTimeField(1)
|
| + timestamp = field.value_from_message(message)
|
| + self.assertEqual(datetime.datetime(2033, 2, 4, 11, 6, 40),
|
| + timestamp)
|
| +
|
| + def testValueFromMessageBadValue(self):
|
| + field = message_types.DateTimeField(1)
|
| + self.assertRaisesWithRegexpMatch(
|
| + messages.DecodeError,
|
| + 'Expected type DateTimeMessage, got VoidMessage: <VoidMessage>',
|
| + field.value_from_message, message_types.VoidMessage())
|
| +
|
| + def testValueFromMessageWithTimeZone(self):
|
| + message = message_types.DateTimeMessage(milliseconds=1991128000000,
|
| + time_zone_offset=300)
|
| + field = message_types.DateTimeField(1)
|
| + timestamp = field.value_from_message(message)
|
| + time_zone = util.TimeZoneOffset(60 * 5)
|
| + self.assertEqual(
|
| + datetime.datetime(2033, 2, 4, 11, 6, 40, tzinfo=time_zone),
|
| + timestamp)
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + unittest.main()
|
|
|