github.com/GGP1/kure@v0.8.4/commands/stats/stats_test.go (about)

     1  package stats
     2  
     3  import (
     4  	"testing"
     5  
     6  	cmdutil "github.com/GGP1/kure/commands"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  func TestStats(t *testing.T) {
    12  	db := cmdutil.SetContext(t)
    13  
    14  	t.Run("Success", func(t *testing.T) {
    15  		cmd := NewCmd(db)
    16  		err := cmd.Execute()
    17  		assert.NoError(t, err)
    18  	})
    19  
    20  	t.Run("Database connection closed", func(t *testing.T) {
    21  		db.Close()
    22  		err := NewCmd(db).Execute()
    23  		assert.Error(t, err)
    24  	})
    25  }