github.com/ethereum-optimism/optimism/l2geth@v0.0.0-20230612200230-50b04ade19e3/cmd/devp2p/dns_route53_test.go (about)

     1  // Copyright 2020 The go-ethereum Authors
     2  // This file is part of go-ethereum.
     3  //
     4  // go-ethereum is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // go-ethereum is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU General Public License
    15  // along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package main
    18  
    19  import (
    20  	"reflect"
    21  	"testing"
    22  
    23  	"github.com/aws/aws-sdk-go/service/route53"
    24  )
    25  
    26  // This test checks that computeChanges/splitChanges create DNS changes in
    27  // leaf-added -> root-changed -> leaf-deleted order.
    28  func TestRoute53ChangeSort(t *testing.T) {
    29  	testTree0 := map[string]recordSet{
    30  		"2kfjogvxdqtxxugbh7gs7naaai.n": {ttl: 3333, values: []string{
    31  			`"enr:-HW4QO1ml1DdXLeZLsUxewnthhUy8eROqkDyoMTyavfks9JlYQIlMFEUoM78PovJDPQrAkrb3LRJ-"`,
    32  			`"vtrymDguKCOIAWAgmlkgnY0iXNlY3AyNTZrMaEDffaGfJzgGhUif1JqFruZlYmA31HzathLSWxfbq_QoQ4"`,
    33  		}},
    34  		"fdxn3sn67na5dka4j2gok7bvqi.n": {ttl: treeNodeTTL, values: []string{`"enrtree-branch:"`}},
    35  		"n":                            {ttl: rootTTL, values: []string{`"enrtree-root:v1 e=2KFJOGVXDQTXXUGBH7GS7NAAAI l=FDXN3SN67NA5DKA4J2GOK7BVQI seq=0 sig=v_-J_q_9ICQg5ztExFvLQhDBGMb0lZPJLhe3ts9LAcgqhOhtT3YFJsl8BWNDSwGtamUdR-9xl88_w-X42SVpjwE"`}},
    36  	}
    37  
    38  	testTree1 := map[string]string{
    39  		"n":                            "enrtree-root:v1 e=JWXYDBPXYWG6FX3GMDIBFA6CJ4 l=C7HRFPF3BLGF3YR4DY5KX3SMBE seq=1 sig=o908WmNp7LibOfPsr4btQwatZJ5URBr2ZAuxvK4UWHlsB9sUOTJQaGAlLPVAhM__XJesCHxLISo94z5Z2a463gA",
    40  		"C7HRFPF3BLGF3YR4DY5KX3SMBE.n": "enrtree://AM5FCQLWIZX2QFPNJAP7VUERCCRNGRHWZG3YYHIUV7BVDQ5FDPRT2@morenodes.example.org",
    41  		"JWXYDBPXYWG6FX3GMDIBFA6CJ4.n": "enrtree-branch:2XS2367YHAXJFGLZHVAWLQD4ZY,H4FHT4B454P6UXFD7JCYQ5PWDY,MHTDO6TMUBRIA2XWG5LUDACK24",
    42  		"2XS2367YHAXJFGLZHVAWLQD4ZY.n": "enr:-HW4QOFzoVLaFJnNhbgMoDXPnOvcdVuj7pDpqRvh6BRDO68aVi5ZcjB3vzQRZH2IcLBGHzo8uUN3snqmgTiE56CH3AMBgmlkgnY0iXNlY3AyNTZrMaECC2_24YYkYHEgdzxlSNKQEnHhuNAbNlMlWJxrJxbAFvA",
    43  		"H4FHT4B454P6UXFD7JCYQ5PWDY.n": "enr:-HW4QAggRauloj2SDLtIHN1XBkvhFZ1vtf1raYQp9TBW2RD5EEawDzbtSmlXUfnaHcvwOizhVYLtr7e6vw7NAf6mTuoCgmlkgnY0iXNlY3AyNTZrMaECjrXI8TLNXU0f8cthpAMxEshUyQlK-AM0PW2wfrnacNI",
    44  		"MHTDO6TMUBRIA2XWG5LUDACK24.n": "enr:-HW4QLAYqmrwllBEnzWWs7I5Ev2IAs7x_dZlbYdRdMUx5EyKHDXp7AV5CkuPGUPdvbv1_Ms1CPfhcGCvSElSosZmyoqAgmlkgnY0iXNlY3AyNTZrMaECriawHKWdDRk2xeZkrOXBQ0dfMFLHY4eENZwdufn1S1o",
    45  	}
    46  
    47  	wantChanges := []*route53.Change{
    48  		{
    49  			Action: sp("CREATE"),
    50  			ResourceRecordSet: &route53.ResourceRecordSet{
    51  				Name: sp("2xs2367yhaxjfglzhvawlqd4zy.n"),
    52  				ResourceRecords: []*route53.ResourceRecord{{
    53  					Value: sp(`"enr:-HW4QOFzoVLaFJnNhbgMoDXPnOvcdVuj7pDpqRvh6BRDO68aVi5ZcjB3vzQRZH2IcLBGHzo8uUN3snqmgTiE56CH3AMBgmlkgnY0iXNlY3AyNTZrMaECC2_24YYkYHEgdzxlSNKQEnHhuNAbNlMlWJxrJxbAFvA"`),
    54  				}},
    55  				TTL:  ip(treeNodeTTL),
    56  				Type: sp("TXT"),
    57  			},
    58  		},
    59  		{
    60  			Action: sp("CREATE"),
    61  			ResourceRecordSet: &route53.ResourceRecordSet{
    62  				Name: sp("c7hrfpf3blgf3yr4dy5kx3smbe.n"),
    63  				ResourceRecords: []*route53.ResourceRecord{{
    64  					Value: sp(`"enrtree://AM5FCQLWIZX2QFPNJAP7VUERCCRNGRHWZG3YYHIUV7BVDQ5FDPRT2@morenodes.example.org"`),
    65  				}},
    66  				TTL:  ip(treeNodeTTL),
    67  				Type: sp("TXT"),
    68  			},
    69  		},
    70  		{
    71  			Action: sp("CREATE"),
    72  			ResourceRecordSet: &route53.ResourceRecordSet{
    73  				Name: sp("h4fht4b454p6uxfd7jcyq5pwdy.n"),
    74  				ResourceRecords: []*route53.ResourceRecord{{
    75  					Value: sp(`"enr:-HW4QAggRauloj2SDLtIHN1XBkvhFZ1vtf1raYQp9TBW2RD5EEawDzbtSmlXUfnaHcvwOizhVYLtr7e6vw7NAf6mTuoCgmlkgnY0iXNlY3AyNTZrMaECjrXI8TLNXU0f8cthpAMxEshUyQlK-AM0PW2wfrnacNI"`),
    76  				}},
    77  				TTL:  ip(treeNodeTTL),
    78  				Type: sp("TXT"),
    79  			},
    80  		},
    81  		{
    82  			Action: sp("CREATE"),
    83  			ResourceRecordSet: &route53.ResourceRecordSet{
    84  				Name: sp("jwxydbpxywg6fx3gmdibfa6cj4.n"),
    85  				ResourceRecords: []*route53.ResourceRecord{{
    86  					Value: sp(`"enrtree-branch:2XS2367YHAXJFGLZHVAWLQD4ZY,H4FHT4B454P6UXFD7JCYQ5PWDY,MHTDO6TMUBRIA2XWG5LUDACK24"`),
    87  				}},
    88  				TTL:  ip(treeNodeTTL),
    89  				Type: sp("TXT"),
    90  			},
    91  		},
    92  		{
    93  			Action: sp("CREATE"),
    94  			ResourceRecordSet: &route53.ResourceRecordSet{
    95  				Name: sp("mhtdo6tmubria2xwg5ludack24.n"),
    96  				ResourceRecords: []*route53.ResourceRecord{{
    97  					Value: sp(`"enr:-HW4QLAYqmrwllBEnzWWs7I5Ev2IAs7x_dZlbYdRdMUx5EyKHDXp7AV5CkuPGUPdvbv1_Ms1CPfhcGCvSElSosZmyoqAgmlkgnY0iXNlY3AyNTZrMaECriawHKWdDRk2xeZkrOXBQ0dfMFLHY4eENZwdufn1S1o"`),
    98  				}},
    99  				TTL:  ip(treeNodeTTL),
   100  				Type: sp("TXT"),
   101  			},
   102  		},
   103  		{
   104  			Action: sp("UPSERT"),
   105  			ResourceRecordSet: &route53.ResourceRecordSet{
   106  				Name: sp("n"),
   107  				ResourceRecords: []*route53.ResourceRecord{{
   108  					Value: sp(`"enrtree-root:v1 e=JWXYDBPXYWG6FX3GMDIBFA6CJ4 l=C7HRFPF3BLGF3YR4DY5KX3SMBE seq=1 sig=o908WmNp7LibOfPsr4btQwatZJ5URBr2ZAuxvK4UWHlsB9sUOTJQaGAlLPVAhM__XJesCHxLISo94z5Z2a463gA"`),
   109  				}},
   110  				TTL:  ip(rootTTL),
   111  				Type: sp("TXT"),
   112  			},
   113  		},
   114  		{
   115  			Action: sp("DELETE"),
   116  			ResourceRecordSet: &route53.ResourceRecordSet{
   117  				Name: sp("2kfjogvxdqtxxugbh7gs7naaai.n"),
   118  				ResourceRecords: []*route53.ResourceRecord{
   119  					{Value: sp(`"enr:-HW4QO1ml1DdXLeZLsUxewnthhUy8eROqkDyoMTyavfks9JlYQIlMFEUoM78PovJDPQrAkrb3LRJ-"`)},
   120  					{Value: sp(`"vtrymDguKCOIAWAgmlkgnY0iXNlY3AyNTZrMaEDffaGfJzgGhUif1JqFruZlYmA31HzathLSWxfbq_QoQ4"`)},
   121  				},
   122  				TTL:  ip(3333),
   123  				Type: sp("TXT"),
   124  			},
   125  		},
   126  		{
   127  			Action: sp("DELETE"),
   128  			ResourceRecordSet: &route53.ResourceRecordSet{
   129  				Name: sp("fdxn3sn67na5dka4j2gok7bvqi.n"),
   130  				ResourceRecords: []*route53.ResourceRecord{{
   131  					Value: sp(`"enrtree-branch:"`),
   132  				}},
   133  				TTL:  ip(treeNodeTTL),
   134  				Type: sp("TXT"),
   135  			},
   136  		},
   137  	}
   138  
   139  	var client route53Client
   140  	changes := client.computeChanges("n", testTree1, testTree0)
   141  	if !reflect.DeepEqual(changes, wantChanges) {
   142  		t.Fatalf("wrong changes (got %d, want %d)", len(changes), len(wantChanges))
   143  	}
   144  
   145  	wantSplit := [][]*route53.Change{
   146  		wantChanges[:4],
   147  		wantChanges[4:8],
   148  	}
   149  	split := splitChanges(changes, 600)
   150  	if !reflect.DeepEqual(split, wantSplit) {
   151  		t.Fatalf("wrong split batches: got %d, want %d", len(split), len(wantSplit))
   152  	}
   153  }
   154  
   155  func sp(s string) *string { return &s }
   156  func ip(i int64) *int64   { return &i }