MockDocumentHandle.java

// Copyright 2010 Google Inc.
//
// 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 com.google.enterprise.connector.util.diffing;

import com.google.enterprise.connector.spi.Document;
import com.google.enterprise.connector.spi.SpiConstants;

import org.json.JSONException;
import org.json.JSONObject;

public class MockDocumentHandle implements DocumentHandle {
  static enum Field {
    DOCUMENT_ID, EXTRA
  }

  private final String documentId;
  private final String extra;

  public MockDocumentHandle(String documentId, String extra) {
    this.documentId = documentId;
    this.extra = extra;
  }

  public String getDocumentId() {
    return documentId;
  }

  String getExtra() {
    return extra;
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result
        + ((documentId == null) ? 0 : documentId.hashCode());
    result = prime * result + ((extra == null) ? 0 : extra.hashCode());
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj) {
      return true;
    }
    if (obj == null) {
      return false;
    }
    if (getClass() != obj.getClass()) {
      return false;
    }
    MockDocumentHandle other = (MockDocumentHandle) obj;
    if (documentId == null) {
      if (other.documentId != null) {
        return false;
      }
    } else if (!documentId.equals(other.documentId)) {
      return false;
    }
    if (extra == null) {
      if (other.extra != null) {
        return false;
      }
    } else if (!extra.equals(other.extra)) {
      return false;
    }
    return true;
  }

  private JSONObject getJson() {
    JSONObject result = new JSONObject();
    try {
      result.put(Field.DOCUMENT_ID.name(), documentId);
      result.put(Field.EXTRA.name(), extra);
      return result;
    } catch (JSONException e) {
      // This cannot happen.
      throw new RuntimeException(
          "internal error: failed to encode snapshot record", e);
    }
  }

  @Override
  public String toString() {
    return getJson().toString();
  }

  public Document getDocument() {
    GenericDocument result = new GenericDocument();
    result.setProperty(SpiConstants.PROPNAME_ACTION,
        SpiConstants.ActionType.ADD.toString());
    result.setProperty(SpiConstants.PROPNAME_DOCID, getDocumentId());
    result.setProperty(SpiConstants.PROPNAME_MIMETYPE, "text/html");
    result.setProperty(SpiConstants.PROPNAME_CONTENT, getExtra());
    return result;
  }
}