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  import java.util.UUID;
27  
28  import javax.management.MalformedObjectNameException;
29  
30  import org.junit.After;
31  import org.junit.Before;
32  import org.junit.Test;
33  import org.neo4j.graphdb.Node;
34  import org.neo4j.graphdb.RelationshipType;
35  import org.neo4j.graphdb.Transaction;
36  import org.neo4j.kernel.ImpermanentGraphDatabase;
37  
38  public class PropertyCountSampleableTest
39  {
40      public ImpermanentGraphDatabase db;
41      public PropertyCountSampleable sampleable;
42  
43      @Test
44      public void emptyDbHasZeroNodesInUse() throws IOException, MalformedObjectNameException
45      {
46          assertThat( sampleable.getValue(), is( 0L ) );
47      }
48  
49      @Test
50      public void addANodeAndSampleableGoesUp() throws IOException, MalformedObjectNameException
51      {
52          addPropertyToReferenceNode( );
53  
54          assertThat( sampleable.getValue(), is( 1L ) );
55  
56          addNodeIntoGraph( );
57          addNodeIntoGraph( );
58  
59          assertThat( sampleable.getValue(), is( 3L ) );
60      }
61  
62      private void addNodeIntoGraph() {
63          Transaction tx = db.beginTx();
64          Node referenceNode = db.getReferenceNode();
65          Node myNewNode = db.createNode();
66          myNewNode.setProperty("id", UUID.randomUUID().toString());
67          myNewNode.createRelationshipTo(referenceNode, new RelationshipType() {
68              public String name() {
69                  return "knows_about";
70              }
71          });
72  
73          tx.success();
74          tx.finish();
75      }
76  
77      private void addPropertyToReferenceNode( )
78      {
79          Transaction tx = db.beginTx();
80          Node n = db.getReferenceNode();
81          n.setProperty( "monkey", "rock!" );
82          tx.success();
83          tx.finish();
84      }
85  
86      @Before
87      public void setUp() throws Exception
88      {
89          db = new ImpermanentGraphDatabase();
90          sampleable = new PropertyCountSampleable( db );
91      }
92  
93      @After
94      public void shutdownDatabase()
95      {
96          this.db.shutdown();
97      }
98  }