org.commonjava.indy.folo.ftest.report.UseChecksumFromTransferDecoratorForTrackingRecord_Sha1FileTest.java Source code

Introduction

Here is the source code for org.commonjava.indy.folo.ftest.report.UseChecksumFromTransferDecoratorForTrackingRecord_Sha1FileTest.java

Source

/**
 * Copyright (C) 2011-2018 Red Hat, Inc. (https://github.com/Commonjava/indy)
 *
 * 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.commonjava.indy.folo.ftest.report;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.io.IOUtils;
import org.commonjava.indy.client.core.util.UrlUtils;
import org.commonjava.indy.folo.client.IndyFoloAdminClientModule;
import org.commonjava.indy.folo.client.IndyFoloContentClientModule;
import org.commonjava.indy.folo.dto.TrackedContentDTO;
import org.commonjava.indy.folo.dto.TrackedContentEntryDTO;
import org.commonjava.indy.ftest.core.category.BytemanTest;
import org.commonjava.indy.model.core.RemoteRepository;
import org.commonjava.indy.model.core.StoreKey;
import org.commonjava.test.http.expect.ExpectationServer;
import org.jboss.byteman.contrib.bmunit.BMRule;
import org.jboss.byteman.contrib.bmunit.BMRules;
import org.jboss.byteman.contrib.bmunit.BMUnitConfig;
import org.jboss.byteman.contrib.bmunit.BMUnitRunner;
import org.junit.Rule;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.util.Set;

import static org.commonjava.indy.model.core.StoreType.remote;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.junit.Assert.assertThat;

@RunWith(BMUnitRunner.class)
@BMUnitConfig(debug = true)
@Category(BytemanTest.class)
public class UseChecksumFromTransferDecoratorForTrackingRecord_Sha1FileTest extends AbstractTrackingReportTest {

    @Rule
    public ExpectationServer server = new ExpectationServer();

    @BMRules(rules = {
            @BMRule(name = "setup_metadata_countdown", targetClass = "^org.commonjava.indy.content.ContentDigester", isInterface = true, targetMethod = "<init>", targetLocation = "ENTRY", action = "System.out.println(\"SETUP COUNTDOWN\"); createCountDown(\"COUNTDOWN\", 1);"),
            @BMRule(name = "prevent_successive_metadata_additions", targetClass = "^org.commonjava.indy.content.ContentDigester", isInterface = true, targetMethod = "addMetadata", targetLocation = "ENTRY", binding = "path:String = $1.getPath();", condition = "path.endsWith(\"path/to/foo.class\") && countDown(\"COUNTDOWN\")", action = "System.out.println(\"RETURN NULL\"); return null;") })
    @Test
    public void run() throws Exception {
        final String trackingId = newName();
        final String repoId = "repo";
        final String path = "/org/bar/foo/1/foo-1.pom.sha1";

        final InputStream stream = new ByteArrayInputStream(
                DigestUtils.md5("This is a test with the same content each time."));

        server.expect(server.formatUrl(repoId, path), 200, stream);

        RemoteRepository rr = new RemoteRepository(repoId, server.formatUrl(repoId));
        rr = client.stores().create(rr, "adding test remote", RemoteRepository.class);

        final ByteArrayOutputStream baos = new ByteArrayOutputStream();
        final InputStream in = client.module(IndyFoloContentClientModule.class).get(trackingId, remote, repoId,
                path);

        IOUtils.copy(in, baos);
        in.close();

        final byte[] bytes = baos.toByteArray();

        final String md5 = md5Hex(bytes);
        final String sha256 = sha256Hex(bytes);

        assertThat(md5, equalTo(DigestUtils.md5Hex(bytes)));
        assertThat(sha256, equalTo(DigestUtils.sha256Hex(bytes)));

        waitForEventPropagation();

        assertThat(client.module(IndyFoloAdminClientModule.class).sealTrackingRecord(trackingId), equalTo(true));

        final TrackedContentDTO report = client.module(IndyFoloAdminClientModule.class)
                .getTrackingReport(trackingId);
        assertThat(report, notNullValue());

        final Set<TrackedContentEntryDTO> downloads = report.getDownloads();

        assertThat(downloads, notNullValue());
        assertThat(downloads.size(), equalTo(1));

        final TrackedContentEntryDTO entry = downloads.iterator().next();

        System.out.println(entry);

        assertThat(entry, notNullValue());
        assertThat(entry.getStoreKey(), equalTo(new StoreKey(remote, repoId)));
        assertThat(entry.getPath(), equalTo(path));
        assertThat(entry.getLocalUrl(), equalTo(client.content().contentUrl(remote, repoId, path)));
        assertThat(entry.getOriginUrl(), equalTo(server.formatUrl(repoId, path)));
        assertThat(entry.getMd5(), equalTo(md5));
        assertThat(entry.getSha256(), equalTo(sha256));
    }

    @Override
    protected boolean createStandardStores() {
        return false;
    }
}