github.com/mysteriumnetwork/node@v0.0.0-20240516044423-365054f76801/firewall/iptables_exec_mock.go (about)

     1  /*
     2   * Copyright (C) 2020 The "MysteriumNetwork/node" Authors.
     3   *
     4   * This program is free software: you can redistribute it and/or modify
     5   * it under the terms of the GNU General Public License as published by
     6   * the Free Software Foundation, either version 3 of the License, or
     7   * (at your option) any later version.
     8   *
     9   * This program is distributed in the hope that it will be useful,
    10   * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12   * GNU General Public License for more details.
    13   *
    14   * You should have received a copy of the GNU General Public License
    15   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    16   */
    17  
    18  package firewall
    19  
    20  import (
    21  	"strings"
    22  )
    23  
    24  type iptablesExecResult struct {
    25  	called bool
    26  	output []string
    27  	err    error
    28  }
    29  
    30  type iptablesExecMock struct {
    31  	mocks map[string]iptablesExecResult
    32  }
    33  
    34  func (iem *iptablesExecMock) Exec(args ...string) ([]string, error) {
    35  	key := argsToKey(args...)
    36  	res := iem.mocks[key]
    37  	res.called = true
    38  	iem.mocks[key] = res
    39  	return res.output, res.err
    40  }
    41  
    42  func (iem *iptablesExecMock) VerifyCalledWithArgs(args ...string) bool {
    43  	key := argsToKey(args...)
    44  	return iem.mocks[key].called
    45  }
    46  
    47  func argsToKey(args ...string) string {
    48  	return strings.Join(args, " ")
    49  }