rpc(mocks): add SetHeader method to server

asteris-llc · Nov 2, 2016 · a7ab883 · a7ab883
1 parent b1acfa6
commit a7ab883
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/rpc/pb/mocks/Grapher_GraphServer.go b/rpc/pb/mocks/Grapher_GraphServer.go
@@ -84,6 +84,20 @@ func (_m *GrapherGraphServer) SendMsg(m interface{}) error {
 	return r0
 }
 
+// SetHeader provides a mock function with given fields: _a0
+func (_m *GrapherGraphServer) SetHeader(_a0 metadata.MD) error {
+	ret := _m.Called(_a0)
+
+	var r0 error
+	if rf, ok := ret.Get(0).(func(metadata.MD) error); ok {
+		r0 = rf(_a0)
+	} else {
+		r0 = ret.Error(0)
+	}
+
+	return r0
+}
+
 // SetTrailer provides a mock function with given fields: _a0
 func (_m *GrapherGraphServer) SetTrailer(_a0 metadata.MD) {
 	_m.Called(_a0)