gitee.com/mysnapcore/mysnapd@v0.1.0/cmd/snap-preseed/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 main 21 22 import "gitee.com/mysnapcore/mysnapd/testutil" 23 24 var ( 25 Run = run 26 ) 27 28 func MockOsGetuid(f func() int) (restore func()) { 29 r := testutil.Backup(&osGetuid) 30 osGetuid = f 31 return r 32 } 33 34 func MockPreseedCore20(f func(dir, key, aaDir string) error) (restore func()) { 35 r := testutil.Backup(&preseedCore20) 36 preseedCore20 = f 37 return r 38 } 39 40 func MockPreseedClassic(f func(dir string) error) (restore func()) { 41 r := testutil.Backup(&preseedClassic) 42 preseedClassic = f 43 return r 44 } 45 46 func MockResetPreseededChroot(f func(dir string) error) (restore func()) { 47 r := testutil.Backup(&preseedResetPreseededChroot) 48 preseedResetPreseededChroot = f 49 return r 50 }