/* Copyright 2014 Google Inc. All rights reserved.
 *
 *  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.jenkins.flakyTestHandler.plugin;

import com.google.jenkins.flakyTestHandler.junit.FlakyTestResult;

import hudson.remoting.VirtualChannel;
import org.kohsuke.stapler.DataBoundConstructor;

import java.io.IOException;
import java.util.Objects;

import hudson.Extension;
import hudson.FilePath;
import hudson.Launcher;
import hudson.model.AbstractBuild;
import hudson.model.BuildListener;
import hudson.model.Descriptor;
import hudson.model.Run;
import hudson.model.TaskListener;
import hudson.tasks.junit.TestDataPublisher;
import hudson.tasks.junit.TestResult;
import hudson.tasks.junit.TestResultAction;
import hudson.tasks.test.AbstractTestResultAction;

/**
 * Publisher for publishing rerun information
 *
 * @author Qingzhou Luo
 */
public class JUnitFlakyTestDataPublisher
    extends TestDataPublisher {

  @DataBoundConstructor
  public JUnitFlakyTestDataPublisher() {
  }

  @Override
  public TestResultAction.Data contributeTestData(Run<?, ?> run, FilePath workspace, Launcher launcher, TaskListener listener, TestResult testResult)
      throws IOException, InterruptedException {
    VirtualChannel channel = launcher.getChannel();
    if(channel == null) {
          throw new InterruptedException("Could not get channel to run a program remotely.");
    }
    FlakyTestResult flakyTestResult = channel.call(new FlakyTestResultCollector(testResult));
    // TODO consider the possibility that there is >1 such action
    flakyTestResult.freeze(run.getAction(AbstractTestResultAction.class), run);
    return new JUnitFlakyTestData(flakyTestResult);
  }

  @Override
  public TestResultAction.Data getTestData(AbstractBuild<?, ?> build, Launcher launcher, BuildListener listener, TestResult testResult) {
      try {
          FilePath workspace = build.getWorkspace();
          if (workspace == null) {
              throw new IOException("no workspace in " + build);
          }
          return contributeTestData(build, workspace, launcher, null, testResult);
      } catch (IOException e) {
          throw new IllegalStateException(e);
      } catch (InterruptedException e) {
          throw new IllegalStateException(e);
      }
  }

  @Override
  public DescriptorImpl getDescriptor() {
    return (DescriptorImpl) super.getDescriptor();
  }


  @Extension
  public static class DescriptorImpl extends Descriptor<TestDataPublisher> {

    @Override
    public String getDisplayName() {
      return "Publish JUnit flaky tests reports";
    }
  }
}


@@ -2203,596 +2203,8 @@
  %7D%0A%0A
-  @Override%0A  public TestResultAction.Data getTestData(AbstractBuild%3C?, ?%3E build, Launcher launcher, BuildListener listener, TestResult testResult) %7B%0A      try %7B%0A          FilePath workspace = build.getWorkspace();%0A          if (workspace == null) %7B%0A              throw new IOException(%22no workspace in %22 + build);%0A          %7D%0A          return contributeTestData(build, workspace, launcher, null, testResult);%0A      %7D catch (IOException e) %7B%0A          throw new IllegalStateException(e);%0A      %7D catch (InterruptedException e) %7B%0A          throw new IllegalStateException(e);%0A      %7D%0A  %7D%0A%0A
   @O

Remove deprecated getTestData method