github.com/ethanhsieh/snapd@v0.0.0-20210615102523-3db9b8e4edc5/daemon/export_access_test.go (about) 1 // -*- Mode: Go; indent-tabs-mode: t -*- 2 3 /* 4 * Copyright (C) 2021 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 "net/http" 24 25 "github.com/snapcore/snapd/polkit" 26 ) 27 28 type ( 29 AccessChecker = accessChecker 30 31 OpenAccess = openAccess 32 AuthenticatedAccess = authenticatedAccess 33 RootAccess = rootAccess 34 SnapAccess = snapAccess 35 ) 36 37 var CheckPolkitActionImpl = checkPolkitActionImpl 38 39 func MockCheckPolkitAction(new func(r *http.Request, ucred *Ucrednet, action string) *APIError) (restore func()) { 40 old := checkPolkitAction 41 checkPolkitAction = new 42 return func() { 43 checkPolkitAction = old 44 } 45 } 46 47 func MockPolkitCheckAuthorization(new func(pid int32, uid uint32, actionId string, details map[string]string, flags polkit.CheckFlags) (bool, error)) (restore func()) { 48 old := polkitCheckAuthorization 49 polkitCheckAuthorization = new 50 return func() { 51 polkitCheckAuthorization = old 52 } 53 }