From 205b9ddfde6d8fb41a0b8d4dcf52c15ecc4a8920 Mon Sep 17 00:00:00 2001
From: ROCm CI Service Account <66695075+rocm-ci@users.noreply.github.com>
Date: Sat, 25 Jun 2022 04:25:59 +0530
Subject: [PATCH] SWDEV-306122 - [catch2][dtest] Added test for
 hipStreamBeginCapture API (#2760)

Change-Id: I9cbd9d05178587883a1a2258ebc2c3516a06d5cc


[ROCm/hip-tests commit: 117f4c484c3e94f19a174bec7616e75a6986490b]
---
 .../catch/unit/graph/hipStreamBeginCapture.cc | 45 ++++++++++++++++++-
 1 file changed, 44 insertions(+), 1 deletion(-)

diff --git a/projects/hip-tests/catch/unit/graph/hipStreamBeginCapture.cc b/projects/hip-tests/catch/unit/graph/hipStreamBeginCapture.cc
index b1d4dfdd06..af550da759 100644
--- a/projects/hip-tests/catch/unit/graph/hipStreamBeginCapture.cc
+++ b/projects/hip-tests/catch/unit/graph/hipStreamBeginCapture.cc
@@ -18,7 +18,7 @@ THE SOFTWARE.
 */
 
 /**
-Testcase Scenarios :
+Testcase Scenarios : Functional
  1) Initiate stream capture with different modes on custom stream.
  Capture stream sequence and replay the sequence in multiple iterations.
  2) End capture and validate that API returns captured graph for
@@ -183,3 +183,46 @@ TEST_CASE("Unit_hipStreamBeginCapture_hipStreamPerThread") {
   HIP_CHECK(hipFree(A_d));
   HIP_CHECK(hipFree(C_d));
 }
+
+
+/* Test verifies hipStreamBeginCapture API Negative scenarios.
+ */
+
+TEST_CASE("Unit_hipStreamBeginCapture_Negative") {
+  hipError_t ret;
+  hipStream_t stream{};
+  HIP_CHECK(hipStreamCreate(&stream));
+
+  SECTION("Stream capture on legacy/null stream returns error code.") {
+    ret = hipStreamBeginCapture(nullptr, hipStreamCaptureModeGlobal);
+    REQUIRE(hipErrorStreamCaptureUnsupported == ret);
+  }
+  SECTION("Capturing hipStream status with same stream again") {
+    HIP_CHECK(hipStreamBeginCapture(stream, hipStreamCaptureModeGlobal));
+    ret = hipStreamBeginCapture(stream, hipStreamCaptureModeGlobal);
+    REQUIRE(hipErrorIllegalState == ret);
+  }
+  SECTION("Creating hipStream with invalid mode") {
+    ret = hipStreamBeginCapture(stream, hipStreamCaptureMode(-1));
+    REQUIRE(hipErrorInvalidValue == ret);
+  }
+  HIP_CHECK(hipStreamDestroy(stream));
+}
+
+TEST_CASE("Unit_hipStreamBeginCapture_Basic") {
+  hipStream_t s1, s2, s3;
+
+  HIP_CHECK(hipStreamCreate(&s1));
+  HIP_CHECK(hipStreamBeginCapture(s1, hipStreamCaptureModeGlobal));
+
+  HIP_CHECK(hipStreamCreate(&s2));
+  HIP_CHECK(hipStreamBeginCapture(s2, hipStreamCaptureModeThreadLocal));
+
+  HIP_CHECK(hipStreamCreate(&s3));
+  HIP_CHECK(hipStreamBeginCapture(s3, hipStreamCaptureModeRelaxed));
+
+  HIP_CHECK(hipStreamDestroy(s1));
+  HIP_CHECK(hipStreamDestroy(s2));
+  HIP_CHECK(hipStreamDestroy(s3));
+}
+