Add AddRoute tests

1 files changed, 77 insertions, 0 deletions

diff --git a/rules_manager_test.go b/rules_manager_test.go
new file mode 100644
index 0000000..53d085d
--- /dev/null
+++ b/rules_manager_test.go
@@ -0,0 +1,77 @@
+package main
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+	"time"
+)
+
+func TestAddRule(t *testing.T) {
+	wrapper := NewTestStorageWrapper(t)
+	wrapper.AddCollection(Rules)
+
+	rulesManager := NewRulesManager(wrapper.Storage).(*rulesManagerImpl)
+
+	checkVersion := func(id RowID) {
+		timeout := time.Tick(1 * time.Second)
+
+		select {
+		case database := <-rulesManager.databaseUpdated:
+			assert.Equal(t, id, database.version)
+		case <-timeout:
+			t.Fatal("timeout")
+		}
+	}
+
+	err := rulesManager.SetFlag(wrapper.Context, "FLAG{test}")
+	assert.NoError(t, err)
+	checkVersion(rulesManager.rulesByName["flag"].ID)
+	emptyID, err := rulesManager.AddRule(wrapper.Context, Rule{Name: "empty", Color: "#fff"})
+	assert.NoError(t, err)
+	assert.NotNil(t, emptyID)
+	checkVersion(emptyID)
+
+	rule1 := Rule{
+		Name:  "rule1",
+		Color: "#eee",
+		Patterns: []Pattern{
+			{Regex: "nope", Flags: RegexFlags{Caseless: true}},
+		},
+	}
+	rule1ID, err := rulesManager.AddRule(wrapper.Context, rule1)
+	assert.NoError(t, err)
+	assert.NotNil(t, rule1ID)
+	checkVersion(rule1ID)
+
+	rule2 := Rule{
+		Name:  "rule2",
+		Color: "#ddd",
+		Patterns: []Pattern{
+			{Regex: "nope", Flags: RegexFlags{Caseless: true}},
+			{Regex: "yep"},
+		},
+	}
+	rule2ID, err := rulesManager.AddRule(wrapper.Context, rule2)
+	assert.NoError(t, err)
+	assert.NotNil(t, rule2ID)
+	checkVersion(rule2ID)
+
+	assert.Len(t, rulesManager.rules, 4)
+	assert.Len(t, rulesManager.rulesByName, 4)
+	assert.Len(t, rulesManager.patterns, 3)
+	assert.Len(t, rulesManager.patternsIds, 3)
+	assert.Equal(t, emptyID, rulesManager.rules[emptyID].ID)
+	assert.Equal(t, emptyID, rulesManager.rulesByName["empty"].ID)
+	assert.Len(t, rulesManager.rules[emptyID].Patterns, 0)
+	assert.Equal(t, rule1ID, rulesManager.rules[rule1ID].ID)
+	assert.Equal(t, rule1ID, rulesManager.rulesByName[rule1.Name].ID)
+	assert.Len(t, rulesManager.rules[rule1ID].Patterns, 1)
+	assert.Equal(t, 1, rulesManager.rules[rule1ID].Patterns[0].internalID)
+	assert.Equal(t, rule2ID, rulesManager.rules[rule2ID].ID)
+	assert.Equal(t, rule2ID, rulesManager.rulesByName[rule2.Name].ID)
+	assert.Len(t, rulesManager.rules[rule2ID].Patterns, 2)
+	assert.Equal(t, 1, rulesManager.rules[rule2ID].Patterns[0].internalID)
+	assert.Equal(t, 2, rulesManager.rules[rule2ID].Patterns[1].internalID)
+
+	wrapper.Destroy(t)
+}