def test_handle_extracted_action(self, mock_origin): mock_origin.return_value = False obj = SnollaWorker(self.cfg, None, None) # Check False self.assertFalse(obj.check_allowed_origins(self.commit)) mock_origin.assert_called_once_with(self.commit['origin'], self.cfg['general']['allowed_origins']) # Check True mock_origin.reset_mock() mock_origin.return_value = True self.assertTrue(obj.check_allowed_origins(self.commit)) mock_origin.assert_called_once_with(self.commit['origin'], self.cfg['general']['allowed_origins'])