MockerResourceWithNetworkClient

Complexity

Total Complexity

5

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Importance

Changes	3
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	last_request()	0	2	1
A	set_response()	0	10	2
A	teardown_method()	0	5	1
A	setup_method()	0	15	1

# -*- coding: utf-8 -*-

import mock


class MockerResourceWithNetworkClient(object):

	resource_module = None

	mock_fluent_request_class = None

	@classmethod
	def setup_method(cls):

		# there are several similar modules-resources.
		# all we need to test them -- mock NetworkClient, to prevent
		# network access.
		# Let's do it with mock.patch.object
		#
		# Mocked client will be accessible in tests as `cls.client`

		# mock FluentRequest
		cls._patcher_fluent_request_class = mock.patch(
			'route4me.sdk._net.FluentRequest'
		)
		cls.mock_fluent_request_class = cls._patcher_fluent_request_class.start()

	@classmethod
	def teardown_method(cls):
		"""Teardown test environment for the entire class
		"""
		cls._patcher_fluent_request_class.stop()

	def set_response(self, status_code=200, data=None, **qw):

		x = mock.MagicMock()
		x.status_code = status_code
		x.json.return_value = data

		for k in qw:
			x.k = qw[k]

		self.mock_fluent_request_class.return_value.send.return_value = x

	def last_request(self):
		return self.mock_fluent_request_class.return_value