/*
 * Copyright (c) 2012, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.ts.tests.jaxrs.api.client.clientrequestcontext;

import java.util.concurrent.atomic.AtomicInteger;

import javax.ws.rs.client.ClientRequestContext;
import javax.ws.rs.core.Response;

import com.sun.ts.lib.harness.EETest.Fault;

public class RemovePropertyProvider extends ContextProvider {
  private AtomicInteger counter;

  public RemovePropertyProvider(AtomicInteger counter) {
    super();
    this.counter = counter;
  }

  @Override
  protected void checkFilterContext(ClientRequestContext context) throws Fault {
    String propName = "PROPERTY";
    switch (counter.incrementAndGet()) {
    case 1:
      Object property = context.getProperty(propName);
      assertFault(property == null, "property already exist");
      context.setProperty(propName, propName);
      break;
    case 2:
      property = context.getProperty(propName);
      assertFault(property != null, "property not exist");
      context.removeProperty(propName);
      break;
    case 3:
      property = context.getProperty(propName);
      assertFault(property == null, "property already exist");
      Response response = Response.ok("NULL").build();
      context.abortWith(response);
      break;
    }
  }

  /**
   * @param conditionTrue
   * @param message
   * @throws Fault
   *           when conditionTrue is not met with message provided
   */
  protected static void //
      assertFault(boolean conditionTrue, Object... message) throws Fault {
    if (!conditionTrue) {
      StringBuilder sb = new StringBuilder();
      for (Object msg : message)
        sb.append(msg).append(" ");
      throw new Fault(sb.toString());
    }
  }

}
