stefanoj3 /
dirstalk
| Total Lines | 54 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | package test |
||
| 2 | |||
| 3 | import ( |
||
| 4 | "sync" |
||
| 5 | |||
| 6 | "net/http" |
||
| 7 | "net/http/httptest" |
||
| 8 | ) |
||
| 9 | |||
| 10 | func NewServerWithAssertion(handler http.HandlerFunc) (*httptest.Server, *ServerAssertion) { |
||
|
introduced
by
|
|||
| 11 | serverAssertion := &ServerAssertion{} |
||
| 12 | |||
| 13 | server := httptest.NewServer(serverAssertion.wrap(handler)) |
||
| 14 | |||
| 15 | return server, serverAssertion |
||
| 16 | } |
||
| 17 | |||
| 18 | type ServerAssertion struct { |
||
| 19 | requests []http.Request |
||
| 20 | requestsMx sync.RWMutex |
||
| 21 | } |
||
| 22 | |||
| 23 | func (s *ServerAssertion) wrap(handler http.HandlerFunc) http.HandlerFunc { |
||
| 24 | return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
||
| 25 | handler(w, r) |
||
| 26 | |||
| 27 | s.requestsMx.Lock() |
||
| 28 | defer s.requestsMx.Unlock() |
||
| 29 | |||
| 30 | s.requests = append(s.requests, *r) |
||
| 31 | }) |
||
| 32 | } |
||
| 33 | |||
| 34 | func (s *ServerAssertion) Range(fn func(index int, r http.Request)) { |
||
| 35 | s.requestsMx.RLock() |
||
| 36 | defer s.requestsMx.RUnlock() |
||
| 37 | |||
| 38 | for i, r := range s.requests { |
||
| 39 | fn(i, r) |
||
| 40 | } |
||
| 41 | } |
||
| 42 | |||
| 43 | func (s *ServerAssertion) At(index int, fn func(r http.Request)) { |
||
| 44 | s.requestsMx.RLock() |
||
| 45 | defer s.requestsMx.RUnlock() |
||
| 46 | |||
| 47 | fn(s.requests[index]) |
||
| 48 | } |
||
| 49 | |||
| 50 | func (s *ServerAssertion) Len() int { |
||
| 51 | s.requestsMx.RLock() |
||
| 52 | defer s.requestsMx.RUnlock() |
||
| 53 | |||
| 54 | return len(s.requests) |
||
| 55 | } |
||
| 56 |
Loading history...