gitee.com/mysnapcore/mysnapd@v0.1.0/daemon/export_api_quotas_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 "gitee.com/mysnapcore/mysnapd/client" 24 "gitee.com/mysnapcore/mysnapd/overlord/servicestate" 25 "gitee.com/mysnapcore/mysnapd/overlord/state" 26 "gitee.com/mysnapcore/mysnapd/snap/quota" 27 ) 28 29 type ( 30 PostQuotaGroupData = postQuotaGroupData 31 ) 32 33 func MockServicestateCreateQuota(f func(st *state.State, name string, parentName string, snaps []string, resourceLimits quota.Resources) (*state.TaskSet, error)) func() { 34 old := servicestateCreateQuota 35 servicestateCreateQuota = f 36 return func() { 37 servicestateCreateQuota = old 38 } 39 } 40 41 func MockServicestateUpdateQuota(f func(st *state.State, name string, opts servicestate.QuotaGroupUpdate) (*state.TaskSet, error)) func() { 42 old := servicestateUpdateQuota 43 servicestateUpdateQuota = f 44 return func() { 45 servicestateUpdateQuota = old 46 } 47 } 48 49 func MockServicestateRemoveQuota(f func(st *state.State, name string) (*state.TaskSet, error)) func() { 50 old := servicestateRemoveQuota 51 servicestateRemoveQuota = f 52 return func() { 53 servicestateRemoveQuota = old 54 } 55 } 56 57 func MockGetQuotaUsage(f func(grp *quota.Group) (*client.QuotaValues, error)) (restore func()) { 58 old := getQuotaUsage 59 getQuotaUsage = f 60 return func() { 61 getQuotaUsage = old 62 } 63 }