123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- // Copyright 2016 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package db
-
- import (
- "context"
- "math"
-
- "code.gitea.io/gitea/modules/unittestbridge"
-
- "xorm.io/xorm"
- )
-
- // Code in this file is mainly used by models.CheckConsistencyFor, which is not in the unit test for various reasons.
- // In the future if we can decouple CheckConsistencyFor into separate unit test code, then this file can be moved into unittest package too.
-
- // NonexistentID an ID that will never exist
- const NonexistentID = int64(math.MaxInt64)
-
- //SetUnitTestEngine is used by unit test code
- func SetUnitTestEngine(eng *xorm.Engine) {
- x = eng
- DefaultContext = &Context{
- Context: context.Background(),
- e: x,
- }
- }
-
- type testCond struct {
- query interface{}
- args []interface{}
- }
-
- // Cond create a condition with arguments for a test
- func Cond(query interface{}, args ...interface{}) interface{} {
- return &testCond{query: query, args: args}
- }
-
- func whereConditions(sess *xorm.Session, conditions []interface{}) {
- for _, condition := range conditions {
- switch cond := condition.(type) {
- case *testCond:
- sess.Where(cond.query, cond.args...)
- default:
- sess.Where(cond)
- }
- }
- }
-
- // LoadBeanIfExists loads beans from fixture database if exist
- func LoadBeanIfExists(bean interface{}, conditions ...interface{}) (bool, error) {
- sess := x.NewSession()
- defer sess.Close()
- whereConditions(sess, conditions)
- return sess.Get(bean)
- }
-
- // BeanExists for testing, check if a bean exists
- func BeanExists(t unittestbridge.Tester, bean interface{}, conditions ...interface{}) bool {
- ta := unittestbridge.NewAsserter(t)
- exists, err := LoadBeanIfExists(bean, conditions...)
- ta.NoError(err)
- return exists
- }
-
- // AssertExistsAndLoadBean assert that a bean exists and load it from the test database
- func AssertExistsAndLoadBean(t unittestbridge.Tester, bean interface{}, conditions ...interface{}) interface{} {
- ta := unittestbridge.NewAsserter(t)
- exists, err := LoadBeanIfExists(bean, conditions...)
- ta.NoError(err)
- ta.True(exists,
- "Expected to find %+v (of type %T, with conditions %+v), but did not",
- bean, bean, conditions)
- return bean
- }
-
- // GetCount get the count of a bean
- func GetCount(t unittestbridge.Tester, bean interface{}, conditions ...interface{}) int {
- ta := unittestbridge.NewAsserter(t)
- sess := x.NewSession()
- defer sess.Close()
- whereConditions(sess, conditions)
- count, err := sess.Count(bean)
- ta.NoError(err)
- return int(count)
- }
-
- // AssertNotExistsBean assert that a bean does not exist in the test database
- func AssertNotExistsBean(t unittestbridge.Tester, bean interface{}, conditions ...interface{}) {
- ta := unittestbridge.NewAsserter(t)
- exists, err := LoadBeanIfExists(bean, conditions...)
- ta.NoError(err)
- ta.False(exists)
- }
-
- // AssertExistsIf asserts that a bean exists or does not exist, depending on
- // what is expected.
- func AssertExistsIf(t unittestbridge.Tester, expected bool, bean interface{}, conditions ...interface{}) {
- ta := unittestbridge.NewAsserter(t)
- exists, err := LoadBeanIfExists(bean, conditions...)
- ta.NoError(err)
- ta.Equal(expected, exists)
- }
-
- // AssertSuccessfulInsert assert that beans is successfully inserted
- func AssertSuccessfulInsert(t unittestbridge.Tester, beans ...interface{}) {
- ta := unittestbridge.NewAsserter(t)
- _, err := x.Insert(beans...)
- ta.NoError(err)
- }
-
- // AssertCount assert the count of a bean
- func AssertCount(t unittestbridge.Tester, bean, expected interface{}) {
- ta := unittestbridge.NewAsserter(t)
- ta.EqualValues(expected, GetCount(ta, bean))
- }
-
- // AssertInt64InRange assert value is in range [low, high]
- func AssertInt64InRange(t unittestbridge.Tester, low, high, value int64) {
- ta := unittestbridge.NewAsserter(t)
- ta.True(value >= low && value <= high,
- "Expected value in range [%d, %d], found %d", low, high, value)
- }
|