Пример #1
0
 public void executeCommand(MojiCommand command) throws IOException {
   Tracker tracker = null;
   CommunicationException lastException = null;
   for (int attempt = 0; attempt < maxAttempts; attempt++) {
     try {
       tracker = trackerFactory.getTracker();
       log.debug("executing {}", command);
       if (maxAttempts > 1) {
         log.debug("Attempt #{}", attempt);
       }
       command.executeWithTracker(tracker);
       return;
     } catch (CommunicationException e) {
       lastException = e;
     } finally {
       if (tracker != null) {
         tracker.close();
       }
     }
   }
   if (maxAttempts > 1) {
     log.debug("All {} attempts failed", maxAttempts);
   }
   throw lastException;
 }
Пример #2
0
  @Test
  public void everyThingClosesWithFirstTrackerFail() throws IOException {
    when(address.size()).thenReturn(2);
    when(mockTrackerFactory.getTracker())
        .thenAnswer(
            new Answer<Tracker>() {
              boolean first = true;

              @Override
              public Tracker answer(InvocationOnMock invocation) throws Throwable {
                if (first) {
                  first = false;
                  throw new CommunicationException("Errror on communication");
                }
                return mockTracker;
              }
            });

    stream.write(1);
    stream.close();

    verify(mockOutputStream).flush();
    verify(mockOutputStream).close();
    verify(mockHttpConnection).disconnect();
    verify(mockTracker).createClose(KEY, DOMAIN, mockDestination, 1);
    verify(mockTracker).close();
    verify(mockWriteLock).unlock();
  }
Пример #3
0
  @Before
  public void setUp() throws IOException {
    URL url = new URL("http://www.last.fm/");
    when(mockDestination.getPath()).thenReturn(url);
    when(mockHttpFactory.newConnection(url)).thenReturn(mockHttpConnection);
    when(mockHttpConnection.getInputStream()).thenReturn(mockInputStream);
    when(mockHttpConnection.getOutputStream()).thenReturn(mockOutputStream);
    when(mockHttpConnection.getResponseMessage()).thenReturn("message");
    when(mockHttpConnection.getResponseCode()).thenReturn(HttpURLConnection.HTTP_OK);
    when(mockTrackerFactory.getTracker()).thenReturn(mockTracker);
    when(address.size()).thenReturn(1);
    when(mockTrackerFactory.getAddresses()).thenReturn(address);

    stream =
        new FileUploadOutputStream(
            mockTrackerFactory, mockHttpFactory, KEY, DOMAIN, mockDestination, mockWriteLock);
  }
Пример #4
0
 Executor(TrackerFactory trackerFactory) {
   this.trackerFactory = trackerFactory;
   maxAttempts = trackerFactory.getAddresses().size();
 }