// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package azure.core.trait;

import azure.core.trait.models.UserActionParam;
import com.azure.core.http.HttpHeaderName;
import com.azure.core.http.RequestConditions;
import com.azure.core.http.rest.Response;
import com.azure.core.util.BinaryData;
import java.time.OffsetDateTime;
import java.time.ZoneOffset;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

public class TraitsTests {

    private TraitsClient client = new TraitsClientBuilder().buildClient();

    @Test
    public void testGet() {
        OffsetDateTime unmodifiedSince = OffsetDateTime.of(2022, 8, 26, 14, 38, 0, 0, ZoneOffset.UTC);
        OffsetDateTime modifiedSince = OffsetDateTime.of(2021, 8, 26, 14, 38, 0, 0, ZoneOffset.UTC);

        client.smokeTest(1, "123",
            new RequestConditions().setIfMatch("\"valid\"")
                .setIfNoneMatch("\"invalid\"")
                .setIfModifiedSince(modifiedSince)
                .setIfUnmodifiedSince(unmodifiedSince));
    }

    @Test
    public void testPost() {
        Response<BinaryData> response
            = client.repeatableActionWithResponse(1, BinaryData.fromObject(new UserActionParam("test")), null);
        Assertions.assertEquals("accepted",
            response.getHeaders().getValue(HttpHeaderName.fromString("repeatability-result")));
    }
}
