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)