cosmos-sdk/telemetry/metrics_test.go
Andrei Ivasko 6cbbd6da75
refactor!: Keyring migration (#9695)
<!--
The default pull request template is for types feat, fix, or refactor.
For other templates, add one of the following parameters to the url:
- template=docs.md
- template=other.md
-->

## Description

The draft PR #9222 
Closes: #7108

<!-- Add a description of the changes that this PR introduces and the files that
are the most critical to review. -->

- implement proto definition for `Record` 
- rename `Info.go` to `legacyInfo.go` within `keyring` package
- implement CLI `migrate` command that migrates all keys from  legacyInfo to proto according to @robert-zaremba migration [algorithm](https://github.com/cosmos/cosmos-sdk/pull/9222/#discussion_r624683839)
- remove legacy keybase entirely.
- add `Migrate` and `MigrateAll` functions  in `keyring.go` for single key and all keys migration
- add tests
- fix tests

---

### Author Checklist

*All items are required. Please add a note to the item if the item is not applicable and
please add links to any relevant follow up issues.*

I have...

- [ ] included the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title
- [ ] added `!` to the type prefix if API or client breaking change
- [ ] targeted the correct branch (see [PR Targeting](https://github.com/cosmos/cosmos-sdk/blob/master/CONTRIBUTING.md#pr-targeting))
- [ ] provided a link to the relevant issue or specification
- [ ] followed the guidelines for [building modules](https://github.com/cosmos/cosmos-sdk/blob/master/docs/building-modules)
- [x] included the necessary unit and integration [tests](https://github.com/cosmos/cosmos-sdk/blob/master/CONTRIBUTING.md#testing)
- [ ] added a changelog entry to `CHANGELOG.md`
- [ ] included comments for [documenting Go code](https://blog.golang.org/godoc)
- [ ] updated the relevant documentation or specification
- [x] reviewed "Files changed" and left comments if necessary
- [ ] confirmed all CI checks have passed

### Reviewers Checklist

*All items are required. Please add a note if the item is not applicable and please add
your handle next to the items reviewed if you only reviewed selected items.*

I have...

- [ ] confirmed the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title
- [ ] confirmed `!` in the type prefix if API or client breaking change
- [ ] confirmed all author checklist items have been addressed 
- [ ] reviewed state machine logic
- [ ] reviewed API design and naming
- [ ] reviewed documentation is accurate
- [ ] reviewed tests and test coverage
- [ ] manually tested (if applicable)
2021-09-20 12:02:15 +00:00

77 lines
1.7 KiB
Go

package telemetry
import (
"encoding/json"
"strings"
"testing"
"time"
"github.com/armon/go-metrics"
"github.com/prometheus/common/expfmt"
"github.com/stretchr/testify/require"
)
func TestMetrics_Disabled(t *testing.T) {
m, err := New(Config{Enabled: false})
require.Nil(t, m)
require.Nil(t, err)
}
func TestMetrics_InMem(t *testing.T) {
m, err := New(Config{
Enabled: true,
EnableHostname: false,
ServiceName: "test",
})
require.NoError(t, err)
require.NotNil(t, m)
emitMetrics()
gr, err := m.Gather(FormatText)
require.NoError(t, err)
require.Equal(t, gr.ContentType, "application/json")
jsonMetrics := make(map[string]interface{})
require.NoError(t, json.Unmarshal(gr.Metrics, &jsonMetrics))
counters := jsonMetrics["Counters"].([]interface{})
require.Equal(t, counters[0].(map[string]interface{})["Count"].(float64), 10.0)
require.Equal(t, counters[0].(map[string]interface{})["Name"].(string), "test.dummy_counter")
}
func TestMetrics_Prom(t *testing.T) {
m, err := New(Config{
Enabled: true,
EnableHostname: false,
ServiceName: "test",
PrometheusRetentionTime: 60,
EnableHostnameLabel: false,
})
require.NoError(t, err)
require.NotNil(t, m)
require.True(t, m.prometheusEnabled)
emitMetrics()
gr, err := m.Gather(FormatPrometheus)
require.NoError(t, err)
require.Equal(t, gr.ContentType, string(expfmt.FmtText))
require.True(t, strings.Contains(string(gr.Metrics), "test_dummy_counter 30"))
}
func emitMetrics() {
ticker := time.NewTicker(time.Second)
timeout := time.After(30 * time.Second)
for {
select {
case <-ticker.C:
metrics.IncrCounter([]string{"dummy_counter"}, 1.0)
case <-timeout:
return
}
}
}