// Copyright (C) 2009 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.pusher;
import com.google.enterprise.connector.spi.Document;
import com.google.enterprise.connector.spi.RepositoryException;
import com.google.enterprise.connector.spi.SpiConstants;
import com.google.enterprise.connector.spi.SpiConstants.FeedType;
import com.google.enterprise.connector.test.ConnectorTestUtils;
import junit.framework.TestCase;
import java.util.HashMap;
import java.util.Map;
/**
* Tests some of the {@link DocUtils} methods.
*/
public class DocUtilsTest extends TestCase {
public final void testGetFeedType() throws RepositoryException {
Map<String, Object> docProps = new HashMap<String, Object>();
// Default to content feed.
docProps.put(SpiConstants.PROPNAME_DOCID, "doc1");
Document document = ConnectorTestUtils.createSimpleDocument(docProps);
assertEquals(FeedType.CONTENT, DocUtils.getFeedType(document));
// Now default to web feed.
docProps.put(SpiConstants.PROPNAME_SEARCHURL, "http://host/doc1.ext");
document = ConnectorTestUtils.createSimpleDocument(docProps);
assertEquals(FeedType.WEB, DocUtils.getFeedType(document));
// Now explicitly set content feed with searchurl.
docProps.put(SpiConstants.PROPNAME_FEEDTYPE, FeedType.CONTENT.name());
document = ConnectorTestUtils.createSimpleDocument(docProps);
assertEquals(FeedType.CONTENT, DocUtils.getFeedType(document));
// Now explicitly set web feed.
docProps.put(SpiConstants.PROPNAME_FEEDTYPE, FeedType.WEB.name());
document = ConnectorTestUtils.createSimpleDocument(docProps);
assertEquals(FeedType.WEB, DocUtils.getFeedType(document));
// Now explicitly set contenturl feed.
docProps.put(SpiConstants.PROPNAME_FEEDTYPE, FeedType.CONTENTURL.name());
document = ConnectorTestUtils.createSimpleDocument(docProps);
assertEquals(FeedType.CONTENTURL, DocUtils.getFeedType(document));
// Now test illegal value. Should go with default behavior.
docProps.put(SpiConstants.PROPNAME_FEEDTYPE, "BOGUS");
document = ConnectorTestUtils.createSimpleDocument(docProps);
assertEquals(FeedType.WEB, DocUtils.getFeedType(document));
}
public void testEnhancedFeedType() throws Exception {
// Test FeedType compatibility.
assertTrue(FeedType.CONTENT.isCompatible(FeedType.CONTENT));
assertFalse(FeedType.CONTENT.isCompatible(FeedType.CONTENTURL));
assertFalse(FeedType.CONTENT.isCompatible(FeedType.WEB));
assertFalse(FeedType.WEB.isCompatible(FeedType.CONTENT));
assertFalse(FeedType.CONTENTURL.isCompatible(FeedType.CONTENT));
assertTrue(FeedType.CONTENTURL.isCompatible(FeedType.WEB));
assertTrue(FeedType.WEB.isCompatible(FeedType.CONTENTURL));
// Test FeedType legacy strings.
assertEquals("incremental", FeedType.CONTENT.toLegacyString());
assertEquals("metadata-and-url", FeedType.WEB.toLegacyString());
assertEquals("metadata-and-url", FeedType.CONTENTURL.toLegacyString());
}
}