Esempio n. 1
0
  @Test
  public void shouldAcceptFolderWithCypFilesAsFixtures() throws Exception {
    // Given two files in the root folder
    File targetFolder = testDir.directory();
    FileUtils.writeToFile(
        new File(targetFolder, "fixture1.cyp"),
        "CREATE (u:User)\n" + "CREATE (a:OtherUser)",
        false);
    FileUtils.writeToFile(
        new File(targetFolder, "fixture2.cyp"),
        "CREATE (u:User)\n" + "CREATE (a:OtherUser)",
        false);

    // And given one file in a sub directory
    File subDir = new File(targetFolder, "subdirectory");
    subDir.mkdir();
    FileUtils.writeToFile(
        new File(subDir, "subDirFixture.cyp"), "CREATE (u:User)\n" + "CREATE (a:OtherUser)", false);

    // When
    try (ServerControls server =
        newInProcessBuilder(targetFolder).withFixture(targetFolder).newServer()) {
      // Then
      HTTP.Response response =
          HTTP.POST(
              server.httpURI().toString() + "db/data/transaction/commit",
              quotedJson("{'statements':[{'statement':'MATCH (n:User) RETURN n'}]}"));

      assertThat(response.get("results").get(0).get("data").size(), equalTo(3));
    }
  }
Esempio n. 2
0
  @Test
  public void shouldBeAbleToRestartServer() throws Exception {
    // Given
    String dbDirectory1 = baseDir.directory("db1").getAbsolutePath();
    String dbDirectory2 = baseDir.directory("db2").getAbsolutePath();

    ConfigurationBuilder config =
        new PropertyFileConfigurator(
            AdvancedServerBuilder.server().usingDatabaseDir(dbDirectory1).createPropertiesFiles(),
            ConsoleLogger.DEV_NULL);

    // When
    NeoServer server = cleanup.add(new AdvancedNeoServer(config, graphDbDependencies()));
    server.start();

    assertNotNull(server.getDatabase().getGraph());
    assertEquals(dbDirectory1, server.getDatabase().getLocation());

    // Change the database location
    setProperty(config.configuration(), Configurator.DATABASE_LOCATION_PROPERTY_KEY, dbDirectory2);
    ServerManagement bean = new ServerManagement(server);
    bean.restartServer();

    // Then
    assertNotNull(server.getDatabase().getGraph());
    assertEquals(dbDirectory2, server.getDatabase().getLocation());
  }
 private ClusterManager.ManagedCluster startClusterSeededWith(File seedDir) {
   ClusterManager.ManagedCluster cluster =
       life.add(new ClusterManager.Builder(targetDir.directory()).withSeedDir(seedDir).build())
           .getDefaultCluster();
   cluster.await(allSeesAllAsAvailable());
   return cluster;
 }
Esempio n. 4
0
  @Test
  public void shouldTakeDirectoryLocationFromConfig() throws Exception {
    String expected = testDirectory.directory().getAbsolutePath();
    addProperty(Configurator.RRDB_LOCATION_PROPERTY_KEY, expected);
    TestableRrdFactory factory = createRrdFactory();

    factory.createRrdDbAndSampler(db, new NullJobScheduler());

    assertThat(factory.directoryUsed, is(expected));
  }
Esempio n. 5
0
  @Test
  public void shouldCreateRrdFileInTempLocationForImpermanentDatabases() throws IOException {
    // Given
    String expected = testDirectory.directory().getAbsolutePath();
    TestableRrdFactory factory = createRrdFactory(expected);

    // When
    factory.createRrdDbAndSampler(db, new NullJobScheduler());

    // Then
    assertThat(factory.directoryUsed, is(expected));
  }
