Unverified Commit b0c74f0a authored by Philipp Winter's avatar Philipp Winter
Browse files

Turn str into bytes in SMTP unit test.

This should fix the following unit test errors:

[ERROR]
Traceback (most recent call last):
  File "/home/travis/build/NullHypothesis/bridgedb/bridgedb/test/test_smtp.py", line 155, in test_getBridges
    self.server.getAndCheckMessageContains("Here are your bridges")
  File "/home/travis/build/NullHypothesis/bridgedb/bridgedb/test/test_smtp.py", line 99, in getAndCheckMessageContains
    assert message.find(text) != -1, ("Message did not contain text '%s'."
builtins.TypeError: argument should be integer or bytes-like object, not 'str'

bridgedb.test.test_smtp.SMTPTests.test_getBridges
===============================================================================
[ERROR]
Traceback (most recent call last):
  File "/home/travis/build/NullHypothesis/bridgedb/bridgedb/test/test_smtp.py", line 171, in test_getBridges_rateLimitExceeded
    self.server.getAndCheckMessageContains("Here are your bridges")
  File "/home/travis/build/NullHypothesis/bridgedb/bridgedb/test/test_smtp.py", line 99, in getAndCheckMessageContains
    assert message.find(text) != -1, ("Message did not contain text '%s'."
builtins.TypeError: argument should be integer or bytes-like object, not 'str'

bridgedb.test.test_smtp.SMTPTests.test_getBridges_rateLimitExceeded
===============================================================================
[ERROR]
Traceback (most recent call last):
  File "/home/travis/build/NullHypothesis/bridgedb/bridgedb/test/test_smtp.py", line 206, in test_getBridges_stressTest
    self.server.getAndCheckMessageContains("Here are your bridges")
  File "/home/travis/build/NullHypothesis/bridgedb/bridgedb/test/test_smtp.py", line 99, in getAndCheckMessageContains
    assert message.find(text) != -1, ("Message did not contain text '%s'."
builtins.TypeError: argument should be integer or bytes-like object, not 'str'
parent 1b20e56b
Loading
Loading
Loading
Loading
+4 −4
Original line number Diff line number Diff line
@@ -152,7 +152,7 @@ class SMTPTests(unittest.TestCase):

        # then check that our local SMTP server received a response
        # and that response contained some bridges
        self.server.getAndCheckMessageContains("Here are your bridges")
        self.server.getAndCheckMessageContains(b"Here are your bridges")

    def test_getBridges_rateLimitExceeded(self):
        if os.environ.get("CI"):
@@ -168,14 +168,14 @@ class SMTPTests(unittest.TestCase):

        # then check that our local SMTP server received a response
        # and that response contained some bridges
        self.server.getAndCheckMessageContains("Here are your bridges")
        self.server.getAndCheckMessageContains(b"Here are your bridges")

        # send another request from the same email address
        sendMail(FROM_ADDRESS)

        # this time, the email response should not contain any bridges
        self.server.getAndCheckMessageContains(
            "You have exceeded the rate limit. Please slow down!")
            b"You have exceeded the rate limit. Please slow down!")

        # then we send another request from the same email address
        sendMail(FROM_ADDRESS)
@@ -203,4 +203,4 @@ class SMTPTests(unittest.TestCase):
                     % random.randint(MIN_FROM_ADDRESS, MAX_FROM_ADDRESS))

        for i in range(NUM_MAILS):
            self.server.getAndCheckMessageContains("Here are your bridges")
            self.server.getAndCheckMessageContains(b"Here are your bridges")