// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

package zero_trust_test

import (
	"context"
	"errors"
	"os"
	"testing"
	"time"

	"github.com/cloudflare/cloudflare-go/v6"
	"github.com/cloudflare/cloudflare-go/v6/internal/testutil"
	"github.com/cloudflare/cloudflare-go/v6/option"
	"github.com/cloudflare/cloudflare-go/v6/zero_trust"
)

func TestGatewayRuleNewWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := cloudflare.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("144c9defac04969c7bfad8efaa8ea194"),
		option.WithAPIEmail("user@example.com"),
	)
	_, err := client.ZeroTrust.Gateway.Rules.New(context.TODO(), zero_trust.GatewayRuleNewParams{
		AccountID:     cloudflare.F("699d98642c564d2e855e9661899b7252"),
		Action:        cloudflare.F(zero_trust.GatewayRuleNewParamsActionAllow),
		Name:          cloudflare.F("block bad websites"),
		Description:   cloudflare.F("Block bad websites based on their host name."),
		DevicePosture: cloudflare.F(`any(device_posture.checks.passed[*] in {"1308749e-fcfb-4ebc-b051-fe022b632644"})`),
		Enabled:       cloudflare.F(true),
		Expiration: cloudflare.F(zero_trust.GatewayRuleNewParamsExpiration{
			ExpiresAt: cloudflare.F(time.Now()),
			Duration:  cloudflare.F(int64(10)),
		}),
		Filters:    cloudflare.F([]zero_trust.GatewayFilter{zero_trust.GatewayFilterHTTP}),
		Identity:   cloudflare.F(`any(identity.groups.name[*] in {"finance"})`),
		Precedence: cloudflare.F(int64(0)),
		RuleSettings: cloudflare.F(zero_trust.RuleSettingParam{
			AddHeaders: cloudflare.F(map[string][]string{
				"My-Next-Header":       {"foo", "bar"},
				"X-Custom-Header-Name": {"somecustomvalue"},
			}),
			AllowChildBypass: cloudflare.F(false),
			AuditSSH: cloudflare.F(zero_trust.RuleSettingAuditSSHParam{
				CommandLogging: cloudflare.F(false),
			}),
			BISOAdminControls: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsParam{
				Copy:     cloudflare.F(zero_trust.RuleSettingBISOAdminControlsCopyRemoteOnly),
				DCP:      cloudflare.F(true),
				DD:       cloudflare.F(true),
				DK:       cloudflare.F(true),
				Download: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsDownloadEnabled),
				DP:       cloudflare.F(false),
				DU:       cloudflare.F(true),
				Keyboard: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsKeyboardEnabled),
				Paste:    cloudflare.F(zero_trust.RuleSettingBISOAdminControlsPasteEnabled),
				Printing: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsPrintingEnabled),
				Upload:   cloudflare.F(zero_trust.RuleSettingBISOAdminControlsUploadEnabled),
				Version:  cloudflare.F(zero_trust.RuleSettingBISOAdminControlsVersionV1),
			}),
			BlockPage: cloudflare.F(zero_trust.RuleSettingBlockPageParam{
				TargetURI:      cloudflare.F("https://example.com"),
				IncludeContext: cloudflare.F(true),
			}),
			BlockPageEnabled: cloudflare.F(true),
			BlockReason:      cloudflare.F("This website is a security risk"),
			BypassParentRule: cloudflare.F(false),
			CheckSession: cloudflare.F(zero_trust.RuleSettingCheckSessionParam{
				Duration: cloudflare.F("300s"),
				Enforce:  cloudflare.F(true),
			}),
			DNSResolvers: cloudflare.F(zero_trust.RuleSettingDNSResolversParam{
				IPV4: cloudflare.F([]zero_trust.DNSResolverSettingsV4Param{{
					IP:                         cloudflare.F("2.2.2.2"),
					Port:                       cloudflare.F(int64(5053)),
					RouteThroughPrivateNetwork: cloudflare.F(true),
					VnetID:                     cloudflare.F("f174e90a-fafe-4643-bbbc-4a0ed4fc8415"),
				}}),
				IPV6: cloudflare.F([]zero_trust.DNSResolverSettingsV6Param{{
					IP:                         cloudflare.F("2001:DB8::"),
					Port:                       cloudflare.F(int64(5053)),
					RouteThroughPrivateNetwork: cloudflare.F(true),
					VnetID:                     cloudflare.F("f174e90a-fafe-4643-bbbc-4a0ed4fc8415"),
				}}),
			}),
			Egress: cloudflare.F(zero_trust.RuleSettingEgressParam{
				IPV4:         cloudflare.F("192.0.2.2"),
				IPV4Fallback: cloudflare.F("192.0.2.3"),
				IPV6:         cloudflare.F("2001:DB8::/64"),
			}),
			IgnoreCNAMECategoryMatches:      cloudflare.F(true),
			InsecureDisableDNSSECValidation: cloudflare.F(false),
			IPCategories:                    cloudflare.F(true),
			IPIndicatorFeeds:                cloudflare.F(true),
			L4override: cloudflare.F(zero_trust.RuleSettingL4overrideParam{
				IP:   cloudflare.F("1.1.1.1"),
				Port: cloudflare.F(int64(0)),
			}),
			NotificationSettings: cloudflare.F(zero_trust.RuleSettingNotificationSettingsParam{
				Enabled:        cloudflare.F(true),
				IncludeContext: cloudflare.F(true),
				Msg:            cloudflare.F("msg"),
				SupportURL:     cloudflare.F("support_url"),
			}),
			OverrideHost: cloudflare.F("example.com"),
			OverrideIPs:  cloudflare.F([]string{"1.1.1.1", "2.2.2.2"}),
			PayloadLog: cloudflare.F(zero_trust.RuleSettingPayloadLogParam{
				Enabled: cloudflare.F(true),
			}),
			Quarantine: cloudflare.F(zero_trust.RuleSettingQuarantineParam{
				FileTypes: cloudflare.F([]zero_trust.RuleSettingQuarantineFileType{zero_trust.RuleSettingQuarantineFileTypeExe}),
			}),
			Redirect: cloudflare.F(zero_trust.RuleSettingRedirectParam{
				TargetURI:            cloudflare.F("https://example.com"),
				IncludeContext:       cloudflare.F(true),
				PreservePathAndQuery: cloudflare.F(true),
			}),
			ResolveDNSInternally: cloudflare.F(zero_trust.RuleSettingResolveDNSInternallyParam{
				Fallback: cloudflare.F(zero_trust.RuleSettingResolveDNSInternallyFallbackNone),
				ViewID:   cloudflare.F("view_id"),
			}),
			ResolveDNSThroughCloudflare: cloudflare.F(true),
			UntrustedCERT: cloudflare.F(zero_trust.RuleSettingUntrustedCERTParam{
				Action: cloudflare.F(zero_trust.RuleSettingUntrustedCERTActionError),
			}),
		}),
		Schedule: cloudflare.F(zero_trust.ScheduleParam{
			Fri:      cloudflare.F("08:00-12:30,13:30-17:00"),
			Mon:      cloudflare.F("08:00-12:30,13:30-17:00"),
			Sat:      cloudflare.F("08:00-12:30,13:30-17:00"),
			Sun:      cloudflare.F("08:00-12:30,13:30-17:00"),
			Thu:      cloudflare.F("08:00-12:30,13:30-17:00"),
			TimeZone: cloudflare.F("America/New York"),
			Tue:      cloudflare.F("08:00-12:30,13:30-17:00"),
			Wed:      cloudflare.F("08:00-12:30,13:30-17:00"),
		}),
		Traffic: cloudflare.F(`http.request.uri matches ".*a/partial/uri.*" and http.request.host in $01302951-49f9-47c9-a400-0297e60b6a10`),
	})
	if err != nil {
		var apierr *cloudflare.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestGatewayRuleUpdateWithOptionalParams(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := cloudflare.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("144c9defac04969c7bfad8efaa8ea194"),
		option.WithAPIEmail("user@example.com"),
	)
	_, err := client.ZeroTrust.Gateway.Rules.Update(
		context.TODO(),
		"f174e90a-fafe-4643-bbbc-4a0ed4fc8415",
		zero_trust.GatewayRuleUpdateParams{
			AccountID:     cloudflare.F("699d98642c564d2e855e9661899b7252"),
			Action:        cloudflare.F(zero_trust.GatewayRuleUpdateParamsActionAllow),
			Name:          cloudflare.F("block bad websites"),
			Description:   cloudflare.F("Block bad websites based on their host name."),
			DevicePosture: cloudflare.F(`any(device_posture.checks.passed[*] in {"1308749e-fcfb-4ebc-b051-fe022b632644"})`),
			Enabled:       cloudflare.F(true),
			Expiration: cloudflare.F(zero_trust.GatewayRuleUpdateParamsExpiration{
				ExpiresAt: cloudflare.F(time.Now()),
				Duration:  cloudflare.F(int64(10)),
			}),
			Filters:    cloudflare.F([]zero_trust.GatewayFilter{zero_trust.GatewayFilterHTTP}),
			Identity:   cloudflare.F(`any(identity.groups.name[*] in {"finance"})`),
			Precedence: cloudflare.F(int64(0)),
			RuleSettings: cloudflare.F(zero_trust.RuleSettingParam{
				AddHeaders: cloudflare.F(map[string][]string{
					"My-Next-Header":       {"foo", "bar"},
					"X-Custom-Header-Name": {"somecustomvalue"},
				}),
				AllowChildBypass: cloudflare.F(false),
				AuditSSH: cloudflare.F(zero_trust.RuleSettingAuditSSHParam{
					CommandLogging: cloudflare.F(false),
				}),
				BISOAdminControls: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsParam{
					Copy:     cloudflare.F(zero_trust.RuleSettingBISOAdminControlsCopyRemoteOnly),
					DCP:      cloudflare.F(true),
					DD:       cloudflare.F(true),
					DK:       cloudflare.F(true),
					Download: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsDownloadEnabled),
					DP:       cloudflare.F(false),
					DU:       cloudflare.F(true),
					Keyboard: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsKeyboardEnabled),
					Paste:    cloudflare.F(zero_trust.RuleSettingBISOAdminControlsPasteEnabled),
					Printing: cloudflare.F(zero_trust.RuleSettingBISOAdminControlsPrintingEnabled),
					Upload:   cloudflare.F(zero_trust.RuleSettingBISOAdminControlsUploadEnabled),
					Version:  cloudflare.F(zero_trust.RuleSettingBISOAdminControlsVersionV1),
				}),
				BlockPage: cloudflare.F(zero_trust.RuleSettingBlockPageParam{
					TargetURI:      cloudflare.F("https://example.com"),
					IncludeContext: cloudflare.F(true),
				}),
				BlockPageEnabled: cloudflare.F(true),
				BlockReason:      cloudflare.F("This website is a security risk"),
				BypassParentRule: cloudflare.F(false),
				CheckSession: cloudflare.F(zero_trust.RuleSettingCheckSessionParam{
					Duration: cloudflare.F("300s"),
					Enforce:  cloudflare.F(true),
				}),
				DNSResolvers: cloudflare.F(zero_trust.RuleSettingDNSResolversParam{
					IPV4: cloudflare.F([]zero_trust.DNSResolverSettingsV4Param{{
						IP:                         cloudflare.F("2.2.2.2"),
						Port:                       cloudflare.F(int64(5053)),
						RouteThroughPrivateNetwork: cloudflare.F(true),
						VnetID:                     cloudflare.F("f174e90a-fafe-4643-bbbc-4a0ed4fc8415"),
					}}),
					IPV6: cloudflare.F([]zero_trust.DNSResolverSettingsV6Param{{
						IP:                         cloudflare.F("2001:DB8::"),
						Port:                       cloudflare.F(int64(5053)),
						RouteThroughPrivateNetwork: cloudflare.F(true),
						VnetID:                     cloudflare.F("f174e90a-fafe-4643-bbbc-4a0ed4fc8415"),
					}}),
				}),
				Egress: cloudflare.F(zero_trust.RuleSettingEgressParam{
					IPV4:         cloudflare.F("192.0.2.2"),
					IPV4Fallback: cloudflare.F("192.0.2.3"),
					IPV6:         cloudflare.F("2001:DB8::/64"),
				}),
				IgnoreCNAMECategoryMatches:      cloudflare.F(true),
				InsecureDisableDNSSECValidation: cloudflare.F(false),
				IPCategories:                    cloudflare.F(true),
				IPIndicatorFeeds:                cloudflare.F(true),
				L4override: cloudflare.F(zero_trust.RuleSettingL4overrideParam{
					IP:   cloudflare.F("1.1.1.1"),
					Port: cloudflare.F(int64(0)),
				}),
				NotificationSettings: cloudflare.F(zero_trust.RuleSettingNotificationSettingsParam{
					Enabled:        cloudflare.F(true),
					IncludeContext: cloudflare.F(true),
					Msg:            cloudflare.F("msg"),
					SupportURL:     cloudflare.F("support_url"),
				}),
				OverrideHost: cloudflare.F("example.com"),
				OverrideIPs:  cloudflare.F([]string{"1.1.1.1", "2.2.2.2"}),
				PayloadLog: cloudflare.F(zero_trust.RuleSettingPayloadLogParam{
					Enabled: cloudflare.F(true),
				}),
				Quarantine: cloudflare.F(zero_trust.RuleSettingQuarantineParam{
					FileTypes: cloudflare.F([]zero_trust.RuleSettingQuarantineFileType{zero_trust.RuleSettingQuarantineFileTypeExe}),
				}),
				Redirect: cloudflare.F(zero_trust.RuleSettingRedirectParam{
					TargetURI:            cloudflare.F("https://example.com"),
					IncludeContext:       cloudflare.F(true),
					PreservePathAndQuery: cloudflare.F(true),
				}),
				ResolveDNSInternally: cloudflare.F(zero_trust.RuleSettingResolveDNSInternallyParam{
					Fallback: cloudflare.F(zero_trust.RuleSettingResolveDNSInternallyFallbackNone),
					ViewID:   cloudflare.F("view_id"),
				}),
				ResolveDNSThroughCloudflare: cloudflare.F(true),
				UntrustedCERT: cloudflare.F(zero_trust.RuleSettingUntrustedCERTParam{
					Action: cloudflare.F(zero_trust.RuleSettingUntrustedCERTActionError),
				}),
			}),
			Schedule: cloudflare.F(zero_trust.ScheduleParam{
				Fri:      cloudflare.F("08:00-12:30,13:30-17:00"),
				Mon:      cloudflare.F("08:00-12:30,13:30-17:00"),
				Sat:      cloudflare.F("08:00-12:30,13:30-17:00"),
				Sun:      cloudflare.F("08:00-12:30,13:30-17:00"),
				Thu:      cloudflare.F("08:00-12:30,13:30-17:00"),
				TimeZone: cloudflare.F("America/New York"),
				Tue:      cloudflare.F("08:00-12:30,13:30-17:00"),
				Wed:      cloudflare.F("08:00-12:30,13:30-17:00"),
			}),
			Traffic: cloudflare.F(`http.request.uri matches ".*a/partial/uri.*" and http.request.host in $01302951-49f9-47c9-a400-0297e60b6a10`),
		},
	)
	if err != nil {
		var apierr *cloudflare.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestGatewayRuleList(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := cloudflare.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("144c9defac04969c7bfad8efaa8ea194"),
		option.WithAPIEmail("user@example.com"),
	)
	_, err := client.ZeroTrust.Gateway.Rules.List(context.TODO(), zero_trust.GatewayRuleListParams{
		AccountID: cloudflare.F("699d98642c564d2e855e9661899b7252"),
	})
	if err != nil {
		var apierr *cloudflare.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestGatewayRuleDelete(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := cloudflare.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("144c9defac04969c7bfad8efaa8ea194"),
		option.WithAPIEmail("user@example.com"),
	)
	_, err := client.ZeroTrust.Gateway.Rules.Delete(
		context.TODO(),
		"f174e90a-fafe-4643-bbbc-4a0ed4fc8415",
		zero_trust.GatewayRuleDeleteParams{
			AccountID: cloudflare.F("699d98642c564d2e855e9661899b7252"),
		},
	)
	if err != nil {
		var apierr *cloudflare.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestGatewayRuleGet(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := cloudflare.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("144c9defac04969c7bfad8efaa8ea194"),
		option.WithAPIEmail("user@example.com"),
	)
	_, err := client.ZeroTrust.Gateway.Rules.Get(
		context.TODO(),
		"f174e90a-fafe-4643-bbbc-4a0ed4fc8415",
		zero_trust.GatewayRuleGetParams{
			AccountID: cloudflare.F("699d98642c564d2e855e9661899b7252"),
		},
	)
	if err != nil {
		var apierr *cloudflare.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}

func TestGatewayRuleResetExpiration(t *testing.T) {
	baseURL := "http://localhost:4010"
	if envURL, ok := os.LookupEnv("TEST_API_BASE_URL"); ok {
		baseURL = envURL
	}
	if !testutil.CheckTestServer(t, baseURL) {
		return
	}
	client := cloudflare.NewClient(
		option.WithBaseURL(baseURL),
		option.WithAPIKey("144c9defac04969c7bfad8efaa8ea194"),
		option.WithAPIEmail("user@example.com"),
	)
	_, err := client.ZeroTrust.Gateway.Rules.ResetExpiration(
		context.TODO(),
		"f174e90a-fafe-4643-bbbc-4a0ed4fc8415",
		zero_trust.GatewayRuleResetExpirationParams{
			AccountID: cloudflare.F("699d98642c564d2e855e9661899b7252"),
		},
	)
	if err != nil {
		var apierr *cloudflare.Error
		if errors.As(err, &apierr) {
			t.Log(string(apierr.DumpRequest(true)))
		}
		t.Fatalf("err should be nil: %s", err.Error())
	}
}
