/*
 * Copyright 2018 - Present Okta, Inc.
 *
 * 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.
 */

package unit

import (
	"context"
	"testing"

	"github.com/jarcoal/httpmock"
	"github.com/okta/okta-sdk-golang/v2/okta"
	"github.com/okta/okta-sdk-golang/v2/tests"
	"github.com/stretchr/testify/require"
)

func Test_429_Will_Automatically_Retry(t *testing.T) {
	httpmock.Activate()
	defer httpmock.DeactivateAndReset()

	ctx, client, err := tests.NewClient(context.TODO(), okta.WithCache(false))
	require.NoError(t, err, "failed to create client")

	httpmock.RegisterResponder("GET", "/api/v1/users",
		tests.MockResponse(
			tests.Mock429Response(),
			tests.MockValidResponse(),
		),
	)

	_, resp, err := client.User.ListUsers(ctx, nil)
	require.Nil(t, err, "Error should have been nil")
	require.NotNil(t, resp, "Response was nil")

	httpmock.GetTotalCallCount()
	info := httpmock.GetCallCountInfo()
	require.Equal(t, 2, info["GET /api/v1/users"], "did not make exactly 2 call to /api/v1/users")
}

func Test_Will_Stop_Retrying_Based_On_Max_Retry_Configuration(t *testing.T) {
	httpmock.Activate()
	defer httpmock.DeactivateAndReset()

	ctx, client, err := tests.NewClient(context.TODO(), okta.WithRequestTimeout(0), okta.WithRateLimitMaxRetries(1))
	require.NoError(t, err)

	httpmock.RegisterResponder("GET", "/api/v1/users",
		tests.MockResponse(
			tests.Mock429Response(),
			tests.Mock429Response(),
		),
	)

	_, _, err = client.User.ListUsers(ctx, nil)
	require.NotNil(t, err, "error was nil, but should have told the user they reached their max retry limit")

	httpmock.GetTotalCallCount()
	info := httpmock.GetCallCountInfo()

	require.False(t, info["GET /api/v1/users"] < 2, "should have done at least one retry call to /api/v1/users")
	require.Equal(t, 2, info["GET /api/v1/users"], "should have thrown error after first retry to /api/v1/users")
}

func Test_Will_Handle_Backoff_Strategy_For_429(t *testing.T) {
	httpmock.Activate()
	defer httpmock.DeactivateAndReset()

	ctx, client, err := tests.NewClient(context.TODO(), okta.WithRequestTimeout(1), okta.WithRateLimitMaxRetries(3))
	require.NoError(t, err, "failed to create client")

	httpmock.RegisterResponder("GET", "/api/v1/users",
		tests.MockResponse(
			tests.Mock429Response(),
			tests.Mock429Response(),
			tests.Mock429Response(),
			tests.MockValidResponse(),
		),
	)

	_, _, err = client.User.ListUsers(ctx, nil)
	require.NotNil(t, err, "error was nil, but should have told the user they reached their max retry limit")

	httpmock.GetTotalCallCount()
	info := httpmock.GetCallCountInfo()

	require.Equal(t, 1, info["GET /api/v1/users"], "should have thrown error before first retry to /api/v1/users due to request timeout")
}

func Test_a_429_with_x_reset_header_throws_error(t *testing.T) {
	httpmock.Activate()
	defer httpmock.DeactivateAndReset()

	ctx, client, err := tests.NewClient(context.TODO())
	require.NoError(t, err, "failed to create client")

	httpmock.RegisterResponder("GET", "/api/v1/users",
		tests.MockResponse(
			tests.Mock429ResponseNoResetHeader(),
		),
	)

	_, _, err = client.User.ListUsers(ctx, nil)

	require.NotNil(t, err, "error should not be nil. It should let user know the reset header is required")
}

func Test_a_429_with_no_date_header_throws_error(t *testing.T) {
	httpmock.Activate()
	defer httpmock.DeactivateAndReset()

	ctx, client, err := tests.NewClient(context.TODO())
	require.NoError(t, err, "failed to create client")

	httpmock.RegisterResponder("GET", "/api/v1/users",
		tests.MockResponse(
			tests.Mock429ResponseNoDateHeader(),
		),
	)

	_, _, err = client.User.ListUsers(ctx, nil)

	require.NotNil(t, err, "error should not be nil. It should let user know the date header is required")
}

func Test_gets_the_correct_backoff_time(t *testing.T) {
	backoff, err := okta.Get429BackoffTime(tests.Mock429Response())
	require.NoError(t, err)

	require.Equal(t, int64(2), backoff, "backoff time should have only been 1 second")
}