package com.linkedin.thirdeye.datalayer.bao;
import java.util.List;
import org.testng.Assert;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import com.linkedin.thirdeye.datalayer.dto.IngraphDashboardConfigDTO;
public class TestIngraphDatasetConfigManager extends AbstractManagerTestBase {
private Long ingraphDashboardConfigId1 = null;
private Long ingraphDashboardConfigId2 = null;
private static String name1 = "dashboard1";
private static String name2 = "dashboard2";
@BeforeClass
void beforeClass() {
super.init();
}
@AfterClass(alwaysRun = true)
void afterClass() {
super.cleanup();
}
@Test
public void testCreateIngraphDashboard() {
IngraphDashboardConfigDTO ingraphDashboardConfigDTO1 = getTestIngraphDashboardConfig(name1);
ingraphDashboardConfigId1 = ingraphDashboardConfigDAO.save(ingraphDashboardConfigDTO1);
Assert.assertNotNull(ingraphDashboardConfigId1);
IngraphDashboardConfigDTO ingraphDashboardConfigDTO2 = getTestIngraphDashboardConfig(name2);
ingraphDashboardConfigId2 = ingraphDashboardConfigDAO.save(ingraphDashboardConfigDTO2);
Assert.assertNotNull(ingraphDashboardConfigId2);
}
@Test(dependsOnMethods = {"testCreateIngraphDashboard"})
public void testFindIngraphDashboard() {
List<IngraphDashboardConfigDTO> ingraphDashboardConfigDTOs = ingraphDashboardConfigDAO.findAll();
Assert.assertEquals(ingraphDashboardConfigDTOs.size(), 2);
IngraphDashboardConfigDTO ingraphDashboardConfigDTO = ingraphDashboardConfigDAO.findByName(name1);
Assert.assertEquals(ingraphDashboardConfigDTO.getName(), name1);
}
@Test(dependsOnMethods = { "testFindIngraphDashboard" })
public void testUpdateIngraphDashboard() {
IngraphDashboardConfigDTO ingraphDashboardConfigDTO = ingraphDashboardConfigDAO.findById(ingraphDashboardConfigId1);
Assert.assertNotNull(ingraphDashboardConfigDTO);
Assert.assertTrue(ingraphDashboardConfigDTO.isBootstrap());
ingraphDashboardConfigDTO.setBootstrap(false);
ingraphDashboardConfigDAO.update(ingraphDashboardConfigDTO);
ingraphDashboardConfigDTO = ingraphDashboardConfigDAO.findById(ingraphDashboardConfigId1);
Assert.assertNotNull(ingraphDashboardConfigDTO);
Assert.assertFalse(ingraphDashboardConfigDTO.isBootstrap());
}
@Test(dependsOnMethods = { "testUpdateIngraphDashboard" })
public void testDeleteIngraphDashboard() {
ingraphDashboardConfigDAO.deleteById(ingraphDashboardConfigId1);
IngraphDashboardConfigDTO ingraphDashboardConfigDTO = ingraphDashboardConfigDAO.findById(ingraphDashboardConfigId1);
Assert.assertNull(ingraphDashboardConfigDTO);
}
}