public void CanBeCanceled()
{
using (TestEnvironment testEnvironment = TestEnvironment.Create())
{
TransientTestFolder folder = testEnvironment.CreateFolder(createFolder: true);
DownloadFile downloadFile = new DownloadFile
{
BuildEngine = _mockEngine,
DestinationFolder = new TaskItem(folder.Path),
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new StringContent(new String('!', 10000000)),
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://largedownload/foo.txt")
}),
SourceUrl = "http://largedownload/foo.txt"
};
Task <bool> task = Task.Run(() => downloadFile.Execute());
downloadFile.Cancel();
task.Wait(TimeSpan.FromSeconds(1)).ShouldBeTrue();
task.Result.ShouldBeFalse();
}
}
public void CanDownloadToFolder()
{
using (TestEnvironment testEnvironment = TestEnvironment.Create())
{
TransientTestFolder folder = testEnvironment.CreateFolder(createFolder: false);
DownloadFile downloadFile = new DownloadFile
{
BuildEngine = _mockEngine,
DestinationFolder = new TaskItem(folder.Path),
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new StringContent("Success!"),
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://success/foo.txt")
}),
SourceUrl = "http://success/foo.txt"
};
downloadFile.Execute().ShouldBeTrue();
FileInfo file = new FileInfo(Path.Combine(folder.Path, "foo.txt"));
file.Exists.ShouldBeTrue(() => file.FullName);
File.ReadAllText(file.FullName).ShouldBe("Success!");
downloadFile.DownloadedFile.ItemSpec.ShouldBe(file.FullName);
}
}
public void SkipUnchangedFiles()
{
using (TestEnvironment testEnvironment = TestEnvironment.Create())
{
TransientTestFolder folder = testEnvironment.CreateFolder(createFolder: true);
DownloadFile downloadFile = new DownloadFile
{
BuildEngine = _mockEngine,
DestinationFolder = new TaskItem(folder.Path),
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new StringContent("C197675A3CC64CAA80680128CF4578C9")
{
Headers =
{
LastModified = DateTimeOffset.UtcNow.AddDays(-1)
}
},
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://success/foo.txt")
}),
SkipUnchangedFiles = true,
SourceUrl = "http://success/foo.txt"
};
testEnvironment.CreateFile(folder, "foo.txt", "C197675A3CC64CAA80680128CF4578C9");
downloadFile.Execute().ShouldBeTrue();
_mockEngine.Log.ShouldContain("Did not download file from \"http://success/foo.txt\"", () => _mockEngine.Log);
}
}
public void InvalidUrlLogsError()
{
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
SourceUrl = "&&&&&"
};
downloadFile.Execute().ShouldBeFalse(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain("MSB3921");
}
public void NotFoundLogsError()
{
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.NotFound)),
SourceUrl = "http://notfound/foo.txt"
};
downloadFile.Execute().ShouldBeFalse(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain("Response status code does not indicate success: 404 (Not Found).");
}
public void RetryOnResponseError()
{
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.RequestTimeout)),
Retries = 1,
RetryDelayMilliseconds = 100,
SourceUrl = "http://notfound/foo.txt"
};
downloadFile.Execute().ShouldBeFalse(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain("MSB3924", () => _mockEngine.Log);
}
public void NotFoundLogsError()
{
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.NotFound)),
SourceUrl = "http://notfound/foo.txt"
};
downloadFile.Execute().ShouldBeFalse(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain(
NativeMethodsShared.IsMono
? @"ERROR : MSB3923: Failed to download file ""http://notfound/foo.txt"". 404 (Not Found)"
: @"ERROR : MSB3923: Failed to download file ""http://notfound/foo.txt"". Response status code does not indicate success: 404 (Not Found).");
}
public void UnknownFileNameLogsError()
{
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new StringContent("Success!"),
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://unknown/")
}),
SourceUrl = "http://unknown/"
};
downloadFile.Execute().ShouldBeFalse(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain("MSB3922", () => _mockEngine.Log);
}
public void CanGetFileNameFromResponseHeader()
{
const string filename = "C6DDD10A99E149F78FA11F133127BF38.txt";
HttpResponseMessage response = new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new StringContent("Success!")
{
Headers =
{
ContentDisposition = new ContentDispositionHeaderValue(DispositionTypeNames.Attachment)
{
FileName = filename
}
}
},
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://success/foo.txt")
};
using (TestEnvironment testEnvironment = TestEnvironment.Create())
{
TransientTestFolder folder = testEnvironment.CreateFolder(createFolder: false);
DownloadFile downloadFile = new DownloadFile
{
BuildEngine = _mockEngine,
DestinationFolder = new TaskItem(folder.Path),
DestinationFileName = new TaskItem(filename),
HttpMessageHandler = new MockHttpMessageHandler((message, token) => response),
SourceUrl = "http://success/foo.txt"
};
downloadFile.Execute().ShouldBeTrue();
FileInfo file = new FileInfo(Path.Combine(folder.Path, filename));
file.Exists.ShouldBeTrue(() => file.FullName);
File.ReadAllText(file.FullName).ShouldBe("Success!");
downloadFile.DownloadedFile.ItemSpec.ShouldBe(file.FullName);
}
}
public void AbortOnTimeout()
{
CancellationTokenSource timeout = new CancellationTokenSource();
timeout.Cancel();
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
HttpMessageHandler = new MockHttpMessageHandler((message, token) =>
{
// Http timeouts manifest as "OperationCanceledExceptions" from the handler, simulate that
throw new OperationCanceledException(timeout.Token);
}),
Retries = 1,
RetryDelayMilliseconds = 100,
SourceUrl = "http://notfound/foo.txt"
};
downloadFile.Execute().ShouldBeFalse(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain("MSB3923", () => _mockEngine.Log);
}
public async Task NoRunawayLoop()
{
DownloadFile downloadFile = null;
bool failed = false;
downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
HttpMessageHandler = new MockHttpMessageHandler((message, token) =>
{
token.ThrowIfCancellationRequested();
downloadFile.Cancel();
if (!failed)
{
failed = true;
return(new HttpResponseMessage(HttpStatusCode.InternalServerError));
}
return(new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new StringContent("Success!"),
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://success/foo.txt")
});
}),
Retries = 2,
RetryDelayMilliseconds = 100,
SourceUrl = "http://notfound/foo.txt"
};
var runaway = Task.Run(() => downloadFile.Execute());
await Task.Delay(TimeSpan.FromSeconds(1));
runaway.IsCompleted.ShouldBeTrue("Task did not cancel");
var result = await runaway;
result.ShouldBeFalse(() => _mockEngine.Log);
}
public void RetryOnDownloadError()
{
const string content = "Foo";
bool hasThrown = false;
using (TestEnvironment testEnvironment = TestEnvironment.Create())
{
TransientTestFolder folder = testEnvironment.CreateFolder(createFolder: false);
DownloadFile downloadFile = new DownloadFile()
{
BuildEngine = _mockEngine,
DestinationFolder = new TaskItem(folder.Path),
HttpMessageHandler = new MockHttpMessageHandler((message, token) => new HttpResponseMessage(HttpStatusCode.OK)
{
Content = new MockHttpContent(content.Length, stream =>
{
if (!hasThrown)
{
hasThrown = true;
throw new WebException("Error", WebExceptionStatus.ReceiveFailure);
}
return(new MemoryStream(Encoding.Unicode.GetBytes(content)).CopyToAsync(stream));
}),
RequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://success/foo.txt")
}),
Retries = 1,
RetryDelayMilliseconds = 100,
SourceUrl = "http://success/foo.txt"
};
downloadFile.Execute().ShouldBeTrue(() => _mockEngine.Log);
_mockEngine.Log.ShouldContain("MSB3924", () => _mockEngine.Log);
}
}
