File: test_bucket.py

package info (click to toggle)
python-b2sdk 2.8.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 3,020 kB
  • sloc: python: 30,902; sh: 13; makefile: 8
file content (59 lines) | stat: -rw-r--r-- 2,054 bytes parent folder | download
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
######################################################################
#
# File: test/integration/test_bucket.py
#
# Copyright 2024 Backblaze Inc. All Rights Reserved.
#
# License https://www.backblaze.com/using_b2_code.html
#
######################################################################
import pytest

from test.helpers import assert_dict_equal_ignore_extra


def test_bucket_notification_rules(bucket, b2_api):
    if 'writeBucketNotifications' not in b2_api.account_info.get_allowed()['capabilities']:
        pytest.skip('Test account does not have writeBucketNotifications capability')

    assert bucket.set_notification_rules([]) == []
    assert bucket.get_notification_rules() == []

    notification_rule = {
        'eventTypes': ['b2:ObjectCreated:*'],
        'isEnabled': True,
        'name': 'test-rule',
        'objectNamePrefix': '',
        'targetConfiguration': {
            'customHeaders': [],
            'targetType': 'webhook',
            'url': 'https://example.com/webhook',
            'hmacSha256SigningSecret': 'stringOf32AlphaNumericCharacters',
        },
    }

    set_notification_rules = bucket.set_notification_rules([notification_rule])
    assert set_notification_rules == bucket.get_notification_rules()
    assert_dict_equal_ignore_extra(
        set_notification_rules,
        [{**notification_rule, 'isSuspended': False, 'suspensionReason': ''}],
    )
    assert bucket.set_notification_rules([]) == []


def test_bucket_update__lifecycle_rules(bucket, b2_api):
    lifecycle_rule = {
        'daysFromHidingToDeleting': 1,
        'daysFromUploadingToHiding': 1,
        'daysFromStartingToCancelingUnfinishedLargeFiles': 1,
        'fileNamePrefix': '',
    }

    old_rules_list = bucket.lifecycle_rules

    updated_bucket = bucket.update(lifecycle_rules=[lifecycle_rule])
    assert updated_bucket.lifecycle_rules == [lifecycle_rule]
    assert bucket.lifecycle_rules is old_rules_list

    updated_bucket = bucket.update(lifecycle_rules=[])
    assert updated_bucket.lifecycle_rules == []