@Test
public void testPersistEntity_EntityWithArray() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithArray()");
System.out.println("------------------------------------------------------------");
Filter filter = new Filter();
List<String> values = new ArrayList<String>();
values.add("Value1");
values.add("Value2");
filter.setValues(values);
String[] simpleArray = new String[] {"Kayal", "Vizhi"};
filter.setSimpleArray(simpleArray);
Tuple[] complexArray = new Tuple[] {new Tuple("a", "A"), new Tuple("b", "B")};
filter.setComplexArray(complexArray);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertEquals(filter.getValues(), retrievedFilter.getValues());
assertEquals(complexArray.length, retrievedFilter.getComplexArray().length);
for (int i = 0; i < complexArray.length; i++) {
assertEquals(complexArray[i], retrievedFilter.getComplexArray()[i]);
}
}
@Test
public void testPersistEntity_complexObject() {
System.out.println("\nTestPersistenceManager.testPersistEntity_complexObject()");
System.out.println("----------------------------------------------------------");
ScriptFile roleSF = new ScriptFile();
roleSF.setName("roleSFName1");
roleSF.setPath("roleSFPath1");
assertNull(roleSF.getId());
ScriptFile playBookSF1 = new ScriptFile();
playBookSF1.setName("playBookSFName1");
playBookSF1.setPath("playBookSFPath1");
assertNull(playBookSF1.getId());
ScriptFile playBookSF2 = new ScriptFile();
playBookSF2.setName("playBookSFName2");
playBookSF2.setPath("playBookSFPath2");
assertNull(playBookSF2.getId());
AnsibleModuleDefinition anModDefinition = new AnsibleModuleDefinition();
List<ScriptFile> playBookSFList = new ArrayList<ScriptFile>();
playBookSFList.add(playBookSF1);
playBookSFList.add(playBookSF2);
Set<ScriptFile> roleSFList = new HashSet<ScriptFile>();
roleSFList.add(roleSF);
anModDefinition.setPlayBooks(playBookSFList);
anModDefinition.setRoles(roleSFList);
assertNull(anModDefinition.getId());
persistenceManager.persistEntity(anModDefinition);
List<AnsibleModuleDefinition> entities =
persistenceManager.getEntities(AnsibleModuleDefinition.class);
assertNotNull(entities);
assertEquals(1, entities.size());
AnsibleModuleDefinition retrievedDefinition = entities.get(0);
assertNotNull(retrievedDefinition);
assertNotNull(retrievedDefinition.getId());
assertEquals(2, retrievedDefinition.getPlayBooks().size());
assertEquals(1, retrievedDefinition.getRoles().size());
}
@Test
public void testPersistEntity_simpleObject() {
System.out.println("\nTestPersistenceManager.testPersistEntity_simpleObject()");
System.out.println("---------------------------------------------------------");
ScriptFile roleSF = new ScriptFile();
roleSF.setName("roleSFName1");
roleSF.setPath("roleSFPath1");
assertNull(roleSF.getId());
persistenceManager.persistEntity(roleSF);
assertNotNull(roleSF.getId());
List<ScriptFile> entities = persistenceManager.getEntities(ScriptFile.class);
assertNotNull(entities);
assertEquals(1, entities.size());
assertEquals(roleSF.getName(), entities.get(0).getName());
assertEquals(roleSF.getPath(), entities.get(0).getPath());
}
@Test
public void testPersistEntity_EntityWithArrayOfMixedType() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithArrayOfMixedType()");
System.out.println(
"----------------------------------------------------------------------------");
Filter filter = new Filter();
Object[] objects = new Object[] {"Hi", new Integer(1)};
filter.setMixedArray(objects);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertNull(retrievedFilter.getMixedArray());
assertNotSame(filter.getMixedArray(), retrievedFilter.getMixedArray());
}
@Test
public void testPersistEntity_EntityWithMap() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithMap()");
System.out.println("----------------------------------------------------------");
Task task = new Task();
task.setName("NewTask");
Map<String, String> taskEntriesMap = new HashMap<String, String>();
taskEntriesMap.put("Key1", "Value1");
taskEntriesMap.put("Key1", "Value1");
task.setTaskEntries(taskEntriesMap);
persistenceManager.persistEntity(task);
List<Task> tasks = persistenceManager.getEntities(Task.class);
assertNotNull(tasks);
assertEquals(1, tasks.size());
Task retrievedTask = tasks.get(0);
assertNotNull(retrievedTask.getId());
assertEquals(task.getName(), retrievedTask.getName());
assertEquals(task.getTaskEntries(), retrievedTask.getTaskEntries());
}
@Test
public void testPersistEntity_EntityWithDate() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithDate()");
System.out.println("------------------------------------------------------------");
DateHolder dateHolder = new DateHolder();
dateHolder.setUtilDate(new Date());
dateHolder.setSqlDate(new java.sql.Date(System.currentTimeMillis()));
List<Date> utilDatesList = new ArrayList<Date>(2);
utilDatesList.add(new Date());
utilDatesList.add(new Date());
Date[] utilDateArray = new Date[] {new Date(), new Date()};
List<java.sql.Date> sqlDatesList = new ArrayList<java.sql.Date>(2);
sqlDatesList.add(new java.sql.Date(System.currentTimeMillis()));
sqlDatesList.add(new java.sql.Date(System.currentTimeMillis()));
java.sql.Date[] sqlDatesArray =
new java.sql.Date[] {
new java.sql.Date(System.currentTimeMillis()),
new java.sql.Date(System.currentTimeMillis())
};
dateHolder.setSqlDatesList(sqlDatesList);
dateHolder.setUtilDatesList(utilDatesList);
dateHolder.setSqlDatesArray(sqlDatesArray);
dateHolder.setUtilDatesArray(utilDateArray);
persistenceManager.persistEntity(dateHolder);
List<DateHolder> dateHolders = persistenceManager.getEntities(DateHolder.class);
assertNotNull(dateHolders);
assertEquals(1, dateHolders.size());
assertEquals(dateHolder.getUtilDate(), dateHolders.get(0).getUtilDate());
assertEquals(dateHolder.getSqlDate(), dateHolders.get(0).getSqlDate());
}
@Test
public void testPersistEntity_EntityWithCollectionOfMixedType() {
System.out.println(
"\nTestPersistenceManager.testPersistEntity_EntityWithCollectionOfMixedType()");
System.out.println(
"----------------------------------------------------------------------------");
Filter filter = new Filter();
List<Object> mixedValues = new ArrayList<Object>();
mixedValues.add("Hi");
mixedValues.add(0);
filter.setMixedValues(mixedValues);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertNull(retrievedFilter.getMixedValues());
assertNotSame(filter.getMixedValues(), retrievedFilter.getMixedValues());
}
@Test
public void testPersistEntity_EntityWithSetOfSimpleType() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithSetOfSimpleType()");
System.out.println("-----------------------------------------------------------------------");
IpPermissionInfo ipPermInfo = new IpPermissionInfo();
List<String> ipRanges = new ArrayList<String>();
ipRanges.add("ipRange1");
ipRanges.add("ipRange2");
ipPermInfo.setIpRanges(ipRanges);
Set<String> groups = new HashSet<String>();
groups.add("groupId1");
groups.add("groupId2");
ipPermInfo.setGroupIds(groups);
persistenceManager.persistEntity(ipPermInfo);
List<IpPermissionInfo> ipPermissionInfos =
persistenceManager.getEntities(IpPermissionInfo.class);
assertNotNull(ipPermissionInfos);
assertEquals(1, ipPermissionInfos.size());
IpPermissionInfo retrievedIpPermissionInfo = ipPermissionInfos.get(0);
assertNotNull(retrievedIpPermissionInfo.getId());
assertEquals(ipPermInfo.getIpRanges().size(), retrievedIpPermissionInfo.getIpRanges().size());
assertEquals(ipPermInfo.getGroupIds().size(), retrievedIpPermissionInfo.getGroupIds().size());
for (String ipRange : ipRanges) {
assertTrue(retrievedIpPermissionInfo.getIpRanges().contains(ipRange));
}
for (String group : groups) {
assertTrue(retrievedIpPermissionInfo.getGroupIds().contains(group));
}
}
@Test
public void testPersistEntity_EntityWithEnum() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithEnum()");
System.out.println("------------------------------------------------------------");
AccountInfo acctInfo = new AccountInfo(InstanceProvider.AWS, "accesskey", "secretKey");
persistenceManager.persistEntity(acctInfo);
List<AccountInfo> accounts = persistenceManager.getEntities(AccountInfo.class);
assertNotNull(accounts);
assertEquals(1, accounts.size());
assertSame(acctInfo.getType(), accounts.get(0).getType());
AccountInfo acctInfo1 = new AccountInfo(InstanceProvider.AWS, "accesskey1", "secretKey1");
List<InstanceProvider> types = new ArrayList<InstanceProvider>(0);
types.add(InstanceProvider.AWS);
types.add(InstanceProvider.OPENSTACK);
acctInfo1.setTypes(types);
persistenceManager.persistEntity(acctInfo1);
String query = "MATCH (acct:AccountInfo{accessKey:\"accesskey1\"}) RETURN acct";
accounts = persistenceManager.getEntities(query, "acct", AccountInfo.class);
assertNotNull(accounts);
assertEquals(1, accounts.size());
assertSame(acctInfo1.getType(), accounts.get(0).getType());
assertEquals(acctInfo1.getTypes().size(), accounts.get(0).getTypes().size());
assertTrue(accounts.get(0).getTypes().size() == 2);
assertSame(acctInfo1.getTypes().get(0), accounts.get(0).getTypes().get(0));
assertSame(acctInfo1.getTypes().get(1), accounts.get(0).getTypes().get(1));
AccountInfo acctInfo2 = new AccountInfo(InstanceProvider.AWS, "accesskey2", "secretKey2");
InstanceProvider[] arrayTypes = new InstanceProvider[2];
arrayTypes[0] = InstanceProvider.AWS;
arrayTypes[1] = InstanceProvider.OPENSTACK;
acctInfo2.setArrayTypes(arrayTypes);
persistenceManager.persistEntity(acctInfo2);
query = "MATCH (acct:AccountInfo{accessKey:\"accesskey2\"}) RETURN acct";
accounts = persistenceManager.getEntities(query, "acct", AccountInfo.class);
assertNotNull(accounts);
assertEquals(1, accounts.size());
assertNotNull(accounts.get(0).getType());
assertNotNull(accounts.get(0).getArrayTypes());
assertTrue(accounts.get(0).getArrayTypes().length == 2);
assertSame(InstanceProvider.AWS, accounts.get(0).getArrayTypes()[0]);
assertSame(InstanceProvider.OPENSTACK, accounts.get(0).getArrayTypes()[1]);
}
@AfterClass
public static void shutDown() {
persistenceManager.shutDownDB();
}
@After
public void clearDb() {
persistenceManager.clearDB();
}
@Test
public void testGetEntities() {
System.out.println("\nTestPersistenceManager.testGetEntities()");
System.out.println("------------------------------------------");
ScriptFile roleSF = new ScriptFile();
roleSF.setName("roleSFName1");
roleSF.setPath("roleSFPath1");
assertNull(roleSF.getId());
ScriptFile playBookSF1 = new ScriptFile();
playBookSF1.setName("playBookSFName1");
playBookSF1.setPath("playBookSFPath1");
assertNull(playBookSF1.getId());
ScriptFile playBookSF2 = new ScriptFile();
playBookSF2.setName("playBookSFName2");
playBookSF2.setPath("playBookSFPath2");
assertNull(playBookSF2.getId());
AnsibleModuleDefinition anModDefinition = new AnsibleModuleDefinition();
List<ScriptFile> playBookSFList = new ArrayList<ScriptFile>();
playBookSFList.add(playBookSF1);
playBookSFList.add(playBookSF2);
Set<ScriptFile> roleSFList = new HashSet<ScriptFile>();
roleSFList.add(roleSF);
anModDefinition.setPlayBooks(playBookSFList);
anModDefinition.setRoles(roleSFList);
Module module = new Module();
module.setName("ModelueName");
module.setPath("ModeluePath");
module.setVersion("ModelueVersion");
module.setDefinition(anModDefinition);
assertNull(module.getId());
persistenceManager.persistEntity(module);
String playBooksQuery =
"MATCH (definition:AnsibleModuleDefinition)-[:playBooks]-"
+ "(file:ScriptFile{name:\"playBookSFName2\"}) RETURN file";
List<ScriptFile> scriptFiles =
persistenceManager.getEntities(playBooksQuery, "file", ScriptFile.class);
assertNotNull(scriptFiles);
assertEquals(1, scriptFiles.size());
assertEquals(playBookSF2.getName(), scriptFiles.get(0).getName());
assertEquals(playBookSF2.getPath(), scriptFiles.get(0).getPath());
playBooksQuery =
"MATCH (definition:AnsibleModuleDefinition)-[:roles]-" + "(file:ScriptFile) RETURN file";
scriptFiles = persistenceManager.getEntities(playBooksQuery, "file", ScriptFile.class);
assertNotNull(scriptFiles);
assertEquals(1, scriptFiles.size());
assertEquals(roleSF.getName(), scriptFiles.get(0).getName());
assertEquals(roleSF.getPath(), scriptFiles.get(0).getPath());
String moduleQuery =
"MATCH (module:Module)-[:definition]-(definition:PuppetModuleDefinition) RETURN module";
List<Module> modules = persistenceManager.getEntities(moduleQuery, "module", Module.class);
assertNotNull(modules);
assertEquals(0, modules.size());
moduleQuery = "MATCH (module:Module)-[:definition]-(AnsibleModuleDefinition) RETURN module";
modules = persistenceManager.getEntities(moduleQuery, "module", Module.class);
assertNotNull(modules);
assertEquals(1, modules.size());
assertEquals(module.getName(), modules.get(0).getName());
assertEquals(module.getPath(), modules.get(0).getPath());
assertEquals(module.getVersion(), modules.get(0).getVersion());
}
public class TestPersistenceManager {
static PersistenceManager persistenceManager = PersistenceManager.get();
@Test
public void testPersistEntity_simpleObject() {
System.out.println("\nTestPersistenceManager.testPersistEntity_simpleObject()");
System.out.println("---------------------------------------------------------");
ScriptFile roleSF = new ScriptFile();
roleSF.setName("roleSFName1");
roleSF.setPath("roleSFPath1");
assertNull(roleSF.getId());
persistenceManager.persistEntity(roleSF);
assertNotNull(roleSF.getId());
List<ScriptFile> entities = persistenceManager.getEntities(ScriptFile.class);
assertNotNull(entities);
assertEquals(1, entities.size());
assertEquals(roleSF.getName(), entities.get(0).getName());
assertEquals(roleSF.getPath(), entities.get(0).getPath());
}
@Test
public void testPersistEntity_complexObject() {
System.out.println("\nTestPersistenceManager.testPersistEntity_complexObject()");
System.out.println("----------------------------------------------------------");
ScriptFile roleSF = new ScriptFile();
roleSF.setName("roleSFName1");
roleSF.setPath("roleSFPath1");
assertNull(roleSF.getId());
ScriptFile playBookSF1 = new ScriptFile();
playBookSF1.setName("playBookSFName1");
playBookSF1.setPath("playBookSFPath1");
assertNull(playBookSF1.getId());
ScriptFile playBookSF2 = new ScriptFile();
playBookSF2.setName("playBookSFName2");
playBookSF2.setPath("playBookSFPath2");
assertNull(playBookSF2.getId());
AnsibleModuleDefinition anModDefinition = new AnsibleModuleDefinition();
List<ScriptFile> playBookSFList = new ArrayList<ScriptFile>();
playBookSFList.add(playBookSF1);
playBookSFList.add(playBookSF2);
Set<ScriptFile> roleSFList = new HashSet<ScriptFile>();
roleSFList.add(roleSF);
anModDefinition.setPlayBooks(playBookSFList);
anModDefinition.setRoles(roleSFList);
assertNull(anModDefinition.getId());
persistenceManager.persistEntity(anModDefinition);
List<AnsibleModuleDefinition> entities =
persistenceManager.getEntities(AnsibleModuleDefinition.class);
assertNotNull(entities);
assertEquals(1, entities.size());
AnsibleModuleDefinition retrievedDefinition = entities.get(0);
assertNotNull(retrievedDefinition);
assertNotNull(retrievedDefinition.getId());
assertEquals(2, retrievedDefinition.getPlayBooks().size());
assertEquals(1, retrievedDefinition.getRoles().size());
}
@Test
public void testPersistEntity_EntityWithMap() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithMap()");
System.out.println("----------------------------------------------------------");
Task task = new Task();
task.setName("NewTask");
Map<String, String> taskEntriesMap = new HashMap<String, String>();
taskEntriesMap.put("Key1", "Value1");
taskEntriesMap.put("Key1", "Value1");
task.setTaskEntries(taskEntriesMap);
persistenceManager.persistEntity(task);
List<Task> tasks = persistenceManager.getEntities(Task.class);
assertNotNull(tasks);
assertEquals(1, tasks.size());
Task retrievedTask = tasks.get(0);
assertNotNull(retrievedTask.getId());
assertEquals(task.getName(), retrievedTask.getName());
assertEquals(task.getTaskEntries(), retrievedTask.getTaskEntries());
}
@Test
public void testPersistEntity_EntityWithListOfSimpleType() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithListOfSimpleType()");
System.out.println("------------------------------------------------------------------------");
Filter filter = new Filter();
List<String> values = new ArrayList<String>();
values.add("Value1");
values.add("Value2");
filter.setValues(values);
List<Object> mixedValues = new ArrayList<Object>();
mixedValues.add("Hi");
mixedValues.add(0);
filter.setMixedValues(mixedValues);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertEquals(filter.getValues(), retrievedFilter.getValues());
}
@Test
public void testPersistEntity_EntityWithSetOfSimpleType() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithSetOfSimpleType()");
System.out.println("-----------------------------------------------------------------------");
IpPermissionInfo ipPermInfo = new IpPermissionInfo();
List<String> ipRanges = new ArrayList<String>();
ipRanges.add("ipRange1");
ipRanges.add("ipRange2");
ipPermInfo.setIpRanges(ipRanges);
Set<String> groups = new HashSet<String>();
groups.add("groupId1");
groups.add("groupId2");
ipPermInfo.setGroupIds(groups);
persistenceManager.persistEntity(ipPermInfo);
List<IpPermissionInfo> ipPermissionInfos =
persistenceManager.getEntities(IpPermissionInfo.class);
assertNotNull(ipPermissionInfos);
assertEquals(1, ipPermissionInfos.size());
IpPermissionInfo retrievedIpPermissionInfo = ipPermissionInfos.get(0);
assertNotNull(retrievedIpPermissionInfo.getId());
assertEquals(ipPermInfo.getIpRanges().size(), retrievedIpPermissionInfo.getIpRanges().size());
assertEquals(ipPermInfo.getGroupIds().size(), retrievedIpPermissionInfo.getGroupIds().size());
for (String ipRange : ipRanges) {
assertTrue(retrievedIpPermissionInfo.getIpRanges().contains(ipRange));
}
for (String group : groups) {
assertTrue(retrievedIpPermissionInfo.getGroupIds().contains(group));
}
}
@Test
public void testPersistEntity_EntityWithArray() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithArray()");
System.out.println("------------------------------------------------------------");
Filter filter = new Filter();
List<String> values = new ArrayList<String>();
values.add("Value1");
values.add("Value2");
filter.setValues(values);
String[] simpleArray = new String[] {"Kayal", "Vizhi"};
filter.setSimpleArray(simpleArray);
Tuple[] complexArray = new Tuple[] {new Tuple("a", "A"), new Tuple("b", "B")};
filter.setComplexArray(complexArray);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertEquals(filter.getValues(), retrievedFilter.getValues());
assertEquals(complexArray.length, retrievedFilter.getComplexArray().length);
for (int i = 0; i < complexArray.length; i++) {
assertEquals(complexArray[i], retrievedFilter.getComplexArray()[i]);
}
}
@Test
public void testPersistEntity_EntityWithCollectionOfMixedType() {
System.out.println(
"\nTestPersistenceManager.testPersistEntity_EntityWithCollectionOfMixedType()");
System.out.println(
"----------------------------------------------------------------------------");
Filter filter = new Filter();
List<Object> mixedValues = new ArrayList<Object>();
mixedValues.add("Hi");
mixedValues.add(0);
filter.setMixedValues(mixedValues);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertNull(retrievedFilter.getMixedValues());
assertNotSame(filter.getMixedValues(), retrievedFilter.getMixedValues());
}
@Test
public void testPersistEntity_EntityWithArrayOfMixedType() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithArrayOfMixedType()");
System.out.println(
"----------------------------------------------------------------------------");
Filter filter = new Filter();
Object[] objects = new Object[] {"Hi", new Integer(1)};
filter.setMixedArray(objects);
persistenceManager.persistEntity(filter);
List<Filter> filters = persistenceManager.getEntities(Filter.class);
assertNotNull(filters);
assertEquals(1, filters.size());
Filter retrievedFilter = filters.get(0);
assertNotNull(retrievedFilter.getId());
assertNull(retrievedFilter.getMixedArray());
assertNotSame(filter.getMixedArray(), retrievedFilter.getMixedArray());
}
@Test
public void testPersistEntity_EntityWithEnum() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithEnum()");
System.out.println("------------------------------------------------------------");
AccountInfo acctInfo = new AccountInfo(InstanceProvider.AWS, "accesskey", "secretKey");
persistenceManager.persistEntity(acctInfo);
List<AccountInfo> accounts = persistenceManager.getEntities(AccountInfo.class);
assertNotNull(accounts);
assertEquals(1, accounts.size());
assertSame(acctInfo.getType(), accounts.get(0).getType());
AccountInfo acctInfo1 = new AccountInfo(InstanceProvider.AWS, "accesskey1", "secretKey1");
List<InstanceProvider> types = new ArrayList<InstanceProvider>(0);
types.add(InstanceProvider.AWS);
types.add(InstanceProvider.OPENSTACK);
acctInfo1.setTypes(types);
persistenceManager.persistEntity(acctInfo1);
String query = "MATCH (acct:AccountInfo{accessKey:\"accesskey1\"}) RETURN acct";
accounts = persistenceManager.getEntities(query, "acct", AccountInfo.class);
assertNotNull(accounts);
assertEquals(1, accounts.size());
assertSame(acctInfo1.getType(), accounts.get(0).getType());
assertEquals(acctInfo1.getTypes().size(), accounts.get(0).getTypes().size());
assertTrue(accounts.get(0).getTypes().size() == 2);
assertSame(acctInfo1.getTypes().get(0), accounts.get(0).getTypes().get(0));
assertSame(acctInfo1.getTypes().get(1), accounts.get(0).getTypes().get(1));
AccountInfo acctInfo2 = new AccountInfo(InstanceProvider.AWS, "accesskey2", "secretKey2");
InstanceProvider[] arrayTypes = new InstanceProvider[2];
arrayTypes[0] = InstanceProvider.AWS;
arrayTypes[1] = InstanceProvider.OPENSTACK;
acctInfo2.setArrayTypes(arrayTypes);
persistenceManager.persistEntity(acctInfo2);
query = "MATCH (acct:AccountInfo{accessKey:\"accesskey2\"}) RETURN acct";
accounts = persistenceManager.getEntities(query, "acct", AccountInfo.class);
assertNotNull(accounts);
assertEquals(1, accounts.size());
assertNotNull(accounts.get(0).getType());
assertNotNull(accounts.get(0).getArrayTypes());
assertTrue(accounts.get(0).getArrayTypes().length == 2);
assertSame(InstanceProvider.AWS, accounts.get(0).getArrayTypes()[0]);
assertSame(InstanceProvider.OPENSTACK, accounts.get(0).getArrayTypes()[1]);
}
@Test
public void testPersistEntity_EntityWithDate() {
System.out.println("\nTestPersistenceManager.testPersistEntity_EntityWithDate()");
System.out.println("------------------------------------------------------------");
DateHolder dateHolder = new DateHolder();
dateHolder.setUtilDate(new Date());
dateHolder.setSqlDate(new java.sql.Date(System.currentTimeMillis()));
List<Date> utilDatesList = new ArrayList<Date>(2);
utilDatesList.add(new Date());
utilDatesList.add(new Date());
Date[] utilDateArray = new Date[] {new Date(), new Date()};
List<java.sql.Date> sqlDatesList = new ArrayList<java.sql.Date>(2);
sqlDatesList.add(new java.sql.Date(System.currentTimeMillis()));
sqlDatesList.add(new java.sql.Date(System.currentTimeMillis()));
java.sql.Date[] sqlDatesArray =
new java.sql.Date[] {
new java.sql.Date(System.currentTimeMillis()),
new java.sql.Date(System.currentTimeMillis())
};
dateHolder.setSqlDatesList(sqlDatesList);
dateHolder.setUtilDatesList(utilDatesList);
dateHolder.setSqlDatesArray(sqlDatesArray);
dateHolder.setUtilDatesArray(utilDateArray);
persistenceManager.persistEntity(dateHolder);
List<DateHolder> dateHolders = persistenceManager.getEntities(DateHolder.class);
assertNotNull(dateHolders);
assertEquals(1, dateHolders.size());
assertEquals(dateHolder.getUtilDate(), dateHolders.get(0).getUtilDate());
assertEquals(dateHolder.getSqlDate(), dateHolders.get(0).getSqlDate());
}
@Test
public void testGetEntities() {
System.out.println("\nTestPersistenceManager.testGetEntities()");
System.out.println("------------------------------------------");
ScriptFile roleSF = new ScriptFile();
roleSF.setName("roleSFName1");
roleSF.setPath("roleSFPath1");
assertNull(roleSF.getId());
ScriptFile playBookSF1 = new ScriptFile();
playBookSF1.setName("playBookSFName1");
playBookSF1.setPath("playBookSFPath1");
assertNull(playBookSF1.getId());
ScriptFile playBookSF2 = new ScriptFile();
playBookSF2.setName("playBookSFName2");
playBookSF2.setPath("playBookSFPath2");
assertNull(playBookSF2.getId());
AnsibleModuleDefinition anModDefinition = new AnsibleModuleDefinition();
List<ScriptFile> playBookSFList = new ArrayList<ScriptFile>();
playBookSFList.add(playBookSF1);
playBookSFList.add(playBookSF2);
Set<ScriptFile> roleSFList = new HashSet<ScriptFile>();
roleSFList.add(roleSF);
anModDefinition.setPlayBooks(playBookSFList);
anModDefinition.setRoles(roleSFList);
Module module = new Module();
module.setName("ModelueName");
module.setPath("ModeluePath");
module.setVersion("ModelueVersion");
module.setDefinition(anModDefinition);
assertNull(module.getId());
persistenceManager.persistEntity(module);
String playBooksQuery =
"MATCH (definition:AnsibleModuleDefinition)-[:playBooks]-"
+ "(file:ScriptFile{name:\"playBookSFName2\"}) RETURN file";
List<ScriptFile> scriptFiles =
persistenceManager.getEntities(playBooksQuery, "file", ScriptFile.class);
assertNotNull(scriptFiles);
assertEquals(1, scriptFiles.size());
assertEquals(playBookSF2.getName(), scriptFiles.get(0).getName());
assertEquals(playBookSF2.getPath(), scriptFiles.get(0).getPath());
playBooksQuery =
"MATCH (definition:AnsibleModuleDefinition)-[:roles]-" + "(file:ScriptFile) RETURN file";
scriptFiles = persistenceManager.getEntities(playBooksQuery, "file", ScriptFile.class);
assertNotNull(scriptFiles);
assertEquals(1, scriptFiles.size());
assertEquals(roleSF.getName(), scriptFiles.get(0).getName());
assertEquals(roleSF.getPath(), scriptFiles.get(0).getPath());
String moduleQuery =
"MATCH (module:Module)-[:definition]-(definition:PuppetModuleDefinition) RETURN module";
List<Module> modules = persistenceManager.getEntities(moduleQuery, "module", Module.class);
assertNotNull(modules);
assertEquals(0, modules.size());
moduleQuery = "MATCH (module:Module)-[:definition]-(AnsibleModuleDefinition) RETURN module";
modules = persistenceManager.getEntities(moduleQuery, "module", Module.class);
assertNotNull(modules);
assertEquals(1, modules.size());
assertEquals(module.getName(), modules.get(0).getName());
assertEquals(module.getPath(), modules.get(0).getPath());
assertEquals(module.getVersion(), modules.get(0).getVersion());
}
@After
public void clearDb() {
persistenceManager.clearDB();
}
@AfterClass
public static void shutDown() {
persistenceManager.shutDownDB();
}
}