Esempio n. 6
0
  private Configurator buildProperties() throws IOException {
    new File(test.directory().getAbsolutePath() + DIRSEP + "conf").mkdirs();

    Properties databaseProperties = new Properties();
    String databasePropertiesFileName =
        test.directory().getAbsolutePath() + DIRSEP + "conf" + DIRSEP + "neo4j.properties";
    databaseProperties.store(new FileWriter(databasePropertiesFileName), null);

    Properties serverProperties = new Properties();
    String serverPropertiesFilename =
        test.directory().getAbsolutePath() + DIRSEP + "conf" + DIRSEP + "neo4j-server.properties";
    serverProperties.setProperty(
        Configurator.DATABASE_LOCATION_PROPERTY_KEY,
        test.directory().getAbsolutePath() + DIRSEP + "data" + DIRSEP + "graph.db");

    serverProperties.setProperty(
        Configurator.DB_TUNING_PROPERTY_FILE_KEY, databasePropertiesFileName);
    serverProperties.setProperty(Configurator.NEO_SERVER_CONFIG_FILE_KEY, serverPropertiesFilename);
    serverProperties.store(new FileWriter(serverPropertiesFilename), null);

    return new PropertyFileConfigurator(new File(serverPropertiesFilename));
  }
  @Test
  public void shouldCloseUnusedWindow() throws Exception {
    // given
    String filename = new File(directory.directory(), "mapped.file").getAbsolutePath();
    RandomAccessFile file = resources.add(new RandomAccessFile(filename, "rw"));
    MappedPersistenceWindow window =
        new MappedPersistenceWindow(0, 8, 16, file.getChannel(), READ_WRITE);

    // when
    boolean wasClosed = window.writeOutAndCloseIfFree(false);

    // then
    assertTrue(wasClosed);
  }
Esempio n. 8
0
  @Test
  public void shouldCreateRrdFileInDbSubdirectory() throws Exception {
    String storeDir = testDirectory.directory().getAbsolutePath();
    db =
        new WrappedDatabase(
            (InternalAbstractGraphDatabase)
                new TestGraphDatabaseFactory().newEmbeddedDatabase(storeDir));
    TestableRrdFactory factory = createRrdFactory();

    // When
    factory.createRrdDbAndSampler(db, new NullJobScheduler());

    // Then
    String rrdParent = new File(factory.directoryUsed).getParent();

    assertThat(rrdParent, is(storeDir));
  }
Esempio n. 9
0
  @Test
  public void shouldMoveAwayInvalidRrdFile() throws IOException {
    // Given
    String expected = new File(testDirectory.directory(), "rrd-test").getAbsolutePath();
    addProperty(Configurator.RRDB_LOCATION_PROPERTY_KEY, expected);

    TestableRrdFactory factory = createRrdFactory();
    createInvalidRrdFile(expected);

    // When
    RrdDbWrapper rrdDbAndSampler = factory.createRrdDbAndSampler(db, new NullJobScheduler());

    // Then
    assertSubdirectoryExists("rrd-test-invalid", factory.directoryUsed);

    rrdDbAndSampler.close();
  }
Esempio n. 10
0
  @Test
  public void shouldHandleStringFixtures() throws Exception {
    // Given two files in the root folder
    File targetFolder = testDir.directory();

    // When
    try (ServerControls server =
        newInProcessBuilder(targetFolder).withFixture("CREATE (a:User)").newServer()) {
      // Then
      HTTP.Response response =
          HTTP.POST(
              server.httpURI().toString() + "db/data/transaction/commit",
              quotedJson("{'statements':[{'statement':'MATCH (n:User) RETURN n'}]}"));

      assertThat(response.get("results").get(0).get("data").size(), equalTo(1));
    }
  }
  private File databaseWithUniquenessConstraint(String label, String propertyKey) {
    File storeDir = new File(targetDir.directory(), "seed");
    GraphDatabaseService graphDb =
        new GraphDatabaseFactory().newEmbeddedDatabase(storeDir.getAbsolutePath());
    try {
      Transaction tx = graphDb.beginTx();
      try {
        graphDb.schema().constraintFor(label(label)).assertPropertyIsUnique(propertyKey).create();

        tx.success();
      } finally {
        tx.finish();
      }
    } finally {
      graphDb.shutdown();
    }
    return storeDir;
  }
