@Test
public void testSetTimeZone() throws IoTDBSQLException, TException {

  String timeZone = "Asia/Shanghai";
  when(client.setTimeZone(any(TSSetTimeZoneReq.class)))
      .thenReturn(new TSStatus(successStatus));
  connection.setClient(client);
  connection.setTimeZone(timeZone);
  assertEquals(connection.getTimeZone(), timeZone);
}