github.com/juju/juju@v0.0.0-20240327075706-a90865de2538/worker/httpserver/mock_test.go (about)

     1  // Copyright 2018 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package httpserver_test
     5  
     6  import (
     7  	"github.com/juju/testing"
     8  	"github.com/prometheus/client_golang/prometheus"
     9  
    10  	"github.com/juju/juju/state"
    11  )
    12  
    13  type stubStateTracker struct {
    14  	testing.Stub
    15  	pool *state.StatePool
    16  }
    17  
    18  func (s *stubStateTracker) Use() (*state.StatePool, error) {
    19  	s.MethodCall(s, "Use")
    20  	return s.pool, s.NextErr()
    21  }
    22  
    23  func (s *stubStateTracker) Done() error {
    24  	s.MethodCall(s, "Done")
    25  	return s.NextErr()
    26  }
    27  
    28  func (s *stubStateTracker) Report() map[string]interface{} {
    29  	s.MethodCall(s, "Report")
    30  	return nil
    31  }
    32  
    33  type stubPrometheusRegisterer struct {
    34  	testing.Stub
    35  }
    36  
    37  func (s *stubPrometheusRegisterer) MustRegister(...prometheus.Collector) {
    38  	panic("should not be called")
    39  }
    40  
    41  func (s *stubPrometheusRegisterer) Register(c prometheus.Collector) error {
    42  	s.MethodCall(s, "Register", c)
    43  	return s.NextErr()
    44  }
    45  
    46  func (s *stubPrometheusRegisterer) Unregister(c prometheus.Collector) bool {
    47  	s.MethodCall(s, "Unregister", c)
    48  	return false
    49  }