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  }