/*
    * 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
    *
    *      https://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.commons.jexl3;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertNotNull;
  import static org.junit.jupiter.api.Assertions.assertNotSame;
  
  import java.util.Map;
  import java.util.Set;
  import java.util.concurrent.atomic.AtomicInteger;
  
  import org.apache.commons.jexl3.internal.ArrayBuilder;
  import org.apache.commons.jexl3.internal.MapBuilder;
  import org.apache.commons.jexl3.internal.SetBuilder;
  import org.junit.jupiter.api.Test;
  
  /**
   * Counting the number of times map,sets,array literals are created.
   * <p>
   * Originally intended to prove one could cache results of constant map literals when used to create
   * libraries (a map of lamba).
   * However, those literals must be newly instantiated since their result may be modified; there is
   * thus no point in trying to cache them. (Think harder and nod, don't try again.)
   * These pointless tests as a reminder of 'why' those behave the way they do.
   * </p>
   */
  class CollectionLiteralTest extends JexlTestCase {
      public static class Arithmetic363 extends JexlArithmetic {
          final AtomicInteger maps = new AtomicInteger();
          final AtomicInteger sets = new AtomicInteger();
          final AtomicInteger arrays = new AtomicInteger();
  
          public Arithmetic363(final boolean strict) {
              super(strict);
          }
  
          @Override public ArrayBuilder arrayBuilder(final int size, final boolean extended) {
              return new CountingArrayBuilder(arrays, size, extended);
          }
          @Override public MapBuilder mapBuilder(final int size, final boolean extended) {
              return new CountingMapBuilder(maps, size, extended);
          }
          @Override public SetBuilder setBuilder(final int size, final boolean extended) {
              return new CountingSetBuilder(sets, size, extended);
          }
      }
  
      static class CountingArrayBuilder extends ArrayBuilder {
          final AtomicInteger count;
  
          public CountingArrayBuilder(final AtomicInteger ai, final int size, final boolean extended) {
              super(size, extended);
              count = ai;
          }
  
          @Override public Object create(final boolean extended) {
              final Object array = super.create(extended);
              count.incrementAndGet();
              return array;
          }
      }
  
      static class CountingMapBuilder extends MapBuilder {
          final AtomicInteger count;
          public CountingMapBuilder(final AtomicInteger ai, final int size, final boolean extended) {
              super(size, extended);
              count = ai;
          }
          @Override public Map<Object, Object> create() {
              final Map<Object, Object> map = super.create();
              count.incrementAndGet();
              return map;
          }
      }
  
      static class CountingSetBuilder extends SetBuilder {
          final AtomicInteger count;
          public CountingSetBuilder(final AtomicInteger ai, final int size, final boolean extended) {
              super(size, extended);
              count = ai;
          }
          @Override public Set<?> create() {
              final Set<?> set = super.create();
              count.incrementAndGet();
             return set;
         }
     }
 
     public CollectionLiteralTest() {
         super("CollectionLiteralTest");
     }
 
     @Test
     void testArrayBuilder() {
         final Arithmetic363 jc = new Arithmetic363(true);
         final JexlEngine jexl = new JexlBuilder().cache(4).arithmetic(jc).create();
         JexlScript script;
         Object result;
 
         script = jexl.createScript("[ (x)->{ 1 + x; }, (y)->{ y - 1; } ]");
         Object previous = null;
         for (int i = 0; i < 4; ++i) {
             result = script.execute(null);
             assertNotNull(result);
             assertNotSame(previous, result);
             previous = result;
             assertEquals(1 + i, jc.arrays.get());
         }
     }
 
     @Test
     void testMapLBuilder() {
         final Arithmetic363 jc = new Arithmetic363(true);
         final JexlEngine jexl = new JexlBuilder().cache(4).arithmetic(jc).create();
         JexlScript script;
         Object result;
 
         script = jexl.createScript("{ 'x':(x)->{ 1 + x; }, 'y' : (y)->{ y - 1; } }");
         Object previous = null;
         for(int i = 0; i < 4; ++i) {
             result = script.execute(null);
             assertNotNull(result);
             assertNotSame(previous, result);
             previous = result;
             assertEquals(1 + i, jc.maps.get());
         }
     }
 
     @Test
     void testSetBuilder() {
         final Arithmetic363 jc = new Arithmetic363(true);
         final JexlEngine jexl = new JexlBuilder().cache(4).arithmetic(jc).create();
         JexlScript script;
         Object result;
 
         script = jexl.createScript("{ (x)->{ 1 + x; }, (y)->{ y - 1; } }");
         Object previous = null;
         for(int i = 0; i < 4; ++i) {
             result = script.execute(null);
             assertNotNull(result);
             assertNotSame(previous, result);
             previous = result;
             assertEquals(1 + i, jc.sets.get());
         }
     }
 
 }