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')
