def testVariantAnnotationSetsSearch(self):
dataset = self.dataRepo.getDatasets()[0]
variantSet = dataset.getVariantSets()[0]
variantAnnotationSet = dataset.getVariantAnnotationSets()[0]
self.assertIsNotNone(variantAnnotationSet)
request = protocol.SearchVariantAnnotationSetsRequest()
request.variantSetId = "b4d=="
path = '/variantannotationsets/search'
response = self.sendJsonPostRequest(path, request.toJsonString())
responseData = protocol.SearchVariantAnnotationSetsResponse. \
fromJsonString(response.data)
self.assertTrue(responseData.validate(responseData.toJsonDict()))
self.assertIsNone(responseData.nextPageToken)
self.assertEqual([], responseData.variantAnnotationSets,
"There should no results for a bad ID")
request.variantSetId = variantSet.getId()
response = self.sendJsonPostRequest(path, request.toJsonString())
responseData = protocol.SearchVariantAnnotationSetsResponse. \
fromJsonString(response.data)
self.assertTrue(
protocol.SearchVariantAnnotationSetsResponse.validate(
responseData.toJsonDict()))
self.assertGreater(len(responseData.variantAnnotationSets), 0,
"Expect some results for a known good ID")
def testSearchVariantAnnotationSets(self):
request = protocol.SearchVariantAnnotationSetsRequest()
request.variantSetId = self.variantSetId
request.pageSize = self.pageSize
self.httpClient.searchVariantAnnotationSets(self.variantSetId)
self.httpClient._runSearchRequest.assert_called_once_with(
request, "variantannotationsets",
protocol.SearchVariantAnnotationSetsResponse)
def searchVariantAnnotationSets(self, variantSetId):
"""
Returns an iterator over the Annotation Sets fulfilling the specified
conditions from the specified variant set.
:param str variantSetId: The ID of the
:class:`ga4gh.protocol.VariantSet` of interest.
:return: An iterator over the :class:`ga4gh.protocol.AnnotationSet`
objects defined by the query parameters.
"""
request = protocol.SearchVariantAnnotationSetsRequest()
request.variant_set_id = variantSetId
request.page_size = pb.int(self._pageSize)
return self._runSearchRequest(
request, "variantannotationsets",
protocol.SearchVariantAnnotationSetsResponse)