github.com/hugh712/snapd@v0.0.0-20200910133618-1a99902bd583/timings/export_test.go (about) 1 // -*- Mode: Go; indent-tabs-mode: t -*- 2 3 /* 4 * Copyright (C) 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 timings 21 22 import ( 23 "time" 24 ) 25 26 func MockTimeDuration(durationFunc func(t1, t2 time.Time) time.Duration) func() { 27 old := timeDuration 28 timeDuration = durationFunc 29 return func() { 30 timeDuration = old 31 } 32 } 33 34 func MockTimeNow(nowFunc func() time.Time) func() { 35 old := timeNow 36 timeNow = nowFunc 37 return func() { 38 timeNow = old 39 } 40 }