def test_query_with_not_equal_filter_raises_type_error(self) -> None: query = datastore_services.Query(filters=BarModel.prop != 1) with self.assertRaisesRegexp( TypeError, 'forbidden filter'): # type: ignore[no-untyped-call] job_utils.get_beam_query_from_ndb_query(query)
def test_query_with_range_like_filter(self) -> None: query = datastore_services.Query(filters=datastore_services.all_of( BarModel.prop >= 3, BarModel.prop < 6)) beam_query = job_utils.get_beam_query_from_ndb_query(query) self.assertEqual(beam_query.filters, (('prop', '>=', 3), ('prop', '<', 6)))
def test_query_with_filter(self) -> None: query = datastore_services.Query(filters=BarModel.prop >= 3) beam_query = job_utils.get_beam_query_from_ndb_query(query) self.assertEqual(beam_query.filters, (('prop', '>=', 3), ))
def test_query_with_namespace(self) -> None: query = datastore_services.Query(namespace='abc') beam_query = job_utils.get_beam_query_from_ndb_query(query) self.assertEqual(beam_query.namespace, 'abc')