prysm-pulse/validator/graffiti/parse_graffiti_test.go
Josh Yudaken f0eb843138
Graffiti ordered index (#8482)
* Added ordered option to graffiti file

* Updated validator to use Ordered graffiti

* Track graffiti ordered index in db

* Update `ordered` to only emit each graffiti once

Co-authored-by: pinglamb <pinglambs@gmail.com>
2021-02-24 22:50:47 +00:00

160 lines
3.6 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/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)
}