Unit tests for models/access.go (#606)

1 files changed, 15 insertions, 0 deletions

diff --git a/models/setup_for_test.go b/models/setup_for_test.go
index b585a7490e..cf88397c26 100644
--- a/models/setup_for_test.go
+++ b/models/setup_for_test.go
@@ -12,6 +12,7 @@ import (
 	"github.com/go-xorm/core"
 	"github.com/go-xorm/xorm"
 	_ "github.com/mattn/go-sqlite3" // for the test engine
+	"github.com/stretchr/testify/assert"
 	"gopkg.in/testfixtures.v2"
 )
 
@@ -45,3 +46,17 @@ func CreateTestEngine() error {
 func PrepareTestDatabase() error {
 	return fixtures.Load()
 }
+
+// LoadFixture load a test fixture from the test database, failing if fixture
+// does not exist
+func LoadTestFixture(t *testing.T, fixture interface{}, conditions... interface{}) {
+	sess := x.NewSession()
+	defer sess.Close()
+
+	for _, cond := range conditions {
+		sess = sess.Where(cond)
+	}
+	has, err := sess.Get(fixture)
+	assert.NoError(t, err)
+	assert.True(t, has)
+}