From 51db9a049e41e2ae594b194e1def9c0fce9da9f4 Mon Sep 17 00:00:00 2001
From: Hector Cao <hector.cao@canonical.com>
Date: Tue, 22 Oct 2024 10:42:28 +0000
Subject: [PATCH] tests : add stress test for quote generation

---
 tests/tests/test_stress_quote.py | 40 ++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 tests/tests/test_stress_quote.py

diff --git a/tests/tests/test_stress_quote.py b/tests/tests/test_stress_quote.py
new file mode 100644
index 0000000..b111538
--- /dev/null
+++ b/tests/tests/test_stress_quote.py
@@ -0,0 +1,40 @@
+#!/usr/bin/env python3
+#
+# Copyright 2024 Canonical Ltd.
+#
+# This program is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License version 3, as published
+# by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranties of MERCHANTABILITY,
+# SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+import Qemu
+
+def test_stress_tdxattest_tsm():
+    """
+    Stress test on quote generation
+    Have a loop to generate 200 quotes
+    """
+    with Qemu.QemuMachine() as qm:
+        machine = qm.qcmd.plugins['machine']
+        machine.enable_quote_socket()
+
+        qm.run()
+        ssh = Qemu.QemuSSH(qm)
+
+        ssh.check_exec('rm -f /etc/tdx-attest.conf')
+        nb_iterations = 200
+        stdout, _ = ssh.check_exec(f'''
+            count={nb_iterations}
+            for i in $(seq $count); do
+              /usr/share/doc/libtdx-attest-dev/examples/test_tdx_attest | grep "Successfully get the TD Quote"
+            done
+            ''')
+        assert stdout.read().decode().count('Successfully get the TD Quote') == nb_iterations