Skip to content

Commit 429d20d

Browse files
committed
feat: added validation of authenticate
1 parent 14cb6ae commit 429d20d

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

src/balance_service/interfaces/nubank.py

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,10 @@ def authenticate(self,
1212
certificate_path: str):
1313
pass
1414

15+
@abstractmethod
16+
def has_certificate(self, certificate_path: str):
17+
pass
18+
1519
@abstractmethod
1620
def get_balance(self):
1721
pass
@@ -25,8 +29,15 @@ def __init__(self,
2529
self.token = token
2630
self.certificate_path = certificate_path
2731
self.bank_service = bank_service
28-
self.bank_service.authenticate(self.token, self.certificate_path)
32+
self.has_certificate = self.bank_service.has_certificate(self.certificate_path)
2933

30-
def get_balance(self):
31-
return self.bank_service.get_balance()
34+
if self.has_certificate:
35+
self.bank_service.authenticate(
36+
self.token,
37+
self.certificate_path
38+
)
3239

40+
def get_balance(self):
41+
if self.has_certificate:
42+
return self.bank_service.get_balance()
43+
return 0

0 commit comments

Comments
 (0)