@@ -0,0 +1,2504 @@
+/*%0A * Licensed to the University of California, Berkeley under one or more contributor license%0A * agreements. See the NOTICE file distributed with this work for additional information regarding%0A * copyright ownership. The ASF licenses this file to You under the Apache License, Version 2.0 (the%0A * %22License%22); you may not use this file except in compliance with the License. You may obtain a%0A * copy of the License at%0A *%0A * http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software distributed under the License%0A * is distributed on an %22AS IS%22 BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express%0A * or implied. See the License for the specific language governing permissions and limitations under%0A * the License.%0A */%0A%0Apackage tachyon.worker.block.allocator;%0A%0Aimport java.io.File;%0Aimport java.io.IOException;%0Aimport java.util.Collections;%0A%0Aimport org.junit.Assert;%0Aimport org.junit.Before;%0Aimport org.junit.Rule;%0Aimport org.junit.Test;%0Aimport org.junit.rules.TemporaryFolder;%0A%0Aimport tachyon.worker.block.BlockMetadataManager;%0Aimport tachyon.worker.block.BlockMetadataManagerView;%0Aimport tachyon.worker.block.evictor.EvictorTestUtils;%0A%0A/*%0A * Test AllocatorFactory by passing different AllocatorType %0A * and test if it generate the correct Allocator instance.%0A */%0Apublic class AllocatorFactoryTest %7B%0A  private BlockMetadataManager mMetaManager;%0A  private BlockMetadataManagerView mManagerView;%0A  %0A  @Rule%0A  public TemporaryFolder mTestFolder = new TemporaryFolder();%0A%0A  @Before%0A  public void before() throws IOException %7B%0A    File tempFolder = mTestFolder.newFolder();%0A    mMetaManager = EvictorTestUtils.defaultMetadataManager(tempFolder.getAbsolutePath());%0A    mManagerView = new BlockMetadataManagerView(mMetaManager, Collections.%3CInteger%3EemptySet(),%0A        Collections.%3CLong%3EemptySet());%0A  %7D%0A  %0A  @Test%0A  public void createGreedyAllocatorTest() %7B%0A    Allocator allocator = AllocatorFactory.create(AllocatorType.GREEDY, mManagerView);%0A    Assert.assertTrue(allocator instanceof GreedyAllocator);%0A  %7D%0A  %0A  @Test%0A  public void createMaxFreeAllocatorTest() %7B%0A    Allocator allocator = AllocatorFactory.create(AllocatorType.MAX_FREE, mManagerView);%0A    Assert.assertTrue(allocator instanceof MaxFreeAllocator);%0A  %7D%0A  %0A  @Test%0A  public void createDefaultAllocatorTypeTest() %7B%0A    Allocator allocator = AllocatorFactory.create(AllocatorType.DEFAULT, mManagerView);%0A    Assert.assertTrue(allocator instanceof MaxFreeAllocator);%0A  %7D%0A%7D%0A

Add unit test of AllocatorFactory