/*

* Copyright 2018 The CovenantSQL Authors.

*

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*/

package utils

import (

"os"

"testing"

. "github.com/smartystreets/goconvey/convey"

)

const (

cpuProfile = ".cpuProfile"

memProfile = ".memProfile"

)

func TestStartStopProfile(t *testing.T) {

Convey("start normally", t, func() {

defer os.Remove(cpuProfile)

defer os.Remove(memProfile)

err := StartProfile(cpuProfile, memProfile)

So(err, ShouldBeNil)

StopProfile()

cpuFileInfo, err := os.Stat(cpuProfile)

So(cpuFileInfo.Size(), ShouldBeGreaterThan, 0)

So(err, ShouldBeNil)

memFileInfo, err := os.Stat(memProfile)

So(memFileInfo.Size(), ShouldBeGreaterThan, 0)

So(err, ShouldBeNil)

})

Convey("start empty", t, func() {

err := StartProfile("", "")

So(err, ShouldBeNil)

StopProfile()

})

Convey("start no such path", t, func() {

err := StartProfile("/not/exist/path", "")

So(err, ShouldNotBeNil)

StopProfile()

err = StartProfile("", "/not/exist/path")

So(err, ShouldNotBeNil)

StopProfile()

})

}