From bebdb2bef9369cb9e034abe91e7d38dd9d947889 Mon Sep 17 00:00:00 2001
From: caijq <callen.cai@rock-chips.com>
Date: Thu, 13 Apr 2023 10:45:56 +0800
Subject: [PATCH] CEC : fix VtsHalTvCecV1_0TargetTest SendMessage test failed.

If CEC DUT is a TV device type,we should reset logical address to tv,
and then set message.initiator to tv.

Bug: 277715429
Test: run VTS

Change-Id: If7f7b9ddce182e5de80c91a30c4ec18294459fbf
Signed-off-by: caijq <callen.cai@rock-chips.com>
---
 .../1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp   | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp b/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp
index 7b426899b1..75c44b7c90 100644
--- a/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp
+++ b/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp
@@ -127,7 +127,15 @@ TEST_P(HdmiCecTest, PhysicalAddress) {
 
 TEST_P(HdmiCecTest, SendMessage) {
     CecMessage message;
-    message.initiator = CecLogicalAddress::PLAYBACK_1;
+    if (hasDeviceType(CecDeviceType::TV))
+    {
+        hdmiCec->clearLogicalAddress();
+        Return<Result> result = hdmiCec->addLogicalAddress(CecLogicalAddress::TV);
+        EXPECT_EQ(result, Result::SUCCESS);
+        message.initiator = CecLogicalAddress::TV;
+    }
+    else
+        message.initiator = CecLogicalAddress::PLAYBACK_1;
     message.destination = CecLogicalAddress::BROADCAST;
     message.body.resize(1);
     message.body[0] = 131;