def setUp(self): yield self._start_container() services = [{'name':'reference1','module':'ion.play.rdf_store.reference_service','class':'ReferenceService'},] sup = yield self._spawn_processes(services) self.rsc = ReferenceServiceClient(proc=sup) self.key='key1' self.ref1='ref1' self.ref2='ref2' self.ref3='ref3' self.ref4='ref4'
class ReferenceTest(IonTestCase): """Testing service classes of resource registry """ @defer.inlineCallbacks def setUp(self): yield self._start_container() services = [{'name':'reference1','module':'ion.play.rdf_store.reference_service','class':'ReferenceService'},] sup = yield self._spawn_processes(services) self.rsc = ReferenceServiceClient(proc=sup) self.key='key1' self.ref1='ref1' self.ref2='ref2' self.ref3='ref3' self.ref4='ref4' @defer.inlineCallbacks def tearDown(self): yield self._stop_container() @defer.inlineCallbacks def test_put_get_delete(self): res = yield self.rsc.add_reference(self.key,self.ref1) self.assertEqual(res,self.key) res = yield self.rsc.add_reference(self.key,self.ref2) res = yield self.rsc.add_reference(self.key,self.ref4) res = yield self.rsc.add_reference(self.key,self.ref3) res = yield self.rsc.get_references(self.key) self.assertEqual(set(res),set([self.ref1,self.ref2,self.ref3,self.ref4])) res = yield self.rsc.del_reference(self.key,self.ref1) self.assertEqual(res,'success') res = yield self.rsc.get_references(self.key) self.assertEqual(set(res),set([self.ref2,self.ref3,self.ref4]))