def setUp(self):
self.triple1 = Triple()
self.triple1.subject = Variable("var1")
self.triple1.property = RDFS.type
self.triple1.object = DEB.Source
self.triple2 = Triple()
self.triple2.subject = URIRef("http://rdf.d.n/r")
self.triple2.property = RDFS.type
self.triple2.object = DEB.Binary
self.triple3 = Triple()
self.triple3.subject = Variable("v1")
self.triple3.property = Variable("v2")
self.triple3.object = Variable("v3")
self.v = QueryStringVisitor()
def setUp(self):
self.triple1 = Triple()
self.triple1.subject = Variable("var1")
self.triple1.property = RDFS.type
self.triple1.object = DEB.Source
self.triple2 = Triple()
self.triple2.subject = URIRef("http://rdf.d.n/r")
self.triple2.property = RDFS.type
self.triple2.object = DEB.Binary
self.triple3 = Triple()
self.triple3.subject = Variable("v1")
self.triple3.property = Variable("v2")
self.triple3.object = Variable("v3")
self.v = QueryStringVisitor()
class VisitorTest(unittest.TestCase):
def setUp(self):
self.triple1 = Triple()
self.triple1.subject = Variable("var1")
self.triple1.property = RDFS.type
self.triple1.object = DEB.Source
self.triple2 = Triple()
self.triple2.subject = URIRef("http://rdf.d.n/r")
self.triple2.property = RDFS.type
self.triple2.object = DEB.Binary
self.triple3 = Triple()
self.triple3.subject = Variable("v1")
self.triple3.property = Variable("v2")
self.triple3.object = Variable("v3")
self.v = QueryStringVisitor()
def test_visit_Variable(self):
expected = "?s"
self.assertEqual(expected, self.v.visit(Variable("s")))
def test_visit_URIRef(self):
uriref = URIRef("http://rdf.debian.net/r")
expected = "<http://rdf.debian.net/r>"
self.assertEqual(expected, self.v.visit(uriref))
def test_visit_Literal(self):
literal = Literal("Literal")
expected = '"Literal"'
self.assertEqual(expected, self.v.visit(literal))
def test_visit_Limit(self):
limit = Limit("10")
expected = "LIMIT 10"
self.assertEqual(expected, self.v.visit(limit))
def test_visit_Offset(self):
offset = Offset("10")
expected = "OFFSET 10"
self.assertEqual(expected, self.v.visit(offset))
def test_visit_Offset(self):
offset = Orderby(Variable("var"))
expected = "ORDER BY ?var"
self.assertEqual(expected, self.v.visit(offset))
def test_visit_Triple(self):
expected = "?var1 <http://www.w3.org/2000/01/rdf-schema#type> \
<http://idi.fundacionctic.org/steamy/debian.owl#Source>."
self.assertEqual(expected, self.v.visit(self.triple1))
expected = "<http://rdf.d.n/r> <http://www.w3.org/2000/01/rdf-schema#type> \
<http://idi.fundacionctic.org/steamy/debian.owl#Binary>."
self.assertEqual(expected, self.v.visit(self.triple2))
def test_visit_Optional(self):
optional = Optional(self.triple3)
expected = "OPTIONAL{?v1 ?v2 ?v3.}"
self.assertEqual(expected, self.v.visit(optional))
optional = Optional(self.triple3, self.triple3)
expected = "OPTIONAL{?v1 ?v2 ?v3.?v1 ?v2 ?v3.}"
self.assertEqual(expected, self.v.visit(optional))
def test_visit_Filter(self):
f1 = FunCall("regex", "arg1", "arg2")
f2 = FunCall("regex", "arg3", "arg4")
binexp = BinaryExpression(f1, "||", f2)
filter = Filter(binexp)
expected = "FILTER(regex(arg1,arg2)||regex(arg3,arg4))"
self.assertEqual(expected, self.v.visit(filter))
f = FunCall("bound", Variable("var"))
unexp = UnaryExpression(f, "!")
filter = Filter(unexp)
expected = "FILTER(!bound(?var))"
self.assertEqual(expected, self.v.visit(filter))
def test_visit_Union(self):
st1 = Triple(Variable("a"), Variable("b"), Variable("c"))
st2 = Triple(Variable("d"), Variable("e"), Variable("f"))
gp1 = [st1, st2]
gp2 = [st1]
union = Union()
union.graphpatterns.append(gp1)
union.graphpatterns.append(gp2)
expected = "{?a ?b ?c.?d ?e ?f.}UNION{?a ?b ?c.}"
self.assertEqual(expected, self.v.visit(union))
gp3 = [st1]
union.graphpatterns.append(gp3)
expected = "{?a ?b ?c.?d ?e ?f.}UNION{?a ?b ?c.}UNION{?a ?b ?c.}"
self.assertEqual(expected, self.v.visit(union))
def test_visit_BinaryExpression(self):
f1 = FunCall("regex", Variable("v1"), "r1")
f2 = FunCall("regex", Variable("v2"), "r2")
f3 = FunCall("regex", Variable("v3"), "r3")
b1 = BinaryExpression(f2, "||", f3)
b2 = BinaryExpression(f1, "||", b1)
expected = 'regex(?v1,r1)||regex(?v2,r2)||regex(?v3,r3)'
self.assertEqual(expected, self.v.visit(b2))
def test_visit_UnaryExpression(self):
u2 = UnaryExpression(Variable("var1"), "!")
u1 = UnaryExpression(u2, "!")
expected = "!!?var1"
self.assertEqual(expected, self.v.visit(u1))
def test_visit_FunCall(self):
fc1 = FunCall("symbol", Variable("v1"), '"string"')
expected = 'symbol(?v1,"string")'
self.assertEqual(expected, self.v.visit(fc1))
fc2 = FunCall("symbol")
expected = 'symbol()'
self.assertEqual(expected, self.v.visit(fc2))
def test_visit_SelectQuery(self):
st2 = Triple(Variable("d"), Variable("e"), Variable("f"))
helper = SelectQueryHelper()
expected = "SELECT WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
helper.set_limit("50")
expected = "SELECT WHERE{}LIMIT 50"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
helper.add_triple_variables(st2)
helper.add_filter(FunCall("regex", Variable("var"), '"regex"'))
expected = "SELECT?e?f?d WHERE{?d ?e ?f.FILTER(regex(?var,\"regex\"))}LIMIT 50"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
self.assertNotEqual(None, Parse(result))
helper.set_offset("2")
expected = "SELECT?e?f?d WHERE{?d ?e ?f.FILTER(regex(?var,\"regex\"))}LIMIT 50 OFFSET 2"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
self.assertNotEqual(None, Parse(result))
helper.add_optional(st2, st2)
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.add_union([st2, st2], [st2], [st2])
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.set_orderby("e")
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.set_orderby("e")
helper.query.modifiers = []
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.add_filter_notbound(Variable("e"))
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.add_or_filter_regex({Variable("a"): "r1", Variable("b"): "r2"})
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
def test_visit_SelectQueryFrom(self):
helper = SelectQueryHelper()
helper.set_from("http://example.com/graph")
expected = "SELECT FROM <http://example.com/graph> WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
helper.add_variable("var")
expected = "SELECT?var FROM <http://example.com/graph> WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
def test_visit_SelectQueryDistinct(self):
helper = SelectQueryHelper()
helper.set_distinct()
expected = "SELECT DISTINCT WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
def __str__(self):
v = QueryStringVisitor()
return v.visit(self.query)
def __str__(self):
v = QueryStringVisitor()
return v.visit(self.query)
class VisitorTest(unittest.TestCase):
def setUp(self):
self.triple1 = Triple()
self.triple1.subject = Variable("var1")
self.triple1.property = RDFS.type
self.triple1.object = DEB.Source
self.triple2 = Triple()
self.triple2.subject = URIRef("http://rdf.d.n/r")
self.triple2.property = RDFS.type
self.triple2.object = DEB.Binary
self.triple3 = Triple()
self.triple3.subject = Variable("v1")
self.triple3.property = Variable("v2")
self.triple3.object = Variable("v3")
self.v = QueryStringVisitor()
def test_visit_Variable(self):
expected = "?s"
self.assertEqual(expected, self.v.visit(Variable("s")))
def test_visit_URIRef(self):
uriref = URIRef("http://rdf.debian.net/r")
expected = "<http://rdf.debian.net/r>"
self.assertEqual(expected, self.v.visit(uriref))
def test_visit_Literal(self):
literal = Literal("Literal")
expected = '"Literal"'
self.assertEqual(expected, self.v.visit(literal))
def test_visit_Limit(self):
limit = Limit("10")
expected = "LIMIT 10"
self.assertEqual(expected, self.v.visit(limit))
def test_visit_Offset(self):
offset = Offset("10")
expected = "OFFSET 10"
self.assertEqual(expected, self.v.visit(offset))
def test_visit_Offset(self):
offset = Orderby(Variable("var"))
expected = "ORDER BY ?var"
self.assertEqual(expected, self.v.visit(offset))
def test_visit_Triple(self):
expected = "?var1 <http://www.w3.org/2000/01/rdf-schema#type> \
<http://idi.fundacionctic.org/steamy/debian.owl#Source>."
self.assertEqual(expected, self.v.visit(self.triple1))
expected = "<http://rdf.d.n/r> <http://www.w3.org/2000/01/rdf-schema#type> \
<http://idi.fundacionctic.org/steamy/debian.owl#Binary>."
self.assertEqual(expected, self.v.visit(self.triple2))
def test_visit_Optional(self):
optional = Optional(self.triple3)
expected = "OPTIONAL{?v1 ?v2 ?v3.}"
self.assertEqual(expected, self.v.visit(optional))
optional = Optional(self.triple3, self.triple3)
expected = "OPTIONAL{?v1 ?v2 ?v3.?v1 ?v2 ?v3.}"
self.assertEqual(expected, self.v.visit(optional))
def test_visit_Filter(self):
f1 = FunCall("regex", "arg1", "arg2")
f2 = FunCall("regex", "arg3", "arg4")
binexp = BinaryExpression(f1, "||", f2)
filter = Filter(binexp)
expected = "FILTER(regex(arg1,arg2)||regex(arg3,arg4))"
self.assertEqual(expected, self.v.visit(filter))
f = FunCall("bound", Variable("var"))
unexp = UnaryExpression(f, "!")
filter = Filter(unexp)
expected = "FILTER(!bound(?var))"
self.assertEqual(expected, self.v.visit(filter))
def test_visit_Union(self):
st1 = Triple(Variable("a"), Variable("b"), Variable("c"))
st2 = Triple(Variable("d"), Variable("e"), Variable("f"))
gp1 = [st1, st2]
gp2 = [st1]
union = Union()
union.graphpatterns.append(gp1)
union.graphpatterns.append(gp2)
expected = "{?a ?b ?c.?d ?e ?f.}UNION{?a ?b ?c.}"
self.assertEqual(expected, self.v.visit(union))
gp3 = [st1]
union.graphpatterns.append(gp3)
expected = "{?a ?b ?c.?d ?e ?f.}UNION{?a ?b ?c.}UNION{?a ?b ?c.}"
self.assertEqual(expected, self.v.visit(union))
def test_visit_BinaryExpression(self):
f1 = FunCall("regex", Variable("v1"), "r1")
f2 = FunCall("regex", Variable("v2"), "r2")
f3 = FunCall("regex", Variable("v3"), "r3")
b1 = BinaryExpression(f2, "||", f3)
b2 = BinaryExpression(f1, "||", b1)
expected = 'regex(?v1,r1)||regex(?v2,r2)||regex(?v3,r3)'
self.assertEqual(expected, self.v.visit(b2))
def test_visit_UnaryExpression(self):
u2 = UnaryExpression(Variable("var1"), "!")
u1 = UnaryExpression(u2, "!")
expected = "!!?var1"
self.assertEqual(expected, self.v.visit(u1))
def test_visit_FunCall(self):
fc1 = FunCall("symbol", Variable("v1"), '"string"')
expected = 'symbol(?v1,"string")'
self.assertEqual(expected, self.v.visit(fc1))
fc2 = FunCall("symbol")
expected = 'symbol()'
self.assertEqual(expected, self.v.visit(fc2))
def test_visit_SelectQuery(self):
st2 = Triple(Variable("d"), Variable("e"), Variable("f"))
helper = SelectQueryHelper()
expected = "SELECT WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
helper.set_limit("50")
expected = "SELECT WHERE{}LIMIT 50"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
helper.add_triple_variables(st2)
helper.add_filter(FunCall("regex", Variable("var"), '"regex"'))
expected = "SELECT?e?f?d WHERE{?d ?e ?f.FILTER(regex(?var,\"regex\"))}LIMIT 50"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
self.assertNotEqual(None, Parse(result))
helper.set_offset("2")
expected = "SELECT?e?f?d WHERE{?d ?e ?f.FILTER(regex(?var,\"regex\"))}LIMIT 50 OFFSET 2"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
self.assertNotEqual(None, Parse(result))
helper.add_optional(st2, st2)
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.add_union([st2, st2], [st2], [st2])
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.set_orderby("e")
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.set_orderby("e")
helper.query.modifiers = []
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.add_filter_notbound(Variable("e"))
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
helper.add_or_filter_regex({Variable("a"): "r1", Variable("b"): "r2"})
result = self.v.visit(helper.query, True)
self.assertNotEqual(None, Parse(result))
def test_visit_SelectQueryFrom(self):
helper = SelectQueryHelper()
helper.set_from("http://example.com/graph")
expected = "SELECT FROM <http://example.com/graph> WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
helper.add_variable("var")
expected = "SELECT?var FROM <http://example.com/graph> WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
def test_visit_SelectQueryDistinct(self):
helper = SelectQueryHelper()
helper.set_distinct()
expected = "SELECT DISTINCT WHERE{}"
result = self.v.visit(helper.query)
self.assertEqual(expected, result)
