123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253 |
- package server
- import (
- "cfTest/cloudflareApi/zone"
- "context"
- "github.com/samber/lo"
- "log"
- "testing"
- )
- //func TestCreateRuleSet(t *testing.T) {
- // type args struct {
- // zoneID string
- // rulesetPhase string
- // ruleset cloudflare.Ruleset
- // }
- //
- // var (
- // True = true
- // CONST604800 = uint(604800)
- // )
- //
- // tests := []struct {
- // name string
- // args args
- // want interface{}
- // wantErr bool
- // }{
- // {
- // name: "Test1",
- // args: args{
- // zoneID: "e29551894461bf16e43a113e5ab09ed3",
- // rulesetPhase: "http_request_cache_settings",
- // ruleset: cloudflare.Ruleset{
- // Rules: []cloudflare.RulesetRule{
- // {
- // Expression: "(starts_with(http.request.uri.path, \"/\"))",
- // Enabled: true,
- // Description: "test1",
- // Action: "set_cache_settings",
- // ActionParameters: &cloudflare.RulesetRuleActionParameters{
- // Cache: &True,
- // CacheKey: &cloudflare.RulesetRuleActionParametersCacheKey{
- // CustomKey: &cloudflare.RulesetRuleActionParametersCustomKey{
- // Query: &cloudflare.RulesetRuleActionParametersCustomKeyQuery{
- // Exclude: &cloudflare.RulesetRuleActionParametersCustomKeyList{
- // List: []string{
- // "abc",
- // },
- // All: false,
- // },
- // },
- // },
- // },
- // EdgeTTL: &cloudflare.RulesetRuleActionParametersEdgeTTL{
- // Mode: "override_origin",
- // Default: &CONST604800,
- // },
- // },
- // },
- // },
- // },
- // },
- // want: nil,
- // wantErr: false,
- // },
- // }
- // for _, tt := range tests {
- // t.Run(tt.name, func(t *testing.T) {
- // got, err := UpdateRuleSet("", "", tt.args.zoneID, tt.args.rulesetPhase, tt.args.ruleset)
- // if (err != nil) != tt.wantErr {
- // t.Errorf("UpdateRuleSet() error = %v, wantErr %v", err, tt.wantErr)
- // return
- // }
- // if !reflect.DeepEqual(got, tt.want) {
- // t.Errorf("UpdateRuleSet() got = %v, want %v", got, tt.want)
- // }
- // })
- // }
- //}
- //func newUint32(a uint32) *uint32 {
- // return &a
- //}
- //func newStr(a string) *string {
- // return &a
- //}
- var (
- apiKey = "2cc87f6213a4c40e5a90c8dced3c9c577eda5"
- apiEmail = "ll@centcloud.net"
- )
- //func TestUpdatePageRule(t *testing.T) {
- //
- // type args struct {
- // apiKey string
- // apiEmail string
- // zoneID string
- // rule *zone.Rule
- // }
- // tests := []struct {
- // name string
- // args args
- // want *cloudflare.PageRule
- // wantErr bool
- // }{
- // {
- // name: "测试1",
- // args: args{
- // apiKey: apiKey,
- // apiEmail: apiEmail,
- // zoneID: "e29551894461bf16e43a113e5ab09ed3",
- // rule: &zone.Rule{
- // Expression: "www.1011test5.com.cdn.sensordb.site/*\n",
- // Ttl: newUint32(18000),
- // QueryArgsType: newStr("ExcludeAll"),
- // QueryArgs: []string{"qwe12", "78u9hi"},
- // OriginHost: newStr("www.baidu.com"),
- // OriginPath: nil,
- // OriginProtocol: newUint32(80),
- // RequestHeader: nil,
- // ResponseHeader: nil,
- // },
- // },
- // want: nil,
- // wantErr: false,
- // },
- // // TODO: Add test cases.
- // }
- // for _, tt := range tests {
- // t.Run(tt.name, func(t *testing.T) {
- // got, err := UpdatePageRule(tt.args.apiKey, tt.args.apiEmail, tt.args.zoneID, tt.args.rule)
- // if (err != nil) != tt.wantErr {
- // t.Errorf("UpdatePageRule() error = %v, wantErr %v", err, tt.wantErr)
- // return
- // }
- // print("%v", got)
- // //if !reflect.DeepEqual(got, tt.want) {
- // // t.Errorf("UpdatePageRule() got = %v, want %v", got, tt.want)
- // //}
- // })
- // }
- //}
- func TestZoneServer_GetRuleSet(t *testing.T) {
- type fields struct {
- UnimplementedZoneCloudflareServer zone.UnimplementedZoneCloudflareServer
- }
- type args struct {
- ctx context.Context
- requestCloudflare *zone.GetRuleSetRequestCloudflare
- }
- tests := []struct {
- name string
- fields fields
- args args
- want *zone.RuleList
- wantErr bool
- }{
- {
- name: "测试1",
- fields: fields{
- UnimplementedZoneCloudflareServer: zone.UnimplementedZoneCloudflareServer{},
- },
- args: args{
- ctx: context.Background(),
- requestCloudflare: &zone.GetRuleSetRequestCloudflare{
- ApiKey: apiKey,
- ApiEmail: apiEmail,
- ZoneId: "e29551894461bf16e43a113e5ab09ed3",
- Phase: zone.Phase_http_request_transform,
- },
- },
- want: nil,
- wantErr: false,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- s := &ZoneServer{
- UnimplementedZoneCloudflareServer: tt.fields.UnimplementedZoneCloudflareServer,
- }
- got, err := s.GetRuleSet(tt.args.ctx, tt.args.requestCloudflare)
- if (err != nil) != tt.wantErr {
- t.Errorf("GetRuleSet() error = %v, wantErr %v", err, tt.wantErr)
- return
- }
- log.Print(got)
- //if !reflect.DeepEqual(got, tt.want) {
- // t.Errorf("GetRuleSet() got = %v, want %v", got, tt.want)
- //}
- })
- }
- }
- func TestZoneServer_UpdateRuleSet(t *testing.T) {
- type fields struct {
- UnimplementedZoneCloudflareServer zone.UnimplementedZoneCloudflareServer
- }
- type args struct {
- ctx context.Context
- requestCloudflare *zone.UpdateRuleSetRequestCloudflare
- }
- tests := []struct {
- name string
- fields fields
- args args
- want *zone.RuleList
- wantErr bool
- }{
- {
- name: "测试1",
- fields: fields{
- UnimplementedZoneCloudflareServer: zone.UnimplementedZoneCloudflareServer{},
- },
- args: args{
- ctx: nil,
- requestCloudflare: &zone.UpdateRuleSetRequestCloudflare{
- ApiKey: apiKey,
- ApiEmail: apiEmail,
- ZoneId: "e29551894461bf16e43a113e5ab09ed3",
- Phase: zone.Phase_http_request_transform,
- Rules: []*zone.Rule{
- {
- Expression: "(http.request.uri.path contains \"www.1011test5.com.cdn.sensordb.site/\")",
- OriginPath: lo.ToPtr[string]("www.baidu.com"),
- RequestHeader: nil,
- ResponseHeader: nil,
- },
- },
- },
- },
- want: nil,
- wantErr: false,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- s := &ZoneServer{
- UnimplementedZoneCloudflareServer: tt.fields.UnimplementedZoneCloudflareServer,
- }
- got, err := s.UpdateRuleSet(tt.args.ctx, tt.args.requestCloudflare)
- if (err != nil) != tt.wantErr {
- t.Errorf("UpdateRuleSet() error = %v, wantErr %v", err, tt.wantErr)
- return
- }
- log.Print(got)
- //if !reflect.DeepEqual(got, tt.want) {
- // t.Errorf("UpdateRuleSet() got = %v, want %v", got, tt.want)
- //}
- })
- }
- }
|