ReassignOwnedTest.java

Go to the documentation of this file.

package com.mapd.parser.extension.ddl;

import static org.junit.Assert.assertEquals;

import com.google.gson.JsonObject;

import org.junit.Test;

import ai.heavy.thrift.calciteserver.TPlanResult;

public class ReassignOwnedTest extends DDLTest {
  public ReassignOwnedTest() {
    resourceDirPath = ReassignOwnedTest.class.getClassLoader().getResource("").getPath();
    jsonTestDir = "reassignowned";
  }

  @Test
  public void reassignOwned() throws Exception {
    final JsonObject expectedJsonObject = getJsonFromFile("reassign_owned.json");
    final TPlanResult result =
            processDdlCommand("REASSIGN OWNED BY user_1, user_2 To user_3;");
    final JsonObject actualJsonObject =
            gson.fromJson(result.plan_result, JsonObject.class);
    assertEquals(expectedJsonObject, actualJsonObject);
  }

  @Test
  public void reassignAllOwned() throws Exception {
    final JsonObject expectedJsonObject = getJsonFromFile("reassign_all_owned.json");
    final TPlanResult result =
            processDdlCommand("REASSIGN ALL OWNED BY user_1, user_2 To user_3;");
    final JsonObject actualJsonObject =
            gson.fromJson(result.plan_result, JsonObject.class);
    assertEquals(expectedJsonObject, actualJsonObject);
  }
}