foreign/csharp/Iggy_SDK.Tests.Integration/PartitionsTests.cs

// // Licensed to the Apache Software Foundation (ASF) under one
// // or more contributor license agreements.  See the NOTICE file
// // distributed with this work for additional information
// // regarding copyright ownership.  The ASF licenses this file
// // to you 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.

using Apache.Iggy.Enums;
using Apache.Iggy.Exceptions;
using Apache.Iggy.Tests.Integrations.Fixtures;
using Apache.Iggy.Tests.Integrations.Helpers;
using Shouldly;

namespace Apache.Iggy.Tests.Integrations;

public class PartitionsTests
{
    [ClassDataSource<PartitionsFixture>(Shared = SharedType.PerClass)]
    public required PartitionsFixture Fixture { get; init; }

    [Test]
    [MethodDataSource<IggyServerFixture>(nameof(IggyServerFixture.ProtocolData))]
    public async Task CreatePartition_HappyPath_Should_CreatePartition_Successfully(Protocol protocol)
    {
        await Should.NotThrowAsync(() =>
            Fixture.Clients[protocol]
                .CreatePartitionsAsync(Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
                    Identifier.String(Fixture.TopicRequest.Name), 3));

        var response = await Fixture.Clients[protocol].GetTopicByIdAsync(
            Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
            Identifier.String(Fixture.TopicRequest.Name));
        response.ShouldNotBeNull();
        response.PartitionsCount.ShouldBe(4u);
    }

    [Test]
    [DependsOn(nameof(CreatePartition_HappyPath_Should_CreatePartition_Successfully))]
    [MethodDataSource<IggyServerFixture>(nameof(IggyServerFixture.ProtocolData))]
    public async Task DeletePartition_Should_DeletePartition_Successfully(Protocol protocol)
    {
        await Should.NotThrowAsync(() =>
            Fixture.Clients[protocol].DeletePartitionsAsync(
                Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
                Identifier.String(Fixture.TopicRequest.Name), 1));

        var response = await Fixture.Clients[protocol].GetTopicByIdAsync(
            Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
            Identifier.String(Fixture.TopicRequest.Name));
        response.ShouldNotBeNull();
        response.PartitionsCount.ShouldBe(3u);
    }

    [Test]
    [DependsOn(nameof(DeletePartition_Should_DeletePartition_Successfully))]
    [MethodDataSource<IggyServerFixture>(nameof(IggyServerFixture.ProtocolData))]
    public async Task DeletePartition_Should_Throw_WhenTopic_DoesNotExist(Protocol protocol)
    {
        await Fixture.Clients[protocol].DeleteTopicAsync(Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
            Identifier.String(Fixture.TopicRequest.Name));
        await Should.ThrowAsync<InvalidResponseException>(() =>
            Fixture.Clients[protocol].DeletePartitionsAsync(
                Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
                Identifier.String(Fixture.TopicRequest.Name), 1));
    }

    [Test]
    [DependsOn(nameof(DeletePartition_Should_Throw_WhenTopic_DoesNotExist))]
    [MethodDataSource<IggyServerFixture>(nameof(IggyServerFixture.ProtocolData))]
    public async Task DeletePartition_Should_Throw_WhenStream_DoesNotExist(Protocol protocol)
    {
        await Fixture.Clients[protocol]
            .DeleteStreamAsync(Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)));
        await Should.ThrowAsync<InvalidResponseException>(() =>
            Fixture.Clients[protocol].DeletePartitionsAsync(
                Identifier.String(Fixture.StreamId.GetWithProtocol(protocol)),
                Identifier.String(Fixture.TopicRequest.Name), 1));
    }
}