content: /*
   * 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.ambari.server.api.query;
  
  import java.util.HashMap;
  import java.util.HashSet;
  import java.util.Map;
  import java.util.Set;
  
  import org.apache.ambari.server.controller.internal.ResourceImpl;
  import org.apache.ambari.server.controller.spi.Predicate;
  import org.apache.ambari.server.controller.spi.Resource;
  import org.apache.ambari.server.controller.utilities.PredicateBuilder;
  import org.apache.ambari.server.controller.utilities.PredicateHelper;
  import org.junit.Assert;
  import org.junit.Test;
  
  /**
   * ExtendedResourcePredicateVisitor tests.
   */
  public class ExtendedResourcePredicateVisitorTest {
    @Test
    public void testGetExtendedPredicate() throws Exception {
  
      Resource resource1 = new ResourceImpl(Resource.Type.Service);
      resource1.setProperty("name", "service1");
      Resource resource2 = new ResourceImpl(Resource.Type.Service);
      resource2.setProperty("name", "service2");
  
      Map<String, Object> resource1SubProperties1 = new HashMap<>();
      resource1SubProperties1.put("sub1/category/p1", 1);
      resource1SubProperties1.put("sub1/category/p2", 2);
      resource1SubProperties1.put("sub1/category/p3", 3);
  
      Map<String, Object> resource1SubProperties2 = new HashMap<>();
      resource1SubProperties2.put("sub1/category/p1", 1);
      resource1SubProperties2.put("sub1/category/p2", 4);
      resource1SubProperties2.put("sub1/category/p3", 6);
  
      Map<String, Object> resource1SubProperties3 = new HashMap<>();
      resource1SubProperties3.put("sub1/category/p1", 1);
      resource1SubProperties3.put("sub1/category/p2", 8);
      resource1SubProperties3.put("sub1/category/p3", 12);
  
      Set<Map<String, Object>> resource1SubPropertiesSet = new HashSet<>();
  
      resource1SubPropertiesSet.add(resource1SubProperties1);
      resource1SubPropertiesSet.add(resource1SubProperties2);
      resource1SubPropertiesSet.add(resource1SubProperties3);
  
      Map<String, Object> resource2SubProperties1 = new HashMap<>();
      resource2SubProperties1.put("sub1/category/p1", 2);
      resource2SubProperties1.put("sub1/category/p2", 2);
      resource2SubProperties1.put("sub1/category/p3", 3);
  
      Map<String, Object> resource2SubProperties2 = new HashMap<>();
      resource2SubProperties2.put("sub1/category/p1", 2);
      resource2SubProperties2.put("sub1/category/p2", 4);
      resource2SubProperties2.put("sub1/category/p3", 6);
  
      Map<String, Object> resource2SubProperties3 = new HashMap<>();
      resource2SubProperties3.put("sub1/category/p1", 2);
      resource2SubProperties3.put("sub1/category/p2", 8);
      resource2SubProperties3.put("sub1/category/p3", 12);
  
      Set<Map<String, Object>> resource2SubPropertiesSet = new HashSet<>();
  
      resource2SubPropertiesSet.add(resource2SubProperties1);
      resource2SubPropertiesSet.add(resource2SubProperties2);
      resource2SubPropertiesSet.add(resource2SubProperties3);
  
      Map<Resource, Set<Map<String, Object>>> extendedPropertyMap = new HashMap<>();
  
      extendedPropertyMap.put(resource1, resource1SubPropertiesSet);
      extendedPropertyMap.put(resource2, resource2SubPropertiesSet);
  
      Predicate predicate = new PredicateBuilder().
          property("sub1/category/p1").equals(1).toPredicate();
  
      ExtendedResourcePredicateVisitor visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      Predicate extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertTrue(extendedPredicate.evaluate(resource1));
      Assert.assertFalse(extendedPredicate.evaluate(resource2));
  
  
      predicate = new PredicateBuilder().
          property("sub1/category/p1").equals(2).toPredicate();
  
      visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertFalse(extendedPredicate.evaluate(resource1));
      Assert.assertTrue(extendedPredicate.evaluate(resource2));
  
  
  
      predicate = new PredicateBuilder().
          property("sub1/category/p2").equals(4).toPredicate();
  
      visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertTrue(extendedPredicate.evaluate(resource1));
      Assert.assertTrue(extendedPredicate.evaluate(resource2));
  
  
  
      predicate = new PredicateBuilder().
          property("sub1/category/p2").equals(5).toPredicate();
  
      visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertFalse(extendedPredicate.evaluate(resource1));
      Assert.assertFalse(extendedPredicate.evaluate(resource2));
  
  
      predicate = new PredicateBuilder().not().
          property("sub1/category/p2").equals(5).toPredicate();
  
      visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertTrue(extendedPredicate.evaluate(resource1));
      Assert.assertTrue(extendedPredicate.evaluate(resource2));
  
  
      predicate = new PredicateBuilder().
          property("sub1/category/p1").equals(1).and().
          property("sub1/category/p2").equals(4).toPredicate();
  
      visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertTrue(extendedPredicate.evaluate(resource1));
      Assert.assertFalse(extendedPredicate.evaluate(resource2));
  
  
      predicate = new PredicateBuilder().
          property("sub1/category/p1").equals(1).or().
          property("sub1/category/p2").equals(4).toPredicate();
  
      visitor = new ExtendedResourcePredicateVisitor(extendedPropertyMap);
      PredicateHelper.visit(predicate, visitor);
  
      extendedPredicate = visitor.getExtendedPredicate();
  
      Assert.assertTrue(extendedPredicate.evaluate(resource1));
      Assert.assertTrue(extendedPredicate.evaluate(resource2));
    }
  }
  