From d2602bbf3800a19231a5fb7e0c84be4e2144aefc Mon Sep 17 00:00:00 2001
From: Hirotaka Wakabayashi <hiwkby@yahoo.com>
Date: Tue, 23 May 2023 08:26:13 +0900
Subject: [PATCH] Adds postgres guestagent test driver

This PR adds postgres guestagent test driver.

Story: 2010761
Task: 48059
Change-Id: Id15129cbd93502ae06cb600b31553c1050ccd9ca
---
 .../datastore/postgres/test_service.py        | 54 +++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 trove/tests/unittests/guestagent/datastore/postgres/test_service.py

diff --git a/trove/tests/unittests/guestagent/datastore/postgres/test_service.py b/trove/tests/unittests/guestagent/datastore/postgres/test_service.py
new file mode 100644
index 0000000000..fac0aa54f5
--- /dev/null
+++ b/trove/tests/unittests/guestagent/datastore/postgres/test_service.py
@@ -0,0 +1,54 @@
+#    Licensed under the Apache License, Version 2.0 (the "License");
+#    you may not use this file except in compliance with the License.
+#    You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS,
+#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#    See the License for the specific language governing permissions and
+#    limitations under the License.
+
+from unittest import mock
+
+from trove.guestagent.datastore.postgres import service
+from trove.tests.unittests import trove_testtools
+
+statement = (
+    "SELECT usename FROM pg_catalog.pg_user WHERE usesuper = true"
+)
+username = "postgres"
+port = 5432
+result = [(1, "one"), (2, 'two'), (3, 'three')]
+
+
+class TestPostgresConnection(trove_testtools.TestCase):
+    def setUp(self):
+        super(TestPostgresConnection, self).setUp()
+
+    # execute is expected to returns nothing
+    @mock.patch(
+        'trove.guestagent.datastore.postgres.service.PostgresConnection')
+    def test_execute(self, mock_postgres_connection):
+        postgres_connection = mock.MagicMock()
+        postgres_connection.execute = mock.MagicMock(return_value=None)
+        mock_postgres_connection.return_value = postgres_connection
+
+        # assertion here
+        connection = service.PostgresConnection(username, port=port)
+        self.assertIsNone(connection.execute(statement),
+                          'postgres_connection_execute does not returns None')
+
+    # query is expected to returns result
+    @mock.patch(
+        'trove.guestagent.datastore.postgres.service.PostgresConnection')
+    def test_query(self, mock_postgres_connection):
+        postgres_connection = mock.MagicMock()
+        postgres_connection.query = mock.MagicMock(return_value=result)
+        mock_postgres_connection.return_value = postgres_connection
+
+        # assertion here
+        connection = service.PostgresConnection(username, port=port)
+        self.assertEqual(result, connection.query(statement),
+                         'postgres_connection_query does not returns expected')