karma/store/store_test.go

package store_test

import (
	"testing"

	"github.com/cloudflare/unsee/models"
	"github.com/cloudflare/unsee/store"
)

type silenceTest struct {
	silences  map[string]models.Silence
	silenceId string
	found     bool
}

var silenceTests = []silenceTest{
	silenceTest{
		silences: map[string]models.Silence{
			"1": models.Silence{},
		},
		silenceId: "1",
		found:     true,
	},
	silenceTest{
		silences: map[string]models.Silence{
			"1": models.Silence{},
			"2": models.Silence{},
			"3": models.Silence{},
		},
		silenceId: "2",
		found:     true,
	},
	silenceTest{
		silences:  map[string]models.Silence{},
		silenceId: "1",
		found:     false,
	},
	silenceTest{
		silences: map[string]models.Silence{
			"2": models.Silence{},
			"3": models.Silence{},
		},
		silenceId: "1",
		found:     false,
	},
}

func TestSilences(t *testing.T) {
	for _, testCase := range silenceTests {
		store.Store.SetSilences(testCase.silences)
		silence := store.Store.GetSilence(testCase.silenceId)
		found := silence != nil
		if found != testCase.found {
			t.Errorf("GetSilence('%s') returned %v, %v was expected", testCase.silenceId, found, testCase.found)
		}
	}
}