Skip to content

Commit f218cd5

Browse files
[Release] Synchronize for release
1 parent c365ffb commit f218cd5

File tree

13 files changed

+58
-16
lines changed

13 files changed

+58
-16
lines changed

bank/backend/src/account_servicer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,17 @@
1313
WithdrawResponse,
1414
)
1515
from bank.v1.errors_pb2 import OverdraftError
16+
from reboot.aio.auth.authorizers import allow
1617
from reboot.aio.contexts import ReaderContext, WriterContext
1718

1819
logging.basicConfig(level=logging.INFO)
1920

2021

2122
class AccountServicer(Account.Servicer):
2223

24+
def authorizer(self):
25+
return allow()
26+
2327
async def Open(
2428
self,
2529
context: WriterContext,

bank/backend/src/bank_servicer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,17 @@
88
TransferRequest,
99
TransferResponse,
1010
)
11+
from reboot.aio.auth.authorizers import allow
1112
from reboot.aio.contexts import TransactionContext
1213

1314
logging.basicConfig(level=logging.INFO)
1415

1516

1617
class BankServicer(Bank.Servicer):
1718

19+
def authorizer(self):
20+
return allow()
21+
1822
async def SignUp(
1923
self,
2024
context: TransactionContext,

bank/backend/tests/account_servicer_test.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from account_servicer import AccountServicer
33
from bank.v1.account_rbt import Account, BalanceResponse
44
from bank.v1.errors_pb2 import OverdraftError
5+
from reboot.aio.applications import Application
56
from reboot.aio.tests import Reboot
67
from unittest import mock
78

@@ -23,7 +24,7 @@ async def asyncTearDown(self) -> None:
2324
async def test_basics(self) -> None:
2425
context = self.rbt.create_external_context(name=f"test-{self.id()}")
2526

26-
await self.rbt.up(servicers=[AccountServicer])
27+
await self.rbt.up(Application(servicers=[AccountServicer]))
2728

2829
# Create the state machine by calling its constructor. The fact that the
2930
# state machine _has_ a constructor means that this step is required
@@ -72,7 +73,7 @@ async def withdraw():
7273
@mock.patch("account_servicer.send_email")
7374
async def test_send_welcome_email(self, mock_send_email) -> None:
7475
await self.rbt.up(
75-
servicers=[AccountServicer],
76+
Application(servicers=[AccountServicer]),
7677
# Normally, `rbt.up()` runs the servicers in a separate process, to
7778
# ensure that accidental use of blocking functions (an easy mistake
7879
# to make) don't cause the test to lock up. However, in this test

bank/backend/tests/bank_servicer_test.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from bank.v1.bank_rbt import Bank, SignUpResponse
55
from bank.v1.errors_pb2 import OverdraftError
66
from bank_servicer import BankServicer
7+
from reboot.aio.applications import Application
78
from reboot.aio.tests import Reboot
89

910

@@ -17,7 +18,9 @@ async def asyncTearDown(self) -> None:
1718
await self.rbt.stop()
1819

1920
async def test_signup(self) -> None:
20-
await self.rbt.up(servicers=[BankServicer, AccountServicer])
21+
await self.rbt.up(
22+
Application(servicers=[BankServicer, AccountServicer])
23+
)
2124
context = self.rbt.create_external_context(name=f"test-{self.id()}")
2225
bank = Bank.ref("my-bank")
2326

@@ -34,7 +37,9 @@ async def test_signup(self) -> None:
3437
self.assertEqual(response.balance, 0)
3538

3639
async def test_transfer(self):
37-
await self.rbt.up(servicers=[BankServicer, AccountServicer])
40+
await self.rbt.up(
41+
Application(servicers=[BankServicer, AccountServicer])
42+
)
3843
context = self.rbt.create_external_context(name=f"test-{self.id()}")
3944
bank = Bank.ref("my-bank")
4045

hello-constructors/backend/src/hello_servicer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,15 @@
77
SendRequest,
88
SendResponse,
99
)
10+
from reboot.aio.auth.authorizers import allow
1011
from reboot.aio.contexts import ReaderContext, WriterContext
1112

1213

1314
class HelloServicer(Hello.Servicer):
1415

16+
def authorizer(self):
17+
return allow()
18+
1519
async def Create(
1620
self,
1721
context: WriterContext,

hello-constructors/backend/tests/hello_servicer_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import unittest
22
from hello_constructors.v1.hello_rbt import Hello
33
from hello_servicer import HelloServicer
4+
from reboot.aio.applications import Application
45
from reboot.aio.tests import Reboot
56

67

@@ -14,7 +15,7 @@ async def asyncTearDown(self) -> None:
1415
await self.rbt.stop()
1516

1617
async def test_hello_constructors(self) -> None:
17-
await self.rbt.up(servicers=[HelloServicer])
18+
await self.rbt.up(Application(servicers=[HelloServicer]))
1819

1920
context = self.rbt.create_external_context(name=f"test-{self.id()}")
2021

hello-legacy-grpc/backend/src/deprecated_greeter_servicer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,14 @@
22
import random
33
from google.protobuf.empty_pb2 import Empty
44
from hello_legacy_grpc.v1 import greeter_pb2, greeter_pb2_grpc
5+
from reboot.aio.auth.authorizers import allow
56

67

78
class DeprecatedGreeterServicer(greeter_pb2_grpc.DeprecatedGreeterServicer):
89

10+
def authorizer(self):
11+
return allow()
12+
913
async def Greet(
1014
self,
1115
request: greeter_pb2.GreetRequest,

hello-legacy-grpc/backend/src/proxy_greeter_servicer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,9 @@ async def Greet(
1515
# new Reboot service. This proxy servicer will forward traffic to
1616
# either the RebootGreeter or the DeprecatedGreeter with a 50/50
1717
# ratio.
18-
context = legacy_context.external_context(name=self.Greet.__name__)
18+
context = legacy_context.external_context(
19+
name="Call into `RebootGreeter`"
20+
)
1921
if random.random() < 0.5:
2022
# Route to RebootGreeter.
2123
reboot_greeter = RebootGreeter.ref("my-greeter")

hello-legacy-grpc/backend/src/reboot_greeter_servicer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
GreetResponse,
88
RebootGreeter,
99
)
10+
from reboot.aio.auth.authorizers import allow
1011
from reboot.aio.contexts import (
1112
Context,
1213
ReaderContext,
@@ -20,6 +21,9 @@
2021

2122
class RebootGreeterServicer(RebootGreeter.Servicer):
2223

24+
def authorizer(self):
25+
return allow()
26+
2327
async def _get_deprecated_salutation(self, context: Context) -> str:
2428
"""Fetch a salutation from the deprecated Greeter service written in
2529
legacy gRPC."""

hello-legacy-grpc/backend/tests/greeter_servicer_test.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from hello_legacy_grpc.v1 import greeter_pb2, greeter_pb2_grpc
55
from hello_legacy_grpc.v1.greeter_rbt import RebootGreeter
66
from proxy_greeter_servicer import ProxyGreeterServicer
7+
from reboot.aio.applications import Application
78
from reboot.aio.tests import Reboot
89
from reboot_greeter_servicer import RebootGreeterServicer
910

@@ -19,8 +20,10 @@ async def asyncTearDown(self) -> None:
1920

2021
async def test_reboot_greeter(self) -> None:
2122
await self.rbt.up(
22-
servicers=[RebootGreeterServicer],
23-
legacy_grpc_servicers=[DeprecatedGreeterServicer],
23+
Application(
24+
servicers=[RebootGreeterServicer],
25+
legacy_grpc_servicers=[DeprecatedGreeterServicer],
26+
),
2427
)
2528

2629
context = self.rbt.create_external_context(name=f"test-{self.id()}")
@@ -46,7 +49,9 @@ async def test_reboot_greeter(self) -> None:
4649
)
4750

4851
async def test_deprecated_greeter(self) -> None:
49-
await self.rbt.up(legacy_grpc_servicers=[DeprecatedGreeterServicer])
52+
await self.rbt.up(
53+
Application(legacy_grpc_servicers=[DeprecatedGreeterServicer])
54+
)
5055

5156
context = self.rbt.create_external_context(name=f"test-{self.id()}")
5257

@@ -66,10 +71,13 @@ async def test_deprecated_greeter(self) -> None:
6671
)
6772
async def test_proxy_greeter(self) -> None:
6873
await self.rbt.up(
69-
servicers=[RebootGreeterServicer],
70-
legacy_grpc_servicers=[
71-
DeprecatedGreeterServicer, ProxyGreeterServicer
72-
],
74+
Application(
75+
servicers=[RebootGreeterServicer],
76+
legacy_grpc_servicers=[
77+
DeprecatedGreeterServicer,
78+
ProxyGreeterServicer,
79+
],
80+
),
7381
local_envoy=True,
7482
)
7583

0 commit comments

Comments
 (0)