Add test when try to convert datetime to Duration, and try to convert timedelta to Timestamp. It raises AttributeError but will turn to TypeError in OSS 34.0 release.

PiperOrigin-RevId: 797464099

Author: anandolee

python/google/protobuf/internal/duration_test.py

@@ -93,6 +93,11 @@ def test_duration_add_annotation(self):
     # Duration + Duration
     self.assertEqual(dr + dr2, dr2 + dr)
 
+  def test_assign_datetime_to_duration(self):
+    message = well_known_types_test_pb2.WKTMessage()
+    with self.assertRaises((TypeError, AttributeError)):
+      message.optional_duration = datetime.datetime.now()
+
 
 if __name__ == '__main__':
   unittest.main()

python/google/protobuf/internal/timestamp_test.py

@@ -90,6 +90,11 @@ def test_timestamp_add_annotation(self):
     # Timestamp + Duration and Duration + Timestamp
     self.assertEqual(ts + msg.optional_duration, msg.optional_duration + ts)
 
+  def test_assign_duration_to_timestamp(self):
+    message = well_known_types_test_pb2.WKTMessage()
+    with self.assertRaises((TypeError, AttributeError)):
+      message.optional_timestamp = datetime.timedelta(microseconds=123)
+
 
 if __name__ == '__main__':
   unittest.main()

python/google/protobuf/internal/well_known_types_test.py

@@ -540,10 +540,10 @@ def testInvalidTimestamp(self):
     self.assertRaisesRegex(ValueError, 'Timestamp is not valid',
                            message.FromSeconds, -62135596801)
     msg = well_known_types_test_pb2.WKTMessage()
-    with self.assertRaises(AttributeError):
+    with self.assertRaises((TypeError, AttributeError)):
       msg.optional_timestamp = 1
 
-    with self.assertRaises(AttributeError):
+    with self.assertRaises((TypeError, AttributeError)):
       msg2 = well_known_types_test_pb2.WKTMessage(optional_timestamp=1)
 
     with self.assertRaises(TypeError):
@@ -606,10 +606,10 @@ def testInvalidDuration(self):
         message.ToJsonString,
     )
     msg = well_known_types_test_pb2.WKTMessage()
-    with self.assertRaises(AttributeError):
+    with self.assertRaises((TypeError, AttributeError)):
       msg.optional_duration = 1
 
-    with self.assertRaises(AttributeError):
+    with self.assertRaises((TypeError, AttributeError)):
       msg2 = well_known_types_test_pb2.WKTMessage(optional_duration=1)
 
     with self.assertRaises(TypeError):