def test_send_inserted_session_provider(self, send_serie):
send_mass([1], [{}], method=self.method, config=self.config, session="ProcessorMock")
args, kwargs = send_serie.call_args
self.assertEqual(args, ([1], [{}],))
self.assertEqual(kwargs["method"], self.method)
self.assertEqual(kwargs["config"], self.config)
self.assertTrue(kwargs["session"].from_provider)
def test_send_mass_concat_arguments(self, send_serie):
self.config.concat_args_size = 3
self.config.concat_args_symbol = "|"
send_mass(
[[1], [2], [3], [4], [5, 5], [6], [7]],
[{}, {}, {}, {}, {}, {}, {}],
method=self.method,
config=self.config,
session=MockRequests
)
send_serie.assert_called_with(
[["1|2|3"], ["4|5|5|6"], ["7"]], [{}, {}, {}],
method=self.method,
config=self.config,
session=MockRequests
)
def test_send_mass_intervals(self, sleep_mock):
# First with an interval
self.config.interval_duration = 250 # 0.25 secs
args_list = self.get_args_list(["test", "test2"])
kwargs_list = self.get_kwargs_list(["test", "test2"])
scc, err = send_mass(args_list, kwargs_list, method=self.method, config=self.config, session=MockRequests)
self.assertEqual(sleep_mock.call_args_list, [call(0), call(0.25), call(0), call(0.25)])
self.check_results(scc, 2)
self.check_results(err, 0)
# Now without an interval
sleep_mock.reset_mock()
self.config.interval_duration = 0
args_list = self.get_args_list(["test", "test2"])
kwargs_list = self.get_kwargs_list(["test", "test2"])
scc, err = send_mass(args_list, kwargs_list, method=self.method, config=self.config, session=MockRequests)
self.assertFalse(sleep_mock.called)
self.check_results(scc, 2)
self.check_results(err, 0)
def test_send_mass(self):
args_list = self.get_args_list(["test", "test2", "404"])
kwargs_list = self.get_kwargs_list(["test", "test2", "404"])
start = datetime.now()
scc, err = send_mass(args_list, kwargs_list, method=self.method, config=self.config, session=MockRequests)
end = datetime.now()
duration = (end - start).total_seconds()
self.assertLess(duration, 0.1)
self.check_results(scc, 2)
self.check_results(err, 1)
def test_send_mass_continuation(self):
self.config.continuation_limit = 10
args_list = self.get_args_list(["test", "next", "404"])
kwargs_list = self.get_kwargs_list(["test", "next", "404"])
scc, err = send_mass(
args_list,
kwargs_list,
method=self.method,
config=self.config,
session=MockRequests
)
self.check_results(scc, 3)
self.check_results(err, 1)