def test_handle_empty_chroot_message(self):
"""Test handling of an empty chroot message."""
message = build_api_test_pb2.TestRequestMessage()
empty_chroot = chroot_lib.Chroot(env={'FEATURES': 'separatedebug'})
chroot_handler = field_handler.ChrootHandler(clear_field=False)
chroot = chroot_handler.handle(message)
self.assertEqual(empty_chroot, chroot)
def test_handle_success(self):
"""Test a successful Chroot message parse from a parent message."""
message = build_api_test_pb2.TestRequestMessage()
message.chroot.path = self.path
message.chroot.cache_dir = self.cache_dir
message.chroot.chrome_dir = self.chrome_dir
message.chroot.env.features.add().feature = 'thing'
# First a no-clear parse.
chroot_handler = field_handler.ChrootHandler(clear_field=False)
chroot = chroot_handler.handle(message)
self.assertEqual(self.expected_chroot, chroot)
self.assertEqual(message.chroot.path, self.path)
# A clear field parse.
clear_chroot_handler = field_handler.ChrootHandler(clear_field=True)
chroot = clear_chroot_handler.handle(message)
self.assertEqual(self.expected_chroot, chroot)
self.assertFalse(message.chroot.path)
def test_parse_chroot_success(self):
"""Test successful Chroot message parse."""
chroot_msg = common_pb2.Chroot()
chroot_msg.path = self.path
chroot_msg.cache_dir = self.cache_dir
chroot_msg.chrome_dir = self.chrome_dir
chroot_msg.env.features.add().feature = 'thing'
chroot_handler = field_handler.ChrootHandler(clear_field=False)
parsed_chroot = chroot_handler.parse_chroot(chroot_msg)
self.assertEqual(self.expected_chroot, parsed_chroot)