You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tajo.apache.org by ji...@apache.org on 2014/12/12 09:22:16 UTC
[02/45] tajo git commit: TAJO-1143: TajoMaster, TajoWorker,
and TajoClient should have diagnosis phase at startup
http://git-wip-us.apache.org/repos/asf/tajo/blob/a0d67bb6/tajo-core/src/test/java/org/apache/tajo/master/rule/TestMasterRules.java
----------------------------------------------------------------------
diff --git a/tajo-core/src/test/java/org/apache/tajo/master/rule/TestMasterRules.java b/tajo-core/src/test/java/org/apache/tajo/master/rule/TestMasterRules.java
new file mode 100644
index 0000000..dc34620
--- /dev/null
+++ b/tajo-core/src/test/java/org/apache/tajo/master/rule/TestMasterRules.java
@@ -0,0 +1,156 @@
+/**
+ * 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.tajo.master.rule;
+
+import static org.junit.Assert.*;
+import static org.hamcrest.CoreMatchers.*;
+
+import org.apache.hadoop.fs.FileSystem;
+import org.apache.hadoop.fs.Path;
+import org.apache.hadoop.fs.permission.FsPermission;
+import org.apache.hadoop.yarn.conf.YarnConfiguration;
+import org.apache.tajo.TajoConstants;
+import org.apache.tajo.conf.TajoConf;
+import org.apache.tajo.conf.TajoConf.ConfVars;
+import org.apache.tajo.rule.EvaluationContext;
+import org.apache.tajo.rule.EvaluationFailedException;
+import org.apache.tajo.rule.EvaluationResult;
+import org.apache.tajo.rule.EvaluationResult.EvaluationResultCode;
+import org.apache.tajo.rule.SelfDiagnosisRuleEngine;
+import org.apache.tajo.rule.SelfDiagnosisRuleSession;
+import org.apache.tajo.rule.base.TajoConfValidationRule;
+import org.apache.tajo.util.CommonTestingUtil;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+public class TestMasterRules {
+
+ private static Path rootFilePath;
+
+ @BeforeClass
+ public static void setUpClass() throws Exception {
+ rootFilePath = CommonTestingUtil.getTestDir();
+ }
+
+ @AfterClass
+ public static void tearDownClass() throws Exception {
+ CommonTestingUtil.cleanupTestDir(rootFilePath.toUri().getPath());
+ }
+
+ @Test
+ public void testTajoConfValidationRule() throws Exception {
+ TajoConf tajoConf = new TajoConf(new YarnConfiguration());
+
+ EvaluationContext context = new EvaluationContext();
+ context.addParameter(TajoConf.class.getName(), tajoConf);
+
+ TajoConfValidationRule validationRule = new TajoConfValidationRule();
+ EvaluationResult result = validationRule.evaluate(context);
+
+ assertThat(result, is(notNullValue()));
+ assertThat(result.getReturnCode(), is(EvaluationResultCode.OK));
+ }
+
+ @Test(expected=EvaluationFailedException.class)
+ public void testTajoConfValidationRuleWithException() throws Exception {
+ TajoConf tajoConf = new TajoConf(new YarnConfiguration());
+ SelfDiagnosisRuleEngine ruleEngine = SelfDiagnosisRuleEngine.getInstance();
+ SelfDiagnosisRuleSession ruleSession = ruleEngine.newRuleSession();
+
+ tajoConf.setVar(TajoConf.ConfVars.ROOT_DIR, "invalid path.");
+
+ EvaluationContext context = new EvaluationContext();
+ context.addParameter(TajoConf.class.getName(), tajoConf);
+
+ ruleSession.withRuleNames("TajoConfValidationRule").fireRules(context);
+
+ fail("EvaluationFailedException exception is expected, but it does not happen.");
+ }
+
+ protected void createTajoDirectories(TajoConf tajoConf) throws Exception {
+ Path tajoRootDir = new Path(rootFilePath, "tajo-root");
+ FileSystem rootFs = tajoRootDir.getFileSystem(tajoConf);
+ FsPermission defaultPermission = FsPermission.createImmutable((short)0700);
+
+ if (!rootFs.exists(tajoRootDir)) {
+ rootFs.mkdirs(tajoRootDir, new FsPermission(defaultPermission));
+ }
+
+ tajoConf.setVar(ConfVars.ROOT_DIR, tajoRootDir.toUri().toString());
+
+ Path tajoSystemDir = new Path(tajoRootDir, TajoConstants.SYSTEM_DIR_NAME);
+ if (!rootFs.exists(tajoSystemDir)) {
+ rootFs.mkdirs(tajoSystemDir, new FsPermission(defaultPermission));
+ }
+
+ Path tajoSystemResourceDir = new Path(tajoSystemDir, TajoConstants.SYSTEM_RESOURCE_DIR_NAME);
+ if (!rootFs.exists(tajoSystemResourceDir)) {
+ rootFs.mkdirs(tajoSystemResourceDir, new FsPermission(defaultPermission));
+ }
+
+ Path tajoWarehouseDir = new Path(tajoRootDir, TajoConstants.WAREHOUSE_DIR_NAME);
+ if (!rootFs.exists(tajoWarehouseDir)) {
+ rootFs.mkdirs(tajoWarehouseDir, new FsPermission(defaultPermission));
+ }
+
+ Path tajoStagingDir = new Path(tajoRootDir, "staging");
+ if (!rootFs.exists(tajoStagingDir)) {
+ rootFs.mkdirs(tajoStagingDir, new FsPermission(defaultPermission));
+ }
+ tajoConf.setVar(ConfVars.STAGING_ROOT_DIR, tajoStagingDir.toUri().toString());
+ }
+
+ @Test
+ public void testFileSystemRule() throws Exception {
+ TajoConf tajoConf = new TajoConf(new YarnConfiguration());
+
+ createTajoDirectories(tajoConf);
+
+ EvaluationContext context = new EvaluationContext();
+ context.addParameter(TajoConf.class.getName(), tajoConf);
+
+ FileSystemRule fsRule = new FileSystemRule();
+ EvaluationResult result = fsRule.evaluate(context);
+
+ assertThat(result, is(notNullValue()));
+ assertThat(result.getReturnCode(), is(EvaluationResultCode.OK));
+ }
+
+ @Test
+ public void testFileSystemRuleWithError() throws Exception {
+ TajoConf tajoConf = new TajoConf(new YarnConfiguration());
+
+ createTajoDirectories(tajoConf);
+ Path systemResourceDir = TajoConf.getSystemResourceDir(tajoConf);
+ FileSystem defaultFs = systemResourceDir.getFileSystem(tajoConf);
+ if (defaultFs.exists(systemResourceDir)) {
+ defaultFs.delete(systemResourceDir, true);
+ }
+
+ EvaluationContext context = new EvaluationContext();
+ context.addParameter(TajoConf.class.getName(), tajoConf);
+
+ FileSystemRule fsRule = new FileSystemRule();
+ EvaluationResult result = fsRule.evaluate(context);
+
+ assertThat(result, is(notNullValue()));
+ assertThat(result.getReturnCode(), is(EvaluationResultCode.ERROR));
+ }
+}