def testFlowAllWithSourceAndOrSinkShouldReturnAQSFilteredByTheSink(self): f = Flow(**self.args) qs_mock = f.all() sink_mock = Mock() source_mock = Mock() rv_mock = f.all(sink=sink_mock) self.assertEqual(((), {"sink": sink_mock.slug}), qs_mock.filter.call_args) qs_mock.reset_mock() rv_mock = f.all(source=source_mock) self.assertEqual(((), {"source": source_mock.slug}), qs_mock.filter.call_args) qs_mock.reset_mock() rv_mock = f.all(sink=sink_mock, source=source_mock) self.assertEqual(((), {"source": source_mock.slug}), qs_mock.filter.call_args) qs2_mock = qs_mock.filter.return_value self.assertEqual(((), {"sink": sink_mock.slug}), qs2_mock.filter.call_args)
def testFlowAllShouldReturnTheFlowQS(self): f = Flow(**self.args) self.assertEqual(f.queryset, f.all())