Change module name to go.fd.io/govpp

[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 aba93a2..f2378f3 100644 (file)
--- a/adapter/mock/mock_stats_adapter.go
+++ b/adapter/mock/mock_stats_adapter.go
@@ -18,12 +18,16 @@
 package mock
 
 import (
-       "git.fd.io/govpp.git/adapter"
+       "go.fd.io/govpp/adapter"
 )
 
+// implements StatsAPI
+var _ adapter.StatsAPI = (*StatsAdapter)(nil)
+
 // StatsAdapter simulates VPP stats socket from which stats can be read
 type StatsAdapter struct {
-       entries []*adapter.StatEntry
+       entries []adapter.StatEntry
+       dir     *adapter.StatDir
 }
 
 // NewStatsAdapter returns a new mock stats adapter.
@@ -42,20 +46,41 @@ 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, stat.Name)
+               statNames = append(statNames, adapter.StatIdentifier{
+                       Name:  stat.Name,
+                       Index: stat.Index,
+               })
        }
        return statNames, nil
 }
 
 // DumpStats mocks all stat entries dump.
-func (a *StatsAdapter)  DumpStats(patterns ...string) ([]*adapter.StatEntry, error) {
+func (a *StatsAdapter) DumpStats(patterns ...string) ([]adapter.StatEntry, error) {
        return a.entries, nil
 }
 
-// MockStats replaces current values of all supported stats by provided value
-func (a *StatsAdapter) MockStats(stats []*adapter.StatEntry) {
+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
+}
+
+// MockStats sets mocked stat entries to be returned by DumpStats.
+func (a *StatsAdapter) MockStats(stats []adapter.StatEntry) {
        a.entries = stats
-}
\ No newline at end of file
+}
+
+// MockStats sets mocked stat dir to be returned by PrepareDir.
+func (a *StatsAdapter) MockDir(dir *adapter.StatDir) {
+       a.dir = dir
+}