prysm-pulse/validator/graffiti/parse_graffiti_test.go
Alexander Mollohan 644d5bbb1a
Graffiti hex support (#8894)
* more or less feature complete, needs unit tests

* Test support and concomitant refactors

* Resolving requested changes on PR

* minor fix before trying to figure out bazel

* bazel change

* minor changes requested in pr

* fixing build errors

* fixing build errors

* fixing PR feedback

* fixing PR feedback

* resolving nitpicks

* resolving nitpicks

* build failures REEEEE

* tests fail if its not this way, so this way it is

* getting the tests to pass

Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com>
2021-06-10 16:54:24 +00:00

222 lines
4.8 KiB
Go

package graffiti
import (
"io/ioutil"
"os"
"path/filepath"
"testing"
types "github.com/prysmaticlabs/eth2-types"
"github.com/prysmaticlabs/prysm/shared/hashutil"
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
"github.com/prysmaticlabs/prysm/shared/testutil/require"
)
func TestParseGraffitiFile_Default(t *testing.T) {
input := []byte(`default: "Mr T was here"`)
dirName := t.TempDir() + "somedir"
err := os.MkdirAll(dirName, os.ModePerm)
require.NoError(t, err)
someFileName := filepath.Join(dirName, "somefile.txt")
require.NoError(t, ioutil.WriteFile(someFileName, input, os.ModePerm))
got, err := ParseGraffitiFile(someFileName)
require.NoError(t, err)
wanted := &Graffiti{
Hash: hashutil.Hash(input),
Default: "Mr T was here",
}
require.DeepEqual(t, wanted, got)
}
func TestParseGraffitiFile_Random(t *testing.T) {
input := []byte(`random:
- "Mr A was here"
- "Mr B was here"
- "Mr C was here"`)
dirName := t.TempDir() + "somedir"
err := os.MkdirAll(dirName, os.ModePerm)
require.NoError(t, err)
someFileName := filepath.Join(dirName, "somefile.txt")
require.NoError(t, ioutil.WriteFile(someFileName, input, os.ModePerm))
got, err := ParseGraffitiFile(someFileName)
require.NoError(t, err)
wanted := &Graffiti{
Hash: hashutil.Hash(input),
Random: []string{
"Mr A was here",
"Mr B was here",
"Mr C was here",
},
}
require.DeepEqual(t, wanted, got)
}
func TestParseGraffitiFile_Ordered(t *testing.T) {
input := []byte(`ordered:
- "Mr D was here"
- "Mr E was here"
- "Mr F was here"`)
dirName := t.TempDir() + "somedir"
err := os.MkdirAll(dirName, os.ModePerm)
require.NoError(t, err)
someFileName := filepath.Join(dirName, "somefile.txt")
require.NoError(t, ioutil.WriteFile(someFileName, input, os.ModePerm))
got, err := ParseGraffitiFile(someFileName)
require.NoError(t, err)
wanted := &Graffiti{
Hash: hashutil.Hash(input),
Ordered: []string{
"Mr D was here",
"Mr E was here",
"Mr F was here",
},
}
require.DeepEqual(t, wanted, got)
}
func TestParseGraffitiFile_Validators(t *testing.T) {
input := []byte(`
specific:
1234: Yolo
555: "What's up"
703727: Meow`)
dirName := t.TempDir() + "somedir"
err := os.MkdirAll(dirName, os.ModePerm)
require.NoError(t, err)
someFileName := filepath.Join(dirName, "somefile.txt")
require.NoError(t, ioutil.WriteFile(someFileName, input, os.ModePerm))
got, err := ParseGraffitiFile(someFileName)
require.NoError(t, err)
wanted := &Graffiti{
Hash: hashutil.Hash(input),
Specific: map[types.ValidatorIndex]string{
1234: "Yolo",
555: "What's up",
703727: "Meow",
},
}
require.DeepEqual(t, wanted, got)
}
func TestParseGraffitiFile_AllFields(t *testing.T) {
input := []byte(`default: "Mr T was here"
random:
- "Mr A was here"
- "Mr B was here"
- "Mr C was here"
ordered:
- "Mr D was here"
- "Mr E was here"
- "Mr F was here"
specific:
1234: Yolo
555: "What's up"
703727: Meow`)
dirName := t.TempDir() + "somedir"
err := os.MkdirAll(dirName, os.ModePerm)
require.NoError(t, err)
someFileName := filepath.Join(dirName, "somefile.txt")
require.NoError(t, ioutil.WriteFile(someFileName, input, os.ModePerm))
got, err := ParseGraffitiFile(someFileName)
require.NoError(t, err)
wanted := &Graffiti{
Hash: hashutil.Hash(input),
Default: "Mr T was here",
Random: []string{
"Mr A was here",
"Mr B was here",
"Mr C was here",
},
Ordered: []string{
"Mr D was here",
"Mr E was here",
"Mr F was here",
},
Specific: map[types.ValidatorIndex]string{
1234: "Yolo",
555: "What's up",
703727: "Meow",
},
}
require.DeepEqual(t, wanted, got)
}
func TestParseHexGraffiti(t *testing.T) {
tests := []struct {
name string
want string
input string
}{
{
name: "standard",
want: "hola mundo!",
input: "hola mundo!",
},
{
name: "standard with hex tag",
want: "hola mundo!",
input: "hex:686f6c61206d756e646f21",
},
{
name: "irregularly cased hex tag",
want: "hola mundo!",
input: "HEX:686f6c61206d756e646f21",
},
{
name: "hex tag without accompanying data",
want: "hex:",
input: "hex:",
},
{
name: "Passing non-hex data with hex tag",
want: "hex:hola mundo!",
input: "hex:hola mundo!",
},
{
name: "unmarked hex input",
want: "0x686f6c61206d756e646f21",
input: "0x686f6c61206d756e646f21",
},
{
name: "Properly tagged hex data with 0x prefix",
want: "hola mundo!",
input: "hex:0x686f6c61206d756e646f21",
},
{
name: "hex tag with 0x prefix and no other data",
want: "hex:0x",
input: "hex:0x",
},
{
name: "hex tag with 0x prefix and invalid hex data",
want: "hex:0xhola mundo",
input: "hex:0xhola mundo",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
out := ParseHexGraffiti(tt.input)
assert.Equal(t, out, tt.want)
})
}
}