FilesExpand file tree

 main

/

test_eventwebhook.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

42 lines (34 loc) · 1.78 KB

 main

/

test_eventwebhook.py

Copy path

Top

File metadata and controls

• Code
• Blame

42 lines (34 loc) · 1.78 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

import json

import unittest

from sendgrid import EventWebhook

class UnitTests(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.PUBLIC_KEY = 'MFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcDQgAEEDr2LjtURuePQzplybdC+u4CwrqDqBaWjcMMsTbhdbcwHBcepxo7yAQGhHPTnlvFYPAZFceEu/1FwCM/QmGUhA=='

cls.SIGNATURE = 'MEUCIQCtIHJeH93Y+qpYeWrySphQgpNGNr/U+UyUlBkU6n7RAwIgJTz2C+8a8xonZGi6BpSzoQsbVRamr2nlxFDWYNH2j/0='

cls.TIMESTAMP = '1588788367'

cls.PAYLOAD = json.dumps({

'event': 'test_event',

'category': 'example_payload',

'message_id': 'message_id',

}, sort_keys=True, separators=(',', ':'))

def test_verify_valid_signature(self):

ew = EventWebhook()

key = ew.convert_public_key_to_ecdsa(self.PUBLIC_KEY)

self.assertTrue(ew.verify_signature(self.PAYLOAD, self.SIGNATURE, self.TIMESTAMP, key))

def test_verify_bad_key(self):

ew = EventWebhook('MFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcDQgAEqTxd43gyp8IOEto2LdIfjRQrIbsd4SXZkLW6jDutdhXSJCWHw8REntlo7aNDthvj+y7GjUuFDb/R1NGe1OPzpA==')

self.assertFalse(ew.verify_signature(self.PAYLOAD, self.SIGNATURE, self.TIMESTAMP))

def test_verify_bad_payload(self):

ew = EventWebhook(self.PUBLIC_KEY)

self.assertFalse(ew.verify_signature('payload', self.SIGNATURE, self.TIMESTAMP))

def test_verify_bad_signature(self):

ew = EventWebhook(self.PUBLIC_KEY)

self.assertFalse(ew.verify_signature(

self.PAYLOAD,

'MEUCIQCtIHJeH93Y+qpYeWrySphQgpNGNr/U+UyUlBkU6n7RAwIgJTz2C+8a8xonZGi6BpSzoQsbVRamr2nlxFDWYNH3j/0=',

self.TIMESTAMP

))

def test_verify_bad_timestamp(self):

ew = EventWebhook(self.PUBLIC_KEY)

self.assertFalse(ew.verify_signature(self.PAYLOAD, self.SIGNATURE, 'timestamp'))