From 0ebb24b4b76891453d49266a7be463534d20f7a2 Mon Sep 17 00:00:00 2001
From: Shraddha Basantwani <shraddha.basantwani@ittiam.com>
Date: Thu, 21 Jan 2021 14:18:36 +0530
Subject: [PATCH] CEC: Add VTS test to clear and add logical address

Test: atest VtsHalTvCecV1_0TargetTest
Bug: 172316717
Change-Id: I9b66082ed360a158e18deb845e8888a39a2a299f
---
 tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp b/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp
index db6839d584..9daefde4b8 100644
--- a/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp
+++ b/tv/cec/1.0/vts/functional/VtsHalTvCecV1_0TargetTest.cpp
@@ -28,6 +28,7 @@
 using ::android::sp;
 using ::android::hardware::hidl_death_recipient;
 using ::android::hardware::Return;
+using ::android::hardware::tv::cec::V1_0::CecLogicalAddress;
 using ::android::hardware::tv::cec::V1_0::IHdmiCec;
 using ::android::hardware::tv::cec::V1_0::Result;
 
@@ -66,6 +67,12 @@ INSTANTIATE_TEST_SUITE_P(
         testing::ValuesIn(android::hardware::getAllHalInstanceNames(IHdmiCec::descriptor)),
         android::hardware::PrintInstanceNameToString);
 
+TEST_P(HdmiCecTest, ClearAddLogicalAddress) {
+    hdmiCec->clearLogicalAddress();
+    Return<Result> ret = hdmiCec->addLogicalAddress(CecLogicalAddress::PLAYBACK_3);
+    EXPECT_EQ(ret, Result::SUCCESS);
+}
+
 TEST_P(HdmiCecTest, CecVersion) {
     Return<int32_t> ret = hdmiCec->getCecVersion();
     EXPECT_GE(ret, CEC_VERSION);