src/budget_sync/test/mock_bug.py

   1 from typing import Optional
   2
   3
   4 class MockBug:
   5     def __init__(self,
   6                  bug_id: int,
   7                  cf_budget_parent: Optional[int],
   8                  cf_budget: str,
   9                  cf_total_budget: str,
  10                  cf_nlnet_milestone: Optional[str],
  11                  cf_payees_list: str):
  12         self.id = bug_id
  13         if cf_budget_parent is not None:
  14             self.cf_budget_parent = cf_budget_parent
  15         self.cf_budget = cf_budget
  16         self.cf_total_budget = cf_total_budget
  17         if cf_nlnet_milestone is None:
  18             cf_nlnet_milestone = "---"
  19         self.cf_nlnet_milestone = cf_nlnet_milestone
  20         self.cf_payees_list = cf_payees_list
  21
  22     def __repr__(self):
  23         cf_budget_parent = getattr(self, "cf_budget_parent", None)
  24         return (f"MockBug(bug_id={self.id!r}, "
  25                 f"cf_budget_parent={cf_budget_parent!r}, "
  26                 f"cf_budget={self.cf_budget!r}, "
  27                 f"cf_total_budget={self.cf_total_budget!r}, "
  28                 f"cf_nlnet_milestone={self.cf_nlnet_milestone!r}, "
  29                 f"cf_payees_list={self.cf_payees_list!r})")