-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
BackupEdgeIT.java
121 lines (105 loc) · 4.33 KB
/
BackupEdgeIT.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
* Copyright (c) 2002-2016 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.coreedge.backup;
import java.io.File;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.neo4j.backup.OnlineBackupSettings;
import org.neo4j.coreedge.discovery.Cluster;
import org.neo4j.coreedge.edge.EdgeGraphDatabase;
import org.neo4j.kernel.configuration.Settings;
import org.neo4j.test.DbRepresentation;
import org.neo4j.test.coreedge.ClusterRule;
import org.neo4j.test.rule.SuppressOutput;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotEquals;
import static org.neo4j.backup.BackupEmbeddedIT.runBackupToolFromOtherJvmToGetExitCode;
import static org.neo4j.coreedge.backup.BackupCoreIT.backupAddress;
import static org.neo4j.coreedge.backup.BackupCoreIT.backupArguments;
import static org.neo4j.coreedge.backup.BackupCoreIT.createSomeData;
import static org.neo4j.coreedge.backup.BackupCoreIT.getConfig;
import static org.neo4j.test.rule.SuppressOutput.suppress;
public class BackupEdgeIT
{
@Rule
public SuppressOutput suppressOutput = suppress( SuppressOutput.System.out, SuppressOutput.System.err );
@Rule
public ClusterRule clusterRule = new ClusterRule( BackupEdgeIT.class )
.withNumberOfCoreMembers( 3 )
.withSharedCoreParam( OnlineBackupSettings.online_backup_enabled, Settings.FALSE )
.withNumberOfEdgeMembers( 1 )
.withSharedEdgeParam( OnlineBackupSettings.online_backup_enabled, Settings.TRUE )
.withInstanceEdgeParam( OnlineBackupSettings.online_backup_server, serverId -> ":" + findFreePort( 8000, 9000 ) );
private Cluster cluster;
private File backupPath;
@Before
public void setup() throws Exception
{
backupPath = clusterRule.testDirectory().cleanDirectory( "backup-db" );
cluster = clusterRule.startCluster();
}
@Test
public void makeSureBackupCanBePerformed() throws Throwable
{
// Run backup
createSomeData( cluster );
EdgeGraphDatabase db = cluster.findAnEdgeMember().database();
DbRepresentation beforeChange = DbRepresentation.of( db );
String backupAddress = backupAddress( db );
System.out.println( backupAddress );
String[] args = backupArguments( backupAddress, backupPath.getPath() );
assertEquals( 0, runBackupToolFromOtherJvmToGetExitCode( args ) );
// Add some new data
DbRepresentation afterChange = DbRepresentation.of( createSomeData( cluster ) );
// Verify that backed up database can be started and compare representation
DbRepresentation backupRepresentation = DbRepresentation.of( backupPath, getConfig() );
assertEquals( beforeChange, backupRepresentation );
assertNotEquals( backupRepresentation, afterChange );
}
private static int findFreePort( int startRange, int endRange )
{
InetSocketAddress address = null;
RuntimeException ex = null;
for ( int port = startRange; port <= endRange; port++ )
{
address = new InetSocketAddress( port );
try
{
new ServerSocket( address.getPort(), 100, address.getAddress() ).close();
ex = null;
break;
}
catch ( IOException e )
{
ex = new RuntimeException( e );
}
}
if ( ex != null )
{
throw ex;
}
assert address != null;
return address.getPort();
}
}