github.com/netdata/go.d.plugin@v0.58.1/agent/module/mock_test.go (about)

     1  // SPDX-License-Identifier: GPL-3.0-or-later
     2  
     3  package module
     4  
     5  import (
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  	"github.com/stretchr/testify/require"
    10  )
    11  
    12  func TestMockModule_Init(t *testing.T) {
    13  	m := &MockModule{}
    14  
    15  	assert.True(t, m.Init())
    16  	m.InitFunc = func() bool { return false }
    17  	assert.False(t, m.Init())
    18  }
    19  
    20  func TestMockModule_Check(t *testing.T) {
    21  	m := &MockModule{}
    22  
    23  	assert.True(t, m.Check())
    24  	m.CheckFunc = func() bool { return false }
    25  	assert.False(t, m.Check())
    26  }
    27  
    28  func TestMockModule_Charts(t *testing.T) {
    29  	m := &MockModule{}
    30  	c := &Charts{}
    31  
    32  	assert.Nil(t, m.Charts())
    33  	m.ChartsFunc = func() *Charts { return c }
    34  	assert.True(t, c == m.Charts())
    35  }
    36  
    37  func TestMockModule_Collect(t *testing.T) {
    38  	m := &MockModule{}
    39  	d := map[string]int64{
    40  		"1": 1,
    41  	}
    42  
    43  	assert.Nil(t, m.Collect())
    44  	m.CollectFunc = func() map[string]int64 { return d }
    45  	assert.Equal(t, d, m.Collect())
    46  }
    47  
    48  func TestMockModule_Cleanup(t *testing.T) {
    49  	m := &MockModule{}
    50  	require.False(t, m.CleanupDone)
    51  
    52  	m.Cleanup()
    53  	assert.True(t, m.CleanupDone)
    54  }