| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.cassandra.db; |
| |
| import java.io.File; |
| import java.net.UnknownHostException; |
| import java.util.List; |
| |
| import com.google.common.collect.Lists; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| import org.apache.cassandra.cql3.CQLTester; |
| import org.apache.cassandra.dht.BootStrapper; |
| import org.apache.cassandra.locator.InetAddressAndPort; |
| import org.apache.cassandra.locator.TokenMetadata; |
| import org.apache.cassandra.service.StorageService; |
| import org.apache.cassandra.utils.FBUtilities; |
| |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertNotSame; |
| import static org.junit.Assert.assertSame; |
| import static org.junit.Assert.assertTrue; |
| import static org.junit.Assert.fail; |
| |
| public class DiskBoundaryManagerTest extends CQLTester |
| { |
| private DiskBoundaryManager dbm; |
| private MockCFS mock; |
| private Directories dirs; |
| |
| @Before |
| public void setup() |
| { |
| DisallowedDirectories.clearUnwritableUnsafe(); |
| TokenMetadata metadata = StorageService.instance.getTokenMetadata(); |
| metadata.updateNormalTokens(BootStrapper.getRandomTokens(metadata, 10), FBUtilities.getBroadcastAddressAndPort()); |
| createTable("create table %s (id int primary key, x text)"); |
| dirs = new Directories(getCurrentColumnFamilyStore().metadata(), Lists.newArrayList(new Directories.DataDirectory(new File("/tmp/1")), |
| new Directories.DataDirectory(new File("/tmp/2")), |
| new Directories.DataDirectory(new File("/tmp/3")))); |
| mock = new MockCFS(getCurrentColumnFamilyStore(), dirs); |
| dbm = mock.diskBoundaryManager; |
| } |
| |
| @Test |
| public void getBoundariesTest() |
| { |
| DiskBoundaries dbv = dbm.getDiskBoundaries(mock); |
| Assert.assertEquals(3, dbv.positions.size()); |
| assertEquals(dbv.directories, dirs.getWriteableLocations()); |
| } |
| |
| @Test |
| public void disallowedDirectoriesTest() |
| { |
| DiskBoundaries dbv = dbm.getDiskBoundaries(mock); |
| Assert.assertEquals(3, dbv.positions.size()); |
| assertEquals(dbv.directories, dirs.getWriteableLocations()); |
| DisallowedDirectories.maybeMarkUnwritable(new File("/tmp/3")); |
| dbv = dbm.getDiskBoundaries(mock); |
| Assert.assertEquals(2, dbv.positions.size()); |
| Assert.assertEquals(Lists.newArrayList(new Directories.DataDirectory(new File("/tmp/1")), |
| new Directories.DataDirectory(new File("/tmp/2"))), |
| dbv.directories); |
| } |
| |
| @Test |
| public void updateTokensTest() throws UnknownHostException |
| { |
| DiskBoundaries dbv1 = dbm.getDiskBoundaries(mock); |
| StorageService.instance.getTokenMetadata().updateNormalTokens(BootStrapper.getRandomTokens(StorageService.instance.getTokenMetadata(), 10), InetAddressAndPort.getByName("127.0.0.10")); |
| DiskBoundaries dbv2 = dbm.getDiskBoundaries(mock); |
| assertFalse(dbv1.equals(dbv2)); |
| } |
| |
| @Test |
| public void alterKeyspaceTest() throws Throwable |
| { |
| //do not use mock to since it will not be invalidated after alter keyspace |
| DiskBoundaryManager dbm = getCurrentColumnFamilyStore().diskBoundaryManager; |
| DiskBoundaries dbv1 = dbm.getDiskBoundaries(mock); |
| execute("alter keyspace "+keyspace()+" with replication = { 'class' : 'SimpleStrategy', 'replication_factor' : 3 }"); |
| DiskBoundaries dbv2 = dbm.getDiskBoundaries(mock); |
| assertNotSame(dbv1, dbv2); |
| DiskBoundaries dbv3 = dbm.getDiskBoundaries(mock); |
| assertSame(dbv2, dbv3); |
| |
| } |
| |
| private static void assertEquals(List<Directories.DataDirectory> dir1, Directories.DataDirectory[] dir2) |
| { |
| if (dir1.size() != dir2.length) |
| fail(); |
| for (int i = 0; i < dir2.length; i++) |
| { |
| if (!dir1.get(i).equals(dir2[i])) |
| fail(); |
| } |
| } |
| |
| // just to be able to override the data directories |
| private static class MockCFS extends ColumnFamilyStore |
| { |
| MockCFS(ColumnFamilyStore cfs, Directories dirs) |
| { |
| super(cfs.keyspace, cfs.getTableName(), 0, cfs.metadata, dirs, false, false, true); |
| } |
| } |
| } |