diff --git a/src/test/java/net/snowflake/ingest/streaming/internal/SnowflakeStreamingIngestClientTest.java b/src/test/java/net/snowflake/ingest/streaming/internal/SnowflakeStreamingIngestClientTest.java index babb1a4f9..0e1a1eda4 100644 --- a/src/test/java/net/snowflake/ingest/streaming/internal/SnowflakeStreamingIngestClientTest.java +++ b/src/test/java/net/snowflake/ingest/streaming/internal/SnowflakeStreamingIngestClientTest.java @@ -1325,35 +1325,12 @@ public void testRegisterBlobSuccessResponseWithEncryptionKeys() throws Exception client.registerBlobs(blobs); Assert.assertEquals(1, client.getEncryptionKeysPerTable().size()); Assert.assertEquals( - "DB_STREAMINGINGEST", - client - .getEncryptionKeysPerTable() - .get(fqn) - .getDatabaseName()); - Assert.assertEquals( - "PUBLIC", - client - .getEncryptionKeysPerTable() - .get(fqn) - .getSchemaName()); - Assert.assertEquals( - "T_STREAMINGINGEST", - client - .getEncryptionKeysPerTable() - .get(fqn) - .getTableName()); - Assert.assertEquals( - "key", - client - .getEncryptionKeysPerTable() - .get(fqn) - .getEncryptionKey()); + "DB_STREAMINGINGEST", client.getEncryptionKeysPerTable().get(fqn).getDatabaseName()); + Assert.assertEquals("PUBLIC", client.getEncryptionKeysPerTable().get(fqn).getSchemaName()); Assert.assertEquals( - 1234, - client - .getEncryptionKeysPerTable() - .get(fqn) - .getEncryptionKeyId()); + "T_STREAMINGINGEST", client.getEncryptionKeysPerTable().get(fqn).getTableName()); + Assert.assertEquals("key", client.getEncryptionKeysPerTable().get(fqn).getEncryptionKey()); + Assert.assertEquals(1234, client.getEncryptionKeysPerTable().get(fqn).getEncryptionKeyId()); } @Test