diff --git a/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.Tests.cs b/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.Tests.cs deleted file mode 100644 index 3ce02f5..0000000 --- a/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.Tests.cs +++ /dev/null @@ -1,92 +0,0 @@ -using System; -using System.IO; -using Microsoft.Extensions.Logging; -using Moq; -using RestSharp; - -namespace Hua.DotNet.Code.Helper.Http -{ - public class RestClientHttpHelperTests - { - [Test] - public void DownloadFile_ShouldReturnTrue_WhenDownloadSucceeds() - { - // Arrange - var loggerMock = new Mock>(); - var helper = new RestClientHttpHelper(loggerMock.Object); - var testUrl = "https://example.com/testfile.txt"; - var testFilePath = Path.GetTempFileName(); - - var restClientMock = new Mock(); - var responseData = System.Text.Encoding.UTF8.GetBytes("Test content"); - restClientMock.Setup(c => c.DownloadData(It.IsAny())).Returns(responseData); - - var type = helper.GetType(); - var getHttpClientMethod = type.GetMethod("GetHttpClient", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance); - var mockGetter = new Func(() => restClientMock.Object); - - var mockMethod = mockGetter.Method; - var mockInstance = mockGetter.Target; - - var proxy = new DynamicProxy(helper, getHttpClientMethod, mockMethod, mockInstance); - - // Act - var result = helper.DownloadFile(testUrl, testFilePath); - - // Assert - Assert.That(result, Is.True); - Assert.That(File.Exists(testFilePath), Is.True); - File.Delete(testFilePath); - } - - [Test] - public void DownloadFile_ShouldReturnFalse_WhenDownloadFails() - { - // Arrange - var loggerMock = new Mock>(); - var helper = new RestClientHttpHelper(loggerMock.Object); - var testUrl = "https://example.com/nonexistentfile.txt"; - var testFilePath = Path.GetTempFileName(); - - var restClientMock = new Mock(); - restClientMock.Setup(c => c.DownloadData(It.IsAny())).Returns((byte[])null); - - var type = helper.GetType(); - var getHttpClientMethod = type.GetMethod("GetHttpClient", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance); - var mockGetter = new Func(() => restClientMock.Object); - - var mockMethod = mockGetter.Method; - var mockInstance = mockGetter.Target; - - var proxy = new DynamicProxy(helper, getHttpClientMethod, mockMethod, mockInstance); - - // Act - var result = helper.DownloadFile(testUrl, testFilePath); - - // Assert - Assert.That(result, Is.False); - Assert.That(File.Exists(testFilePath), Is.False); - } - } - - public class DynamicProxy - { - private readonly object _target; - private readonly System.Reflection.MethodInfo _originalMethod; - private readonly System.Reflection.MethodInfo _mockMethod; - private readonly object _mockInstance; - - public DynamicProxy(object target, System.Reflection.MethodInfo originalMethod, System.Reflection.MethodInfo mockMethod, object mockInstance) - { - _target = target; - _originalMethod = originalMethod; - _mockMethod = mockMethod; - _mockInstance = mockInstance; - } - - public object Invoke(object[] parameters) - { - return _mockMethod.Invoke(_mockInstance, parameters); - } - } -} \ No newline at end of file diff --git a/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.cs b/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.cs index 1151f36..a5d5080 100644 --- a/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.cs +++ b/src/Hua.Dotnet.Code/Helper/Http/RestClientHttpHelper.cs @@ -1,5 +1,6 @@ using System; using System.Collections.Generic; +using System.IO; using System.Net.Http; using System.Threading; using System.Threading.Tasks; @@ -77,12 +78,12 @@ public class RestClientHttpHelper : IHttpHelper var client = GetHttpClient(); var request = new RestRequest(url, Method.Get); var response = client.DownloadData(request); - if (response != null) - { - System.IO.File.WriteAllBytes(fileFullName, response); - return true; - } - return false; + if (response == null) return false; + + var dir = Path.GetDirectoryName(fileFullName); + Directory.CreateDirectory(dir); + File.WriteAllBytes(fileFullName, response); + return true; } catch (Exception ex) { diff --git a/test/Hua.Dotnet.Code.Test/Hua.Dotnet.Code.Test.csproj b/test/Hua.Dotnet.Code.Test/Hua.Dotnet.Code.Test.csproj index 8b5797e..4c02342 100644 --- a/test/Hua.Dotnet.Code.Test/Hua.Dotnet.Code.Test.csproj +++ b/test/Hua.Dotnet.Code.Test/Hua.Dotnet.Code.Test.csproj @@ -1,4 +1,4 @@ - + net8.0 @@ -10,11 +10,12 @@ - - - - + + + + + diff --git a/test/Hua.Dotnet.Code.Test/RestClientHttpHelperTests.cs b/test/Hua.Dotnet.Code.Test/RestClientHttpHelperTests.cs new file mode 100644 index 0000000..d3cc14c --- /dev/null +++ b/test/Hua.Dotnet.Code.Test/RestClientHttpHelperTests.cs @@ -0,0 +1,78 @@ +using Hua.DotNet.Code.Helper.Http; +using Microsoft.Extensions.Logging; +using Moq; + +namespace Hua.Dotnet.Code.Test +{ + [TestFixture] + public class RestClientHttpHelperTests + { + //[Test] + //public async Task GetAsync_ShouldReturnSuccessResponse() + //{ + // // Arrange + // var mockLogger = new Mock>(); + // var helper = new RestClientHttpHelper(mockLogger.Object); + + // // Act + // var result = await helper.GetAsync("https://api.example.com"); + + // // Assert + // Assert.IsTrue(result.IsSuccessStatusCode); + //} + + //[Test] + //public async Task PostAsync_ShouldHandleException() + //{ + // // Arrange + // var mockLogger = new Mock>(); + // var helper = new RestClientHttpHelper(mockLogger.Object); + + // // Act & Assert + // Assert.ThrowsAsync(() => + // helper.PostAsync("https://api.example.com", new StringContent(""))); + //} + + [Test] + public void DownloadFile_ShouldCreateLocalFile() + { + // Arrange + var testPath = Path.Combine(Path.GetTempPath(), "testfile.dat"); + var mockResponse = new HttpResponseMessage(System.Net.HttpStatusCode.OK) + { + Content = new StreamContent(new MemoryStream(new byte[1024])) + }; + + var mockLogger = new Mock>(); + var helper = new RestClientHttpHelper(mockLogger.Object); + + // Act + var result = helper.DownloadFile("https://example.com/file", testPath); + // Assert + if (result&& File.Exists(testPath)) + { + Assert.Pass(); + } + + // Cleanup + File.Delete(testPath); + } + + [Test] + public void DownloadFile_ShouldHandleIOException() + { + // Arrange + var invalidPath = "D:\\Temp\\123\\123.jpg"; + var mockLogger = new Mock>(); + var helper = new RestClientHttpHelper(mockLogger.Object); + helper.DownloadFile("http://gips3.baidu.com/it/u=1821127123,1149655687&fm=3028&app=3028&f=JPEG&fmt=auto?w=720&h=1280", invalidPath); + + if (File.Exists(invalidPath)) + { + Assert.Pass("ÏÂÔسɹ¦!"); + } + + Assert.Pass("δ֪Òì³£!"); + } + } +} \ No newline at end of file