def setUp(self) -> None:
self.api_token = os.getenv("PUSHOVER_API_TOKEN")
self.user_key = os.getenv("PUSHOVER_USER_KEY")
self.assertIsNotNone(self.api_token, "You must export PUSHOVER_API_TOKEN as env variable")
self.assertIsNotNone(self.user_key, "You must export PUSHOVER_USER_KEY as env variable")
self.notifier = PushoverNotifier(
title_prefix="Test", config={"enable": True, "api_token": self.api_token, "user_key": self.user_key}
)
def testShowcaseBadNotifications(self):
notifiers = [
PushoverNotifier(
title_prefix="Harvester 1",
config={
"enable": True,
"api_token": self.api_token,
"user_key": self.user_key
},
),
PushoverNotifier(
title_prefix="Harvester 2",
config={
"enable": True,
"api_token": self.api_token,
"user_key": self.user_key
},
),
PushoverNotifier(
title_prefix="Harvester 3",
config={
"enable": True,
"api_token": self.api_token,
"user_key": self.user_key
},
),
]
disconnected_hdd = Event(
type=EventType.USER,
priority=EventPriority.HIGH,
service=EventService.HARVESTER,
message=
"Disconnected HDD? The total plot count decreased from 101 to 42.",
)
network_issues = Event(
type=EventType.USER,
priority=EventPriority.NORMAL,
service=EventService.HARVESTER,
message=
"Experiencing networking issues? Harvester did not participate in any "
"challenge for 120 seconds. It's now working again.",
)
offline = Event(
type=EventType.USER,
priority=EventPriority.HIGH,
service=EventService.HARVESTER,
message=
"Your harvester appears to be offline! No events for the past 712 seconds.",
)
events = [disconnected_hdd, offline, network_issues]
for notifier, event in zip(notifiers, events):
success = notifier.send_events_to_user(events=[event])
self.assertTrue(success)
class TestPushoverNotifier(unittest.TestCase):
def setUp(self) -> None:
api_token = os.getenv("PUSHOVER_API_TOKEN")
user_key = os.getenv("PUSHOVER_USER_KEY")
self.assertIsNotNone(
api_token, "You must export PUSHOVER_API_TOKEN as env variable")
self.assertIsNotNone(
user_key, "You must export PUSHOVER_USER_KEY as env variable")
self.notifier = PushoverNotifier(title_prefix="Test",
config={
"enable": True,
"api_token": api_token,
"user_key": user_key
})
def testLowPrioriyNotifications(self):
errors = self.notifier.send_events_to_user(events=[
Event(
type=EventType.USER,
priority=EventPriority.LOW,
service=EventService.HARVESTER,
message="Low priority notification 1.",
),
Event(
type=EventType.USER,
priority=EventPriority.LOW,
service=EventService.HARVESTER,
message="Low priority notification 2.",
),
])
self.assertFalse(errors)
def testNormalPrioriyNotifications(self):
errors = self.notifier.send_events_to_user(events=[
Event(
type=EventType.USER,
priority=EventPriority.NORMAL,
service=EventService.HARVESTER,
message="Normal priority notification.",
)
])
self.assertFalse(errors)
def testHighPrioriyNotifications(self):
errors = self.notifier.send_events_to_user(events=[
Event(
type=EventType.USER,
priority=EventPriority.HIGH,
service=EventService.HARVESTER,
message="This is a high priority notification!",
)
])
self.assertFalse(errors)
def testShowcaseGoodNotifications(self):
notifiers = [
PushoverNotifier(
title_prefix="Harvester 1",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
PushoverNotifier(
title_prefix="Harvester 2",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
PushoverNotifier(
title_prefix="Harvester 3",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
]
found_proof_event = Event(
type=EventType.USER,
priority=EventPriority.LOW,
service=EventService.HARVESTER,
message="Found 1 proof(s)!",
)
for notifier in notifiers:
errors = notifier.send_events_to_user(events=[found_proof_event])
self.assertFalse(errors)
class TestPushoverNotifier(unittest.TestCase):
def setUp(self) -> None:
self.api_token = os.getenv("PUSHOVER_API_TOKEN")
self.user_key = os.getenv("PUSHOVER_USER_KEY")
self.assertIsNotNone(self.api_token, "You must export PUSHOVER_API_TOKEN as env variable")
self.assertIsNotNone(self.user_key, "You must export PUSHOVER_USER_KEY as env variable")
self.notifier = PushoverNotifier(
title_prefix="Test", config={"enable": True, "api_token": self.api_token, "user_key": self.user_key}
)
@unittest.skipUnless(os.getenv("PUSHOVER_API_TOKEN"), "Run only if token available")
def testLowPriorityNotifications(self):
errors = self.notifier.send_events_to_user(
events=[
Event(
type=EventType.USER,
priority=EventPriority.LOW,
service=EventService.HARVESTER,
message="Low priority notification 1.",
),
Event(
type=EventType.USER,
priority=EventPriority.LOW,
service=EventService.HARVESTER,
message="Low priority notification 2.",
),
]
)
self.assertFalse(errors)
@unittest.skipUnless(os.getenv("PUSHOVER_API_TOKEN"), "Run only if token available")
def testNormalPriorityNotifications(self):
errors = self.notifier.send_events_to_user(
events=[
Event(
type=EventType.USER,
priority=EventPriority.NORMAL,
service=EventService.HARVESTER,
message="Normal priority notification.",
)
]
)
self.assertFalse(errors)
@unittest.skipUnless(os.getenv("PUSHOVER_API_TOKEN"), "Run only if token available")
def testHighPriorityNotifications(self):
errors = self.notifier.send_events_to_user(
events=[
Event(
type=EventType.USER,
priority=EventPriority.HIGH,
service=EventService.HARVESTER,
message="This is a high priority notification!",
)
]
)
self.assertFalse(errors)
@unittest.skipUnless(os.getenv("SHOWCASE_NOTIFICATIONS"), "Only for showcasing")
def testShowcaseGoodNotifications(self):
notifiers = [
PushoverNotifier(
title_prefix="Harvester 1",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
PushoverNotifier(
title_prefix="Harvester 2",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
PushoverNotifier(
title_prefix="Harvester 3",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
]
found_proof_event = Event(
type=EventType.USER,
priority=EventPriority.LOW,
service=EventService.HARVESTER,
message="Found 1 proof(s)!",
)
for notifier in notifiers:
errors = notifier.send_events_to_user(events=[found_proof_event])
self.assertFalse(errors)
@unittest.skipUnless(os.getenv("SHOWCASE_NOTIFICATIONS"), "Only for showcasing")
def testShowcaseBadNotifications(self):
notifiers = [
PushoverNotifier(
title_prefix="Harvester 1",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
PushoverNotifier(
title_prefix="Harvester 2",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
PushoverNotifier(
title_prefix="Harvester 3",
config={"enable": True, "api_token": self.api_token, "user_key": self.user_key},
),
]
disconnected_hdd = Event(
type=EventType.USER,
priority=EventPriority.HIGH,
service=EventService.HARVESTER,
message="Disconnected HDD? The total plot count decreased from 101 to 42.",
)
network_issues = Event(
type=EventType.USER,
priority=EventPriority.NORMAL,
service=EventService.HARVESTER,
message="Experiencing networking issues? Harvester did not participate in any "
"challenge for 120 seconds. It's now working again.",
)
offline = Event(
type=EventType.USER,
priority=EventPriority.HIGH,
service=EventService.HARVESTER,
message="Your harvester appears to be offline! No events for the past 712 seconds.",
)
events = [disconnected_hdd, offline, network_issues]
for notifier, event in zip(notifiers, events):
errors = notifier.send_events_to_user(events=[event])
self.assertFalse(errors)