Esempio n. 12
0
  @Test
  public void shouldHandleFixturesWithSyntaxErrorsGracefully() throws Exception {
    // Given two files in the root folder
    File targetFolder = testDir.directory();
    FileUtils.writeToFile(
        new File(targetFolder, "fixture1.cyp"), "this is not a valid cypher statement", false);

    // When
    try (ServerControls ignore =
        newInProcessBuilder(targetFolder).withFixture(targetFolder).newServer()) {
      fail("Should have thrown exception");
    } catch (RuntimeException e) {
      assertThat(
          e.getMessage(),
          equalTo(
              "Failed to install fixtures: Invalid input 't': expected <init> "
                  + "(line 1, column 1)\n\"this is not a valid cypher statement\"\n ^"));
    }
  }
Esempio n. 13
0
  @Test
  public void shouldAccepSingleCypherFileAsFixture() throws Exception {
    // Given
    File targetFolder = testDir.directory();
    File fixture = new File(targetFolder, "fixture.cyp");
    FileUtils.writeToFile(fixture, "CREATE (u:User)" + "CREATE (a:OtherUser)", false);

    // When
    try (ServerControls server =
        newInProcessBuilder(targetFolder).withFixture(fixture).newServer()) {
      // Then
      HTTP.Response response =
          HTTP.POST(
              server.httpURI().toString() + "db/data/transaction/commit",
              quotedJson("{'statements':[{'statement':'MATCH (n:User) RETURN n'}]}"));

      assertThat(response.status(), equalTo(200));
      assertThat(response.get("results").get(0).get("data").size(), equalTo(1));
    }
  }
Esempio n. 14
0
 private TestableRrdFactory createRrdFactory() {
   return new TestableRrdFactory(
       config, new File(testDirectory.directory(), "rrd").getAbsolutePath());
 }
Esempio n. 15
0
  @Test
  public void testPullStorm() throws Throwable {
    // given

    ClusterManager clusterManager =
        new ClusterManager(
            ClusterManager.clusterWithAdditionalArbiters(2, 1),
            testDirectory.directory(),
            stringMap(
                HaSettings.pull_interval.name(), "0",
                HaSettings.tx_push_factor.name(), "1"));

    clusterManager.start();

    try {
      ClusterManager.ManagedCluster cluster = clusterManager.getDefaultCluster();
      cluster.await(ClusterManager.masterAvailable());
      cluster.await(ClusterManager.masterSeesSlavesAsAvailable(1));

      // Create data
      final HighlyAvailableGraphDatabase master = cluster.getMaster();
      {
        Transaction tx = master.beginTx();
        for (int i = 0; i < 1000; i++) {
          master.createNode().setProperty("foo", "bar");
        }
        tx.success();
        tx.finish();
      }

      // Slave goes down
      HighlyAvailableGraphDatabase slave = cluster.getAnySlave();
      ClusterManager.RepairKit repairKit = cluster.fail(slave);

      // Create more data
      for (int i = 0; i < 1000; i++) {
        {
          Transaction tx = master.beginTx();
          for (int j = 0; j < 1000; j++) {
            master.createNode().setProperty("foo", "bar");
            master.createNode().setProperty("foo", "bar");
          }
          tx.success();
          tx.finish();
        }
      }

      // Slave comes back online
      repairKit.repair();

      cluster.await(ClusterManager.masterSeesSlavesAsAvailable(1));

      // when

      // Create 20 concurrent transactions
      System.out.println("Pull storm");
      ExecutorService executor = Executors.newFixedThreadPool(20);
      for (int i = 0; i < 20; i++) {
        executor.submit(
            new Runnable() {
              @Override
              public void run() {
                Transaction tx = master.beginTx();
                master.createNode().setProperty("foo", "bar");
                tx.success();
                tx.finish(); // This should cause lots of concurrent calls to pullUpdate()
              }
            });
      }

      executor.shutdown();
      executor.awaitTermination(1, TimeUnit.MINUTES);

      System.out.println("Pull storm done");

      // then

      long masterLastCommittedTxId = lastCommittedTxId(master);
      for (HighlyAvailableGraphDatabase member : cluster.getAllMembers()) {
        assertEquals(masterLastCommittedTxId, lastCommittedTxId(member));
      }
    } finally {
      System.err.println("Shutting down");
      clusterManager.shutdown();
      System.err.println("Shut down");
    }
  }