From dc8cd86e03ebc47166b5300958c319a56caec6d9 Mon Sep 17 00:00:00 2001 From: "helmi.nour" Date: Thu, 12 Jan 2023 22:18:30 +0100 Subject: [PATCH] log database engine and transaction id --- rai-sdk/src/test/java/com/relationalai/DatabaseTest.java | 2 ++ rai-sdk/src/test/java/com/relationalai/EngineTest.java | 2 ++ rai-sdk/src/test/java/com/relationalai/ExecuteAsyncTest.java | 2 ++ rai-sdk/src/test/java/com/relationalai/OAuthClientTest.java | 2 ++ rai-sdk/src/test/java/com/relationalai/UnitTest.java | 5 +++++ rai-sdk/src/test/java/com/relationalai/UserTest.java | 2 ++ 6 files changed, 15 insertions(+) diff --git a/rai-sdk/src/test/java/com/relationalai/DatabaseTest.java b/rai-sdk/src/test/java/com/relationalai/DatabaseTest.java index 51f6293a..b7f36500 100644 --- a/rai-sdk/src/test/java/com/relationalai/DatabaseTest.java +++ b/rai-sdk/src/test/java/com/relationalai/DatabaseTest.java @@ -40,6 +40,8 @@ Database find(Database[] databases, String name) { void testDatabase() throws HttpError, InterruptedException, IOException { var client = createClient(); + log.info("database: " + databaseName); + try { client.deleteDatabase(databaseName); } catch (HttpError e) { diff --git a/rai-sdk/src/test/java/com/relationalai/EngineTest.java b/rai-sdk/src/test/java/com/relationalai/EngineTest.java index 5eb9bc6a..f184c498 100644 --- a/rai-sdk/src/test/java/com/relationalai/EngineTest.java +++ b/rai-sdk/src/test/java/com/relationalai/EngineTest.java @@ -37,6 +37,8 @@ Engine find(Engine[] engines, String name) { @Test void testEngine() throws HttpError, InterruptedException, IOException { var client = createClient(); + log.info("engine: " + engineName); + var engine = client.getEngine(engineName); assertEquals(engine.name, engineName); assertEquals(engine.state, "PROVISIONED"); diff --git a/rai-sdk/src/test/java/com/relationalai/ExecuteAsyncTest.java b/rai-sdk/src/test/java/com/relationalai/ExecuteAsyncTest.java index e5778f71..01448692 100644 --- a/rai-sdk/src/test/java/com/relationalai/ExecuteAsyncTest.java +++ b/rai-sdk/src/test/java/com/relationalai/ExecuteAsyncTest.java @@ -45,6 +45,8 @@ void testExecuteAsync() throws HttpError, InterruptedException, IOException, URI var rsp = client.execute(databaseName, engineName, query, true); + log.info("transaction id: " + rsp.transaction.id); + var results = new ArrayList () { { add(new ArrowRelation("/:output/Int64/Int64/Int64/Int64", Arrays.asList(new Object[] {1L, 2L, 3L, 4L, 5L}) )); diff --git a/rai-sdk/src/test/java/com/relationalai/OAuthClientTest.java b/rai-sdk/src/test/java/com/relationalai/OAuthClientTest.java index 100c0b14..f589a494 100644 --- a/rai-sdk/src/test/java/com/relationalai/OAuthClientTest.java +++ b/rai-sdk/src/test/java/com/relationalai/OAuthClientTest.java @@ -36,6 +36,8 @@ public class OAuthClientTest extends UnitTest { @Test void testOAuthClient() throws HttpError, InterruptedException, IOException { var client = createClient(); + log.info("client name: " + clientName); + var rsp = client.findOAuthClient(clientName); if (rsp != null) { client.deleteOAuthClient(rsp.id); diff --git a/rai-sdk/src/test/java/com/relationalai/UnitTest.java b/rai-sdk/src/test/java/com/relationalai/UnitTest.java index 9a3e3a70..3046b57d 100644 --- a/rai-sdk/src/test/java/com/relationalai/UnitTest.java +++ b/rai-sdk/src/test/java/com/relationalai/UnitTest.java @@ -22,12 +22,15 @@ import java.util.Map; import java.util.UUID; import java.util.function.Predicate; +import java.util.logging.Logger; public abstract class UnitTest { static UUID uuid = UUID.randomUUID(); static String databaseName = String.format("java-sdk-test-%s", uuid); static String engineName = String.format("java-sdk-test-%s", uuid); + static final Logger log = Logger.getLogger(UnitTest.class.getName()); + static Config getConfig() throws IOException { var filename = String.format("%s/.rai/config", System.getenv("HOME")); if ((new File(filename)).exists()) { @@ -49,6 +52,7 @@ static Client createClient() throws IOException { var cfg = getConfig(); var customHeaders = (Map) Json.deserialize(getenv("CUSTOM_HEADERS", "{}"), Map.class); + log.info("custom headers: " + Json.serialize(customHeaders)); var testClient = new Client(cfg); var httpHeaders = testClient.getDefaultHttpHeaders(); for (var header : customHeaders.entrySet()) { @@ -59,6 +63,7 @@ static Client createClient() throws IOException { // Ensure that the test database exists. void ensureDatabase() throws HttpError, InterruptedException, IOException { + log.info(String.format("engine: %s, database: %s", engineName, databaseName)); ensureDatabase(createClient()); } diff --git a/rai-sdk/src/test/java/com/relationalai/UserTest.java b/rai-sdk/src/test/java/com/relationalai/UserTest.java index c5d6e9e2..b0e1d21e 100644 --- a/rai-sdk/src/test/java/com/relationalai/UserTest.java +++ b/rai-sdk/src/test/java/com/relationalai/UserTest.java @@ -40,6 +40,8 @@ public class UserTest extends UnitTest { void testUser() throws HttpError, InterruptedException, IOException { var client = createClient(); + log.info("user email: " + userEmail); + var rsp = client.findUser(userEmail); if (rsp != null) { client.deleteUser(rsp.id);