def test_wdq_to_wdqs_detect_tree(self):
result = wdq_to_wdqs('TREE[1][2][3]')
self.mock_string_wdqs_search.assert_not_called()
self.mock_tree_wdqs_search.assert_called_once_with('1', '2', '3')
self.mock_claim_wdqs_search.assert_not_called()
self.mock_claim_qualifiers_sparql.assert_not_called()
self.mock_sanitize_to_wdq_result.assert_called_once_with('tree_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_tree(self):
result = wdq_to_wdqs('TREE[1][2][3]')
self.mock_string_wdqs_search.assert_not_called()
self.mock_tree_wdqs_search.assert_called_once_with('1', '2', '3')
self.mock_claim_wdqs_search.assert_not_called()
self.mock_claim_qualifiers_sparql.assert_not_called()
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'tree_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_string(self):
result = wdq_to_wdqs('STRING[123:"test"]')
self.assertEqual(result, 'sanitized_data')
self.mock_string_wdqs_search.assert_called_once_with('123', 'test')
self.mock_tree_wdqs_search.assert_not_called()
self.mock_claim_wdqs_search.assert_not_called()
self.mock_claim_qualifiers_sparql.assert_not_called()
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'string_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_string(self):
result = wdq_to_wdqs('STRING[123:"test"]')
self.assertEqual(result, 'sanitized_data')
self.mock_string_wdqs_search.assert_called_once_with('123', 'test')
self.mock_tree_wdqs_search.assert_not_called()
self.mock_claim_wdqs_search.assert_not_called()
self.mock_claim_qualifiers_sparql.assert_not_called()
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'string_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_claim_w_item(self):
result = wdq_to_wdqs('CLAIM[123:456]')
self.mock_string_wdqs_search.assert_not_called()
self.mock_tree_wdqs_search.assert_not_called()
self.mock_claim_wdqs_search.assert_called_once_with(
'123', q_value='456', qualifiers=None)
self.mock_claim_qualifiers_sparql.assert_not_called()
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'claim_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_claim_w_qualifiers(self):
result = wdq_to_wdqs('CLAIM[123:456]{STRING[bad] OR CLAIM[good,bad]}')
self.mock_string_wdqs_search.assert_not_called()
self.mock_tree_wdqs_search.assert_not_called()
self.mock_claim_wdqs_search.assert_called_once_with(
'123', q_value='456', qualifiers='claim_qualifier')
self.mock_claim_qualifiers_sparql.assert_called_once_with(
'123', '{STRING[bad] OR CLAIM[good,bad]}')
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'claim_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_claim_w_item(self):
result = wdq_to_wdqs('CLAIM[123:456]')
self.mock_string_wdqs_search.assert_not_called()
self.mock_tree_wdqs_search.assert_not_called()
self.mock_claim_wdqs_search.assert_called_once_with('123',
q_value='456',
qualifiers=None)
self.mock_claim_qualifiers_sparql.assert_not_called()
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'claim_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_detect_claim_w_qualifiers(self):
result = wdq_to_wdqs(
'CLAIM[123:456]{STRING[bad] OR CLAIM[good,bad]}')
self.mock_string_wdqs_search.assert_not_called()
self.mock_tree_wdqs_search.assert_not_called()
self.mock_claim_wdqs_search.assert_called_once_with(
'123', q_value='456', qualifiers='claim_qualifier')
self.mock_claim_qualifiers_sparql.assert_called_once_with(
'123', '{STRING[bad] OR CLAIM[good,bad]}')
self.mock_sanitize_to_wdq_result.assert_called_once_with(
'claim_result')
self.assertEqual(result, 'sanitized_data')
def test_wdq_to_wdqs_other_type_raises_error(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('AROUND[test]')
def test_wdq_to_wdqs_cannot_handle_multiple_claims_tree(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('TREE[1][2][3] AND something')
def test_wdq_to_wdqs_cannot_handle_multiple_claims_string(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('STRING[123:"test"] AND something')
def test_wdq_to_wdqs_cannot_handle_multiple_claims(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('CLAIM[123:456,789:0] AND STRING[1:"test"]')
def test_wdq_to_wdqs_comma_in_claim_not_ok(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('CLAIM[123:456,789:0]')
def test_wdq_to_wdqs_other_type_raises_error(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('AROUND[test]')
def test_wdq_to_wdqs_cannot_handle_multiple_claims_tree(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('TREE[1][2][3] AND something')
def test_wdq_to_wdqs_cannot_handle_multiple_claims_string(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('STRING[123:"test"] AND something')
def test_wdq_to_wdqs_cannot_handle_multiple_claims(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('CLAIM[123:456,789:0] AND STRING[1:"test"]')
def test_wdq_to_wdqs_comma_in_claim_not_ok(self):
with self.assertRaises(NotImplementedError):
wdq_to_wdqs('CLAIM[123:456,789:0]')
