def test_acknowledge_cbsd_update(self):
cbsd = self._prepare_base_cbsd(). \
updated(). \
build()
self.session.add(cbsd)
self.session.commit()
self.stub.AcknowledgeCbsdUpdate(
AcknowledgeCbsdUpdateRequest(id=SOME_ID), )
self.assertFalse(cbsd.should_deregister)
def test_acknowledge_non_existent_cbsd_update(self):
with self.assertRaises(grpc.RpcError) as err:
self.stub.AcknowledgeCbsdUpdate(
AcknowledgeCbsdUpdateRequest(id=SOME_ID), )
self.assertEqual(grpc.StatusCode.NOT_FOUND, err.exception.code())