/*
* Licensed to DuraSpace under one or more contributor license agreements.
* See the NOTICE file distributed with this work for additional information
* regarding copyright ownership.
*
* DuraSpace 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
*
* 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.fcrepo.kernel.modeshape.services;
import static javax.jcr.query.Query.JCR_SQL2;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import static org.mockito.Matchers.anyString;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.mockito.MockitoAnnotations.initMocks;
import static org.modeshape.jcr.api.JcrConstants.JCR_PATH;
import static org.springframework.test.util.ReflectionTestUtils.setField;
import java.util.HashMap;
import java.util.Map;
import javax.jcr.Node;
import javax.jcr.NodeIterator;
import javax.jcr.Property;
import javax.jcr.Repository;
import javax.jcr.RepositoryException;
import javax.jcr.Session;
import javax.jcr.Value;
import javax.jcr.ValueFactory;
import javax.jcr.Workspace;
import javax.jcr.query.Query;
import javax.jcr.query.QueryManager;
import javax.jcr.query.QueryResult;
import javax.jcr.query.Row;
import javax.jcr.query.RowIterator;
import javax.jcr.query.qom.QueryObjectModel;
import javax.jcr.query.qom.QueryObjectModelFactory;
import org.fcrepo.kernel.api.FedoraRepository;
import org.fcrepo.kernel.api.FedoraTypes;
import org.fcrepo.kernel.api.services.RepositoryService;
import org.fcrepo.kernel.modeshape.FedoraRepositoryImpl;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.modeshape.jcr.api.NamespaceRegistry;
/**
* <p>RepositoryServiceImplTest class.</p>
*
* @author ksclarke
*/
public class RepositoryServiceImplTest implements FedoraTypes {
private static final String TESTPID = "testObj";
private static final String TESTDSID = "testDs";
private static final String MOCKPREFIX = "mock1";
private static final Long EXPECTED_SIZE = 5L;
private static final Long EXPECTED_COUNT = 0L;
private RepositoryService testObj;
private FedoraRepository testRepo;
@Mock
private Repository mockRepo;
@Mock
private Session mockSession;
@Mock
private Node mockRootNode;
@Mock
private Node mockDsNode;
@Mock
private Property mockProp;
@Mock
private NodeIterator mockNI;
@Mock
private RowIterator mockRowIterator;
@Mock
private Workspace mockWorkspace;
@Mock
private QueryManager mockQueryManager;
@Mock
private QueryObjectModel mockQueryOM;
@Mock
private QueryResult mockQueryResult;
@Mock
private Query mockQuery;
@Mock
private RowIterator mockRI;
@Mock
private Value mockValue;
@Mock
private Row mockRow;
@Mock
private QueryObjectModelFactory mockQOMFactory;
@Mock
private ValueFactory mockFactory;
@Mock
private NamespaceRegistry mockNamespaceRegistry;
private Map<String, String> expectedNS;
@Before
public void setUp() {
initMocks(this);
final String relPath = "/" + TESTPID + "/" + TESTDSID;
final String[] mockPrefixes = {MOCKPREFIX};
expectedNS = new HashMap<>();
testRepo = new FedoraRepositoryImpl(mockRepo);
try {
when(mockSession.getRootNode()).thenReturn(mockRootNode);
when(mockRootNode.getNode(relPath)).thenReturn(mockDsNode);
when(mockRootNode.getProperty("fedora:size")).thenReturn(mockProp);
when(mockProp.getLong()).thenReturn(EXPECTED_SIZE);
when(mockRepo.login()).thenReturn(mockSession);
testObj = new RepositoryServiceImpl();
setField(testObj, "repository", testRepo);
when(mockSession.getNode("/objects")).thenReturn(mockRootNode);
when(mockRootNode.getNodes()).thenReturn(mockNI);
when(mockNI.getSize()).thenReturn(EXPECTED_SIZE);
when(mockSession.getWorkspace()).thenReturn(mockWorkspace);
when(mockWorkspace.getQueryManager()).thenReturn(mockQueryManager);
when(mockWorkspace.getNamespaceRegistry()).thenReturn(
mockNamespaceRegistry);
when(mockNamespaceRegistry.getPrefixes()).thenReturn(mockPrefixes);
when(mockQueryManager.createQuery(anyString(), eq(JCR_SQL2)))
.thenReturn(mockQuery);
when(mockQuery.execute()).thenReturn(mockQueryResult);
when(mockQueryResult.getRows()).thenReturn(mockRI);
when(mockRI.hasNext()).thenReturn(true, false);
when(mockRI.nextRow()).thenReturn(mockRow);
when(mockRow.getValue(CONTENT_SIZE)).thenReturn(mockValue);
when(mockValue.getLong()).thenReturn(EXPECTED_SIZE);
expectedNS
.put(MOCKPREFIX, mockNamespaceRegistry.getURI(MOCKPREFIX));
} catch (final RepositoryException e) {
e.printStackTrace();
fail(e.getMessage());
}
}
@Test
public void testGetRepositorySize() {
final Long actual = testObj.getRepositorySize();
assertEquals(EXPECTED_SIZE, actual);
}
@Test
public void testGetRepositoryObjectCount() {
final Long actual = testObj.getRepositoryObjectCount();
assertEquals(EXPECTED_COUNT, actual);
}
@Test
public void testGetObjectSize() throws RepositoryException {
when(mockRepo.login()).thenReturn(mockSession);
when(mockSession.getWorkspace()).thenReturn(mockWorkspace);
when(mockWorkspace.getQueryManager()).thenReturn(mockQueryManager);
when(
mockQueryManager.createQuery("SELECT [" + CONTENT_SIZE +
"] FROM [" + FEDORA_BINARY + "]", JCR_SQL2))
.thenReturn(mockQuery);
when(mockQuery.execute()).thenReturn(mockQueryResult);
when(mockQueryResult.getRows()).thenReturn(mockRowIterator);
when(mockRowIterator.hasNext()).thenReturn(true, true, true, false);
when(mockRowIterator.nextRow()).thenReturn(mockRow, mockRow, mockRow);
when(mockRow.getValue(CONTENT_SIZE)).thenReturn(mockValue);
when(mockValue.getLong()).thenReturn(5L, 10L, 1L);
final long count = testObj.getRepositorySize();
assertEquals("Got wrong count!", 16L, count);
verify(mockSession).logout();
verify(mockSession, never()).save();
}
@Test
public void testGetObjectCount() throws RepositoryException {
when(mockRepo.login()).thenReturn(mockSession);
when(mockSession.getWorkspace()).thenReturn(mockWorkspace);
when(mockWorkspace.getQueryManager()).thenReturn(mockQueryManager);
when(
mockQueryManager.createQuery("SELECT [" + JCR_PATH +
"] FROM [" + FEDORA_CONTAINER + "]", JCR_SQL2))
.thenReturn(mockQuery);
when(mockQuery.execute()).thenReturn(mockQueryResult);
when(mockQueryResult.getRows()).thenReturn(mockRowIterator);
when(mockRowIterator.getSize()).thenReturn(3L);
final long count = testObj.getRepositoryObjectCount();
assertEquals(3L, count);
verify(mockSession).logout();
verify(mockSession, never()).save();
}
}