| #!/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 protorpc.message_types.""" |
| |
| __author__ = 'rafek@google.com (Rafe Kaplan)' |
| |
| |
| import datetime |
| |
| import unittest |
| |
| from protorpc import message_types |
| from protorpc import messages |
| from protorpc import test_util |
| from protorpc 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() |