LarsRepositoryFixture.java

/*******************************************************************************
 * Copyright (c) 2015 IBM Corp.
 *
 * 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.ibm.ws.lars.testutils.fixtures;

import static org.junit.Assert.assertEquals;

import java.net.URI;
import java.net.URISyntaxException;

import com.ibm.ws.repository.connections.RepositoryConnection;
import com.ibm.ws.repository.connections.RestRepositoryConnection;
import com.ibm.ws.repository.transport.client.RepositoryReadableClient;
import com.ibm.ws.repository.transport.client.RepositoryWriteableClient;
import com.ibm.ws.repository.transport.model.Asset;

/**
 *
 */
public class LarsRepositoryFixture extends RepositoryFixture {

    protected final String repositoryUrl;

    public static LarsRepositoryFixture createFixture(String repositoryUrl, String apiKey, String adminId, String adminPassword, String userId, String userPassword) {
        RestRepositoryConnection adminConnection = new RestRepositoryConnection(adminId, adminPassword, apiKey, repositoryUrl);
        RestRepositoryConnection userConnection = new RestRepositoryConnection(userId, userPassword, apiKey, repositoryUrl);

        RepositoryReadableClient adminClient = adminConnection.createClient();
        RepositoryWriteableClient writableClient = (RepositoryWriteableClient) adminClient;
        RepositoryReadableClient userClient = userConnection.createClient();
        return new LarsRepositoryFixture(adminConnection, userConnection, adminClient, writableClient, userClient, repositoryUrl);
    }

    protected LarsRepositoryFixture(RepositoryConnection adminConnection, RepositoryConnection userConnection, RepositoryReadableClient adminClient,
                                    RepositoryWriteableClient writableClient, RepositoryReadableClient userClient, String repositoryUrl) {
        super(adminConnection, userConnection, adminClient, writableClient, userClient);
        this.repositoryUrl = repositoryUrl;
    }

    @Override
    protected void createCleanRepository() throws Exception {
        for (Asset asset : adminClient.getAllAssets()) {
            writableClient.deleteAssetAndAttachments(asset.get_id());
        }
        assertEquals("Repository resources present", 0, adminConnection.getAllResourcesWithDupes().size());
    }

    @Override
    protected void cleanupRepository() throws Exception {
        for (Asset asset : adminClient.getAllAssets()) {
            writableClient.deleteAssetAndAttachments(asset.get_id());
        }
    }

    @Override
    public boolean isUpdateSupported() {
        return false;
    }

    @Override
    public String getHostedFileRoot() throws URISyntaxException {
        // Test files are served under /testFiles on the same host and port as the test repository
        URI uri = new URI(repositoryUrl);
        URI baseUri = new URI(uri.getScheme(), null, uri.getHost(), uri.getPort(), "/testFiles", null, null);
        return baseUri.toString();
    }

    @Override
    public String toString() {
        return "LARS repo";
    }

}