| /* |
| * 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. |
| */ |
| |
| namespace DotPulsar.IntegrationTests.Services; |
| |
| using Abstraction; |
| using System; |
| using System.Net.Http; |
| using System.Text; |
| using System.Threading; |
| using System.Threading.Tasks; |
| using Xunit.Abstractions; |
| using Xunit.Sdk; |
| |
| public abstract class PulsarServiceBase : IPulsarService |
| { |
| protected readonly IMessageSink MessageSink; |
| private readonly CancellationTokenSource _cts; |
| private readonly HttpClient _adminClient; |
| |
| protected PulsarServiceBase(IMessageSink messageSink) |
| { |
| MessageSink = messageSink; |
| _cts = new CancellationTokenSource(); |
| _adminClient = new HttpClient(); |
| } |
| |
| public abstract Task InitializeAsync(); |
| |
| public async Task DisposeAsync() |
| { |
| _adminClient.Dispose(); |
| _cts.Dispose(); |
| |
| try |
| { |
| await OnDispose(); |
| } |
| catch (Exception e) |
| { |
| MessageSink.OnMessage(new DiagnosticMessage("Error disposing: {0}", e)); |
| } |
| } |
| |
| protected virtual Task OnDispose() |
| => Task.CompletedTask; |
| |
| public abstract Uri GetBrokerUri(); |
| |
| public abstract Uri GetWebServiceUri(); |
| |
| public async Task<HttpResponseMessage?> CreatePartitionedTopic(string restTopic, int numPartitions) |
| { |
| var content = new StringContent(numPartitions.ToString(), Encoding.UTF8, "application/json"); |
| return await _adminClient.PutAsync($"{GetWebServiceUri()}admin/v2/{restTopic}/partitions", content, _cts.Token).ConfigureAwait(false); |
| } |
| } |