codegen/ec2/src/main/java/org/jclouds/codegen/ec2/queryapi/parser/AmazonEC2QueryAPIValidator.java

/**
 *
 * Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
 *
 * ====================================================================
 * Licensed 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.jclouds.codegen.ec2.queryapi.parser;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;

import java.util.Set;

import org.jclouds.codegen.ec2.queryapi.AmazonEC2QueryAPI;
import org.jclouds.codegen.ec2.queryapi.Category;
import org.jclouds.codegen.ec2.queryapi.Content;
import org.jclouds.codegen.ec2.queryapi.DataType;
import org.jclouds.codegen.ec2.queryapi.Query;

import com.google.common.base.Function;
import com.google.common.collect.Iterables;
import com.google.common.collect.Sets;

public class AmazonEC2QueryAPIValidator extends AmazonEC2QueryAPIExpectations {
   protected AmazonEC2QueryAPIValidator validateQueriesInCategory(String categoryName) {
      for (String query : expectedQueryNamesForCategoryName.get(categoryName)) {
         validateQueryInCategory(query, categoryName);
      }
      return this;
   }

   protected AmazonEC2QueryAPIValidator validateQueryInCategory(String queryName,
            String categoryName) {
      Category category = getModel().getCategories().get(categoryName);
      checkState(category != null, String.format("category %1$s not present", categoryName));
      Query query = category.getQueries().get(queryName);
      validateTopLevelType(query);
      checkState(category != null, String.format("query %1$s not present in category %2$s",
               queryName, categoryName));
      checkState(query.getResponseType().equals(query.getType() + "Response"), String.format(
               "Invalid responseType for %1$s [%2$s] should be %1$sResponse", queryName, query
                        .getResponseType()));
      checkNotNull(query.getDescription(), String.format("%1$s: getDescription()", query.getType()));

      validateTopLevelType(query);

      DataType response = getModel().getDataTypes().get(query.getResponseType());
      checkNotNull(response, String.format("response %1$s not present in domain for %2$s", query
               .getResponseType(), query.getType()));
      checkState(query.getResponseType().equals(query.getType() + "Response"));
      checkState(response.getType().equals(query.getType() + "Response"));
      validateTopLevelType(response);
      return validateDataType(query);
   }

   protected void validateTopLevelType(DataType type) {
      checkState(type.getAncestor() == null, String
               .format("%1$s should not have an ancestor", type));
      checkNotNull(type.getExampleCode(), String.format("%1$s: getExampleCode()", type.getType()));
      checkNotNull(type.getExampleHTML(), String.format("%1$s: getExampleHTML()", type.getType()));
   }

   protected AmazonEC2QueryAPIValidator validateDataType(DataType dataType) {
      checkState(dataType.getSee().size() >= 1, "see should have at least one entry");
      Set<String> fieldNames = Sets.newTreeSet(Iterables.transform(dataType.getContents(),
               new Function<Content, String>() {
                  public String apply(Content field) {
                     return field.getName();
                  }
               }));
      checkState(Sets.difference(fieldNames,
               expectedFieldNamesForDataTypeName.get(dataType.getType())).size() == 0, String
               .format("%1$s hasn't the correct fields.  has [%2$s] should have [%3$s]", dataType
                        .getType(), fieldNames, expectedFieldNamesForDataTypeName.get(dataType
                        .getType())));
      return this;
   }

   private AmazonEC2QueryAPI model;

   public AmazonEC2QueryAPIValidator validateCommands() {
      for (String categoryName : expectedQueryNamesForCategoryName.keySet()) {
         for (String queryName : expectedQueryNamesForCategoryName.get(categoryName)) {
            validateQueryInCategory(queryName, categoryName);
         }
      }
      return this;
   }

   public AmazonEC2QueryAPIValidator validateDomain() {
      for (String dataTypeName : expectedFieldNamesForDataTypeName.keySet()) {
         if (!this.queryNames.contains(dataTypeName))
            validateDataType(checkNotNull(getModel().getDataTypes().get(dataTypeName), dataTypeName));
      }
      return this;
   }

   public AmazonEC2QueryAPIValidator validateCategories() {
      checkState(Sets.difference(getModel().getCategories().keySet(),
               expectedQueryNamesForCategoryName.keySet()).size() == 0);
      checkState(getModel().getCategories().keySet().size() == expectedQueryNamesForCategoryName
               .keySet().size());
      return this;
   }

   public void setModel(AmazonEC2QueryAPI model) {
      this.model = model;
   }

   public AmazonEC2QueryAPI getModel() {
      return model;
   }

}