Index: tools/telemetry/third_party/gsutilz/third_party/protorpc/protorpc/message_types_test.py |
diff --git a/tools/telemetry/third_party/gsutilz/third_party/protorpc/protorpc/message_types_test.py b/tools/telemetry/third_party/gsutilz/third_party/protorpc/protorpc/message_types_test.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..b061cdfa1db24d35b07513f3f100b22c3e3489a0 |
--- /dev/null |
+++ b/tools/telemetry/third_party/gsutilz/third_party/protorpc/protorpc/message_types_test.py |
@@ -0,0 +1,88 @@ |
+#!/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() |