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
|
######################################################################
#
# File: test/unit/test_base.py
#
# Copyright 2019 Backblaze Inc. All Rights Reserved.
#
# License https://www.backblaze.com/using_b2_code.html
#
######################################################################
import re
import unittest
from contextlib import contextmanager
from typing import Type
import pytest
@pytest.mark.usefixtures('unit_test_console_tool_class', 'b2_uri_args')
class TestBase(unittest.TestCase):
console_tool_class: Type
@contextmanager
def assertRaises(self, exc, msg=None):
try:
yield
except exc as e:
if msg is not None:
if msg != str(e):
assert False, f"expected message '{msg}', but got '{str(e)}'"
else:
assert False, f'should have thrown {exc}'
@contextmanager
def assertRaisesRegexp(self, expected_exception, expected_regexp):
try:
yield
except expected_exception as e:
if not re.search(expected_regexp, str(e)):
assert False, f"expected message '{expected_regexp}', but got '{str(e)}'"
else:
assert False, f'should have thrown {expected_exception}'
|
