statsclient: allow index as pattern

[govpp.git] / adapter / mock / mock_stats_adapter.go

diff --git a/adapter/mock/mock_stats_adapter.go b/adapter/mock/mock_stats_adapter.go
index 55b1831..08d18d4 100644 (file)
--- a/adapter/mock/mock_stats_adapter.go
+++ b/adapter/mock/mock_stats_adapter.go
@@ -46,10 +46,13 @@ func (a *StatsAdapter) Disconnect() error {
 }
 
 // ListStats mocks name listing for all stats.
-func (a *StatsAdapter) ListStats(patterns ...string) ([]string, error) {
-       var statNames []string
+func (a *StatsAdapter) ListStats(patterns ...string) ([]adapter.StatIdentifier, error) {
+       var statNames []adapter.StatIdentifier
        for _, stat := range a.entries {
-               statNames = append(statNames, string(stat.Name))
+               statNames = append(statNames, adapter.StatIdentifier{
+                       Name:  stat.Name,
+                       Index: stat.Index,
+               })
        }
        return statNames, nil
 }
@@ -63,6 +66,10 @@ func (a *StatsAdapter) PrepareDir(prefixes ...string) (*adapter.StatDir, error)
        return a.dir, nil
 }
 
+func (a *StatsAdapter) PrepareDirOnIndex(indexes ...uint32) (*adapter.StatDir, error) {
+       return a.dir, nil
+}
+
 func (a *StatsAdapter) UpdateDir(dir *adapter.StatDir) error {
        *dir = *a.dir
        return nil