github.com/rigado/snapd@v2.42.5-go-mod+incompatible/daemon/api_debug_pprof_test.go (about) 1 // -*- Mode: Go; indent-tabs-mode: t -*- 2 3 /* 4 * Copyright (C) 2014-2019 Canonical Ltd 5 * 6 * This program is free software: you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License version 3 as 8 * published by the Free Software Foundation. 9 * 10 * This program is distributed in the hope that it will be useful, 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 * GNU General Public License for more details. 14 * 15 * You should have received a copy of the GNU General Public License 16 * along with this program. If not, see <http://www.gnu.org/licenses/>. 17 * 18 */ 19 20 package daemon 21 22 import ( 23 "bytes" 24 "io/ioutil" 25 "net/http" 26 "net/http/httptest" 27 28 "gopkg.in/check.v1" 29 ) 30 31 var _ = check.Suite(&pprofDebugSuite{}) 32 33 type pprofDebugSuite struct { 34 apiBaseSuite 35 } 36 37 func (s *pprofDebugSuite) TestGetPprofCmdline(c *check.C) { 38 req, err := http.NewRequest("GET", "/v2/debug/pprof/cmdline", nil) 39 c.Assert(err, check.IsNil) 40 41 rr := httptest.NewRecorder() 42 getPprof(debugPprofCmd, req, nil).ServeHTTP(rr, req) 43 44 rsp := rr.Result() 45 c.Assert(rsp, check.NotNil) 46 47 c.Assert(rsp.StatusCode, check.Equals, 200) 48 data, err := ioutil.ReadAll(rsp.Body) 49 c.Assert(err, check.IsNil) 50 51 cmdline, err := ioutil.ReadFile("/proc/self/cmdline") 52 c.Assert(err, check.IsNil) 53 cmdline = bytes.TrimRight(cmdline, "\x00") 54 c.Assert(string(data), check.DeepEquals, string(cmdline)) 55 }