コード例 #1
0
  @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());
  }
コード例 #2
0
  @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());
  }