View Javadoc

    1   /**
    2    * Copyright (c) 2002-2011 "Neo Technology,"
    3    * Network Engine for Objects in Lund AB [http://neotechnology.com]
    4    *
    5    * This file is part of Neo4j.
    6    *
    7    * Neo4j is free software: you can redistribute it and/or modify
    8    * it under the terms of the GNU General Public License as published by
    9    * the Free Software Foundation, either version 3 of the License, or
    10   * (at your option) any later version.
    11   *
    12   * This program is distributed in the hope that it will be useful,
    13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
    14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    15   * GNU General Public License for more details.
    16   *
    17   * You should have received a copy of the GNU General Public License
    18   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    19   */
    20  package org.neo4j.server.rrd;
    21
    22  import static org.hamcrest.MatcherAssert.assertThat;
    23  import static org.hamcrest.core.Is.is;
    24
    25  import java.io.IOException;
    26
    27  import javax.management.MalformedObjectNameException;
    28
    29  import org.junit.After;
    30  import org.junit.Before;
    31  import org.junit.Test;
    32  import org.neo4j.graphdb.DynamicRelationshipType;
    33  import org.neo4j.graphdb.Node;
    34  import org.neo4j.graphdb.Transaction;
    35  import org.neo4j.kernel.ImpermanentGraphDatabase;
    36
    37  public class RelationshipCountSampleableTest
    38  {
    39      public ImpermanentGraphDatabase db;
    40      public RelationshipCountSampleable sampleable;
    41
    42      @Test
    43      public void emptyDbHasZeroRelationships() throws IOException, MalformedObjectNameException
    44      {
    45          assertThat( sampleable.getValue(), is( 0L ) );
    46      }
    47
    48      @Test
    49      public void addANodeAndSampleableGoesUp() throws IOException, MalformedObjectNameException
    50      {
    51          createARelationship( db );
    52
    53          assertThat( sampleable.getValue(), is( 1L ) );
    54      }
    55
    56      private void createARelationship( ImpermanentGraphDatabase db )
    57      {
    58          Transaction tx = db.beginTx();
    59          Node node1 = db.createNode();
    60          Node node2 = db.createNode();
    61          node1.createRelationshipTo( node2, DynamicRelationshipType.withName( "friend" ) );
    62          tx.success();
    63          tx.finish();
    64      }
    65
    66      @Before
    67      public void setUp() throws Exception
    68      {
    69          db = new ImpermanentGraphDatabase();
    70          sampleable = new RelationshipCountSampleable( db );
    71      }
    72
    73      @After
    74      public void shutdownDatabase()
    75      {
    76          this.db.shutdown();
    77      }
    78  }