tests

Complexity

Total Complexity

9

Size/Duplication

Total Lines	65
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	CallableMock.has_been_called()	0	3	1
A	CallableMock.__init__()	0	6	1
A	CallableMock.return_value()	0	2	1
A	CallableMock.has_been_called_times()	0	3	1
A	CallableMock.__call__()	0	7	3
A	CallableMock.has_been_called_with()	0	15	2

# -*- coding: utf-8 -*-
# Copyright (C) 2018 Greenbone Networks GmbH
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import functools


class CallableMock:

    def __init__(self, func):
        # look like the function we are wrapping
        functools.update_wrapper(self, func)
        self.calls = []
        self.func = func
        self.result = None

    def __call__(self, *args, **kwargs):
        self.calls.append({'args': args, 'kwargs': kwargs})

        if self.result is None and not self.func is None:
            return self.func(self, *args, **kwargs)

        return self.result

    def return_value(self, value):
        self.result = value

    def has_been_called(self):
        assert len(self.calls) > 0, "{0} havn't been called.".format(
            self.func.__name__)

    def has_been_called_times(self, times):
        assert len(self.calls) == times, "{name} haven't been called {times}" \
            " times.".format(name=self.func.__name__, times=times)

    def has_been_called_with(self, *args, **kwargs):
        if len(self.calls) == 0:
            assert False

        lastcall = self.calls[-1]

        # not sure if this is correct
        assert lastcall['args'] == args and lastcall['kwargs'] == kwargs, \
            "Expected arguments {eargs} {ekwargs} of {name} don't match." \
            "Received: {rargs} {rkwargs}".format(
                name=self.func.__name__,
                eargs=args,
                ekwargs=kwargs,
                rargs=lastcall['args'],
                rkwargs=lastcall['kwargs']
            )