def save_and_get_complex():
'''Test saving entities containing mapped collection properties'''
entity1 = TestEntities.ComplexEntity()
entity1.setId("complex1")
strings = ArrayList()
strings.add("one")
strings.add("two")
entity1.setStringList(strings)
ints = HashSet()
ints.add(1)
ints.add(2)
entity1.setIntSet(ints)
extended = HashMap()
extended.put("prop1", "one")
extended.put("prop2", "two")
entity1.setExtendedProps(extended)
service = EntityService(TestEntities.ComplexEntity)
service.save(entity1)
entity2 = service.get("complex1")
assertNotNull(entity2)
assertEquals(entity2.getId(), entity1.getId())
assertTrue(entity2.getStringList().contains("one"))
assertTrue(entity2.getStringList().contains("two"))
assertTrue(entity2.getIntSet().contains(java.lang.Long(1)))
assertTrue(entity2.getIntSet().contains(java.lang.Long(2)))
assertNotNull(entity2.getExtendedProps())
assertEquals(entity2.getExtendedProps().get("prop1"), "one")
assertEquals(entity2.getExtendedProps().get("prop2"), "two")
def save_multiple():
'''Test saving multiple entities as a batch'''
entities = [
TestEntities.SimpleEntity("e1", "string1", 1, 1.1, 1.1, 1),
TestEntities.SimpleEntity("e2", "string2", 2, 2.2, 2.2, 2),
TestEntities.SimpleEntity("e3", "string3", 3, 3.3, 3.3, 3),
TestEntities.SimpleEntity("e4", "string4", 4, 4.4, 4.4, 4),
TestEntities.SimpleEntity("e5", "string5", 5, 5.5, 5.5, 5),
TestEntities.SimpleEntity("e6", "string6", 6, 6.6, 6.6, 6)
]
srv = EntityService(TestEntities.SimpleEntity)
srv.saveAll(entities)
saved = []
for e in entities:
saved.append(srv.get(e.getId()))
assertEquals(len(entities), len(saved))
for cnt in range(0, len(entities)):
assertEquals(entities[cnt].getId(), saved[cnt].getId())
assertEquals(entities[cnt].getStringProperty(),
saved[cnt].getStringProperty())
assertEquals(entities[cnt].getIntProperty(),
saved[cnt].getIntProperty())
assertEquals(entities[cnt].getFloatProperty(),
saved[cnt].getFloatProperty(), None, 0.0001)
assertEquals(entities[cnt].getDoubleProperty(),
saved[cnt].getDoubleProperty(), None, 0.0001)
assertEquals(entities[cnt].getLongProperty(),
saved[cnt].getLongProperty())
def save_timeout():
'''Test timeouts when saving recs'''
newent = TestEntities.SimpleEntity("ne1", "string one", 1, 1.1, 1.1, 1)
service = EntityService(TestEntities.SimpleEntity)
try:
# save with too short timeout
success = service.save(newent, 1)
assertFalse(success, "Expected save request to timeout")
except HBaseException, e:
fail("Save failed with unexpected exception")
def query_by_string():
srv = EntityService(TestEntities.IndexedEntity)
# test indexing of a value with multiple entries
q = srv.query()
q.using(Criteria.eq("stringProperty", java.lang.String('duck')))
matches = q.execute()
assertEquals(len(matches), 3)
assertEquals(matches[0].getId(), 'e1')
assertEquals(matches[0].getStringProperty(), 'duck')
assertEquals(matches[1].getId(), 'e2')
assertEquals(matches[1].getStringProperty(), 'duck')
assertEquals(matches[2].getId(), 'e3')
assertEquals(matches[2].getStringProperty(), 'duck')
q = srv.query()
q.using(Criteria.eq("stringProperty", java.lang.String('goose')))
matches = q.execute()
assertEquals(len(matches), 1)
assertEquals(matches[0].getId(), 'e4')
assertEquals(matches[0].getStringProperty(), 'goose')
def query_by_int():
srv = EntityService(TestEntities.IndexedEntity)
# test indexing of integer values
q = srv.query()
q.using(Criteria.eq("intKey", java.lang.Integer(2)))
matches = q.execute()
print matches
assertEquals(len(matches), 3)
assertEquals(matches[0].getId(), 'e4',
"Indexed entries should be in reverse timestamp order")
assertEquals(matches[0].getIntKey(), 2)
assertEquals(matches[1].getId(), 'e3',
"Indexed entries should be in reverse timestamp order")
assertEquals(matches[1].getIntKey(), 2)
assertEquals(matches[2].getId(), 'e2',
"Indexed entries should be in reverse timestamp order")
assertEquals(matches[2].getIntKey(), 2)
q = srv.query()
q.using(Criteria.eq("intKey", java.lang.Integer(1)))
matches = q.execute()
assertEquals(len(matches), 1)
assertEquals(matches[0].getId(), 'e1')
assertEquals(matches[0].getIntKey(), 1)
def setup():
hc.setUp()
HUtil.setPool(HTablePool(hc.conf, 5))
# create a dummy HBase table for testing
import db.hbase
admin = db.hbase.Admin(hc.conf)
if not admin.exists("test_indexed"):
admin.create("test_indexed", {"props:": {}})
if not admin.exists("test_indexed-by_intcol"):
admin.create("test_indexed-by_intcol", {"props:": {}, "__idx__:": {}})
if not admin.exists("test_indexed-by_stringcol"):
admin.create("test_indexed-by_stringcol", {
"props:": {},
"__idx__:": {}
})
srv = EntityService(TestEntities.IndexedEntity)
now = java.lang.System.currentTimeMillis()
srv.save(TestEntities.IndexedEntity("e1", "duck", 1, now - 100))
srv.save(TestEntities.IndexedEntity("e2", "duck", 2, now - 80))
srv.save(TestEntities.IndexedEntity("e3", "duck", 2, now - 60))
srv.save(TestEntities.IndexedEntity("e4", "goose", 2, now - 40))
def save_and_get():
entity1 = TestEntities.SimpleEntity()
entity1.setId("entity1")
entity1.setStringProperty("all my words")
entity1.setIntProperty(123)
entity1.setFloatProperty(1.1)
entity1.setDoubleProperty(123.456789)
entity1.setLongProperty(444444444444)
service = EntityService(TestEntities.SimpleEntity)
service.save(entity1)
entity2 = service.get("entity1")
assertNotNull(entity2)
assertEquals(entity2.getId(), entity1.getId())
assertEquals(entity2.getStringProperty(), entity1.getStringProperty())
assertEquals(entity2.getIntProperty(), entity1.getIntProperty())
assertEquals(entity2.getFloatProperty(), entity1.getFloatProperty(), None,
0.0001)
assertEquals(entity2.getDoubleProperty(), entity1.getDoubleProperty(),
None, 0.0001)
assertEquals(entity2.getLongProperty(), entity1.getLongProperty())
# update the retrieved entity and check for the changes
entity2.setStringProperty("new stuff")
entity2.setDoubleProperty(9.8765432101)
entity2.setIntProperty(-17)
service.save(entity2)
entity3 = service.get(entity2.getId())
assertEquals(entity3.getId(), "entity1")
assertEquals(entity3.getStringProperty(), "new stuff")
assertEquals(entity3.getIntProperty(), -17)
assertEquals(entity3.getFloatProperty(), 1.1, None, 0.0001)
assertEquals(entity3.getDoubleProperty(), 9.8765432101, None, 0.0001)
assertEquals(entity3.getLongProperty(), 444444444444)
