70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
|
package nicehash
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"dappco.re/go/proxy"
|
||
|
|
)
|
||
|
|
|
||
|
|
type upstreamStateStrategy struct {
|
||
|
|
active bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *upstreamStateStrategy) Connect() {}
|
||
|
|
|
||
|
|
func (s *upstreamStateStrategy) Submit(jobID, nonce, result, algo string) int64 {
|
||
|
|
return 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *upstreamStateStrategy) Disconnect() {}
|
||
|
|
|
||
|
|
func (s *upstreamStateStrategy) IsActive() bool { return s.active }
|
||
|
|
|
||
|
|
func TestNonceSplitter_Upstreams_Good(t *testing.T) {
|
||
|
|
splitter := &NonceSplitter{
|
||
|
|
mappers: []*NonceMapper{
|
||
|
|
{strategy: &upstreamStateStrategy{active: true}, active: true},
|
||
|
|
{strategy: &upstreamStateStrategy{active: false}, active: false, suspended: 1},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
stats := splitter.Upstreams()
|
||
|
|
|
||
|
|
if stats.Active != 1 {
|
||
|
|
t.Fatalf("expected one active upstream, got %d", stats.Active)
|
||
|
|
}
|
||
|
|
if stats.Error != 1 {
|
||
|
|
t.Fatalf("expected one error upstream, got %d", stats.Error)
|
||
|
|
}
|
||
|
|
if stats.Total != 2 {
|
||
|
|
t.Fatalf("expected total to equal active + sleep + error, got %d", stats.Total)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNonceSplitter_Upstreams_Bad(t *testing.T) {
|
||
|
|
var splitter *NonceSplitter
|
||
|
|
|
||
|
|
stats := splitter.Upstreams()
|
||
|
|
|
||
|
|
if stats != (proxy.UpstreamStats{}) {
|
||
|
|
t.Fatalf("expected zero-value stats for nil splitter, got %+v", stats)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNonceSplitter_Upstreams_Ugly(t *testing.T) {
|
||
|
|
splitter := &NonceSplitter{
|
||
|
|
mappers: []*NonceMapper{
|
||
|
|
{strategy: &upstreamStateStrategy{active: false}, active: false},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
stats := splitter.Upstreams()
|
||
|
|
|
||
|
|
if stats.Error != 1 {
|
||
|
|
t.Fatalf("expected an unready mapper to be counted as error, got %+v", stats)
|
||
|
|
}
|
||
|
|
if stats.Total != 1 {
|
||
|
|
t.Fatalf("expected total to remain internally consistent, got %+v", stats)
|
||
|
|
}
|
||
|
|
}
|