mirror of
https://github.com/ap-pauloafonso/ratio-spoof.git
synced 2026-04-20 08:02:07 +00:00
cleanup
This commit is contained in:
parent
42ec97200b
commit
84f621707f
6 changed files with 21 additions and 7 deletions
|
|
@ -52,17 +52,17 @@ func NewEmulation(code string) (*Emulation, error) {
|
|||
return nil, err
|
||||
}
|
||||
|
||||
peerG, err := generator.NewRegexPeerIdGenerator(c.PeerID.Generator, c.PeerID.Regex)
|
||||
peerG, err := generator.NewRegexPeerIdGenerator(c.PeerID.Regex)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
keyG, err := generator.NewDefaultKeyGenerator(c.Key.Generator)
|
||||
keyG, err := generator.NewDefaultKeyGenerator()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
roudingG, err := generator.NewDefaultRoudingGenerator(c.Rounding.Generator)
|
||||
roudingG, err := generator.NewDefaultRoudingGenerator()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
|
|
|||
|
|
@ -6,7 +6,7 @@ import (
|
|||
"strings"
|
||||
)
|
||||
|
||||
func NewDefaultKeyGenerator(generatorCode string) (*DefaultKeyGenerator, error) {
|
||||
func NewDefaultKeyGenerator() (*DefaultKeyGenerator, error) {
|
||||
randomBytes := make([]byte, 4)
|
||||
rand.Read(randomBytes)
|
||||
str := hex.EncodeToString(randomBytes)
|
||||
|
|
|
|||
14
internal/generator/key_test.go
Normal file
14
internal/generator/key_test.go
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
package generator
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestDeaultKeyGenerator(t *testing.T) {
|
||||
t.Run("Key has 8 length", func(t *testing.T) {
|
||||
obj, _ := NewDefaultKeyGenerator()
|
||||
key := obj.Key()
|
||||
if len(key) != 8 {
|
||||
t.Error("Keys must have length of 8")
|
||||
}
|
||||
|
||||
})
|
||||
}
|
||||
|
|
@ -8,7 +8,7 @@ type RegexPeerIdGenerator struct {
|
|||
generated string
|
||||
}
|
||||
|
||||
func NewRegexPeerIdGenerator(generatorCode, pattern string) (*RegexPeerIdGenerator, error) {
|
||||
func NewRegexPeerIdGenerator(pattern string) (*RegexPeerIdGenerator, error) {
|
||||
result, err := regen.Generate(pattern)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
|
|
|||
|
|
@ -2,7 +2,7 @@ package generator
|
|||
|
||||
type DefaultRoundingGenerator struct{}
|
||||
|
||||
func NewDefaultRoudingGenerator(code string) (*DefaultRoundingGenerator, error) {
|
||||
func NewDefaultRoudingGenerator() (*DefaultRoundingGenerator, error) {
|
||||
return &DefaultRoundingGenerator{}, nil
|
||||
|
||||
}
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@ package generator
|
|||
import "testing"
|
||||
|
||||
func TestNextAmountReport(t *testing.T) {
|
||||
r, _ := NewDefaultRoudingGenerator("")
|
||||
r, _ := NewDefaultRoudingGenerator()
|
||||
|
||||
d, u, l := r.Round(656497856, 46479878, 7879879, 1024)
|
||||
//same
|
||||
|
|
|
|||
Loading…
Reference in a new issue