/
test_backend.py
40 lines (28 loc) · 1.09 KB
/
test_backend.py
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
import os
import pytest
from celery import states
from celery.backends.azureblockblob import AzureBlockBlobBackend
pytest.importorskip('azure')
@pytest.mark.skipif(
not os.environ.get('AZUREBLOCKBLOB_URL'),
reason='Environment variable AZUREBLOCKBLOB_URL required'
)
class test_AzureBlockBlobBackend:
def test_crud(self, manager):
backend = AzureBlockBlobBackend(
app=manager.app,
url=os.environ["AZUREBLOCKBLOB_URL"])
key_values = {("akey%d" % i).encode(): "avalue%d" % i
for i in range(5)}
for key, value in key_values.items():
backend._set_with_state(key, value, states.SUCCESS)
actual_values = backend.mget(key_values.keys())
expected_values = list(key_values.values())
assert expected_values == actual_values
for key in key_values:
backend.delete(key)
def test_get_missing(self, manager):
backend = AzureBlockBlobBackend(
app=manager.app,
url=os.environ["AZUREBLOCKBLOB_URL"])
assert backend.get(b"doesNotExist") is None