Update/tests (#20)

* Some really basic testing for the generator.

* Basic tests for handler and stat factories.

Co-authored-by: ipchama <[email protected]>

Author: ipchama

generator/factory_test.go

@@ -0,0 +1,61 @@
+package generator_test
+
+import (
+	"github.com/ipchama/dhammer/generator"
+	"github.com/ipchama/dhammer/stats"
+	"testing"
+)
+
+type TestHammerConfig struct {
+	hType string
+}
+
+func (t *TestHammerConfig) HammerType() string {
+	return t.hType
+}
+
+type TestGenerator struct {
+}
+
+func (t *TestGenerator) Init() error {
+	return nil
+}
+
+func (t *TestGenerator) Update(i interface{}) error {
+	return nil
+}
+
+func (t *TestGenerator) Run() {
+}
+
+func (t *TestGenerator) Stop() error {
+	return nil
+}
+
+func (t *TestGenerator) DeInit() error {
+	return nil
+}
+
+func TestNew(t *testing.T) {
+
+	o := &TestHammerConfig{
+		hType: "__TEST__",
+	}
+
+	if _, err := generator.New(nil, o, func(string) bool { return true }, func(error) bool { return true }, func(stats.StatValue) bool { return true }); err == nil {
+		t.Errorf("Generator factory did not return error for unknown type.")
+	}
+
+	if err := generator.AddGenerator(o.hType, func(g generator.GeneratorInitParams) generator.Generator { return &TestGenerator{} }); err != nil {
+		t.Errorf("Generator factory failed to add new type.")
+	}
+
+	if err := generator.AddGenerator(o.hType, func(g generator.GeneratorInitParams) generator.Generator { return &TestGenerator{} }); err == nil {
+		t.Errorf("Generator factory allowed duplicate type.")
+	}
+
+	if _, err := generator.New(nil, o, func(string) bool { return true }, func(error) bool { return true }, func(stats.StatValue) bool { return true }); err != nil {
+		t.Errorf("Generator factory failed to return known type.")
+	}
+
+}

handler/factory_test.go

@@ -0,0 +1,62 @@
+package handler_test
+
+import (
+	"github.com/ipchama/dhammer/handler"
+	"github.com/ipchama/dhammer/message"
+	"github.com/ipchama/dhammer/stats"
+	"testing"
+)
+
+type TestHammerConfig struct {
+	hType string
+}
+
+func (t *TestHammerConfig) HammerType() string {
+	return t.hType
+}
+
+type TestHandler struct {
+}
+
+func (t *TestHandler) Init() error {
+	return nil
+}
+
+func (t *TestHandler) ReceiveMessage(m message.Message) bool {
+	return true
+}
+
+func (t *TestHandler) Run() {
+}
+
+func (t *TestHandler) Stop() error {
+	return nil
+}
+
+func (t *TestHandler) DeInit() error {
+	return nil
+}
+
+func TestNew(t *testing.T) {
+
+	o := &TestHammerConfig{
+		hType: "__TEST__",
+	}
+
+	if _, err := handler.New(nil, o, func(string) bool { return true }, func(error) bool { return true }, func(stats.StatValue) bool { return true }); err == nil {
+		t.Errorf("Handler factory did not return error for unknown type.")
+	}
+
+	if err := handler.AddHandler(o.hType, func(h handler.HandlerInitParams) handler.Handler { return &TestHandler{} }); err != nil {
+		t.Errorf("Handler factory failed to add new type.")
+	}
+
+	if err := handler.AddHandler(o.hType, func(h handler.HandlerInitParams) handler.Handler { return &TestHandler{} }); err == nil {
+		t.Errorf("Handler factory allowed duplicate type.")
+	}
+
+	if _, err := handler.New(nil, o, func(string) bool { return true }, func(error) bool { return true }, func(stats.StatValue) bool { return true }); err != nil {
+		t.Errorf("Handler factory failed to return known type.")
+	}
+
+}

stats/factory_test.go

@@ -0,0 +1,64 @@
+package stats_test
+
+import (
+	"github.com/ipchama/dhammer/stats"
+	"testing"
+)
+
+type TestHammerConfig struct {
+	hType string
+}
+
+func (t *TestHammerConfig) HammerType() string {
+	return t.hType
+}
+
+type TestStats struct {
+}
+
+func (t *TestStats) Init() error {
+	return nil
+}
+
+func (t *TestStats) AddStat(s stats.StatValue) bool {
+	return true
+}
+
+func (t *TestStats) Run() {
+}
+
+func (t *TestStats) String() string {
+	return ""
+}
+
+func (t *TestStats) Stop() error {
+	return nil
+}
+
+func (t *TestStats) DeInit() error {
+	return nil
+}
+
+func TestNew(t *testing.T) {
+
+	o := &TestHammerConfig{
+		hType: "__TEST__",
+	}
+
+	if _, err := stats.New(o, func(string) bool { return true }, func(error) bool { return true }); err == nil {
+		t.Errorf("Stats factory did not return error for unknown type.")
+	}
+
+	if err := stats.AddStatter(o.hType, func(h stats.StatsInitParams) stats.Stats { return &TestStats{} }); err != nil {
+		t.Errorf("Stats factory failed to add new type.")
+	}
+
+	if err := stats.AddStatter(o.hType, func(h stats.StatsInitParams) stats.Stats { return &TestStats{} }); err == nil {
+		t.Errorf("Stats factory allowed duplicate type.")
+	}
+
+	if _, err := stats.New(o, func(string) bool { return true }, func(error) bool { return true }); err != nil {
+		t.Errorf("Stats factory failed to return known type.")
+	}
+
+}