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_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_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]')