File size: 3,158 Bytes
7c89ed7 | 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 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | # This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at https://mozilla.org/MPL/2.0/.
# Copyright (c) 2017 Mozilla Corporation
from mq.plugins.remove_dots import message
class TestRemoveDotsPlugin():
def setup(self):
self.plugin = message()
def test_first_dot_removal(self):
msg = {
'source': 'cloudtrail',
'details': {
'responseelements': {
'fileuploadurls': {
'.ds_store': 'astringvalue',
'js/app.js': 'astringvalue'
}
}
}
}
(retmessage, retmeta) = self.plugin.onMessage(msg, {})
expected_message = {
'source': 'cloudtrail',
'details': {
'responseelements': {
'fileuploadurls': {
'ds_store': 'astringvalue',
'js/app.js': 'astringvalue'
}
}
}
}
assert retmessage == expected_message
assert retmeta == {}
def test_last_dot_removal(self):
msg = {
'source': 'cloudtrail',
'details': {
'responseelements': {
'fileuploadurls': {
'ds_store.': 'astringvalue',
'js/app.js': 'astringvalue'
}
}
}
}
(retmessage, retmeta) = self.plugin.onMessage(msg, {})
expected_message = {
'source': 'cloudtrail',
'details': {
'responseelements': {
'fileuploadurls': {
'ds_store': 'astringvalue',
'js/app.js': 'astringvalue'
}
}
}
}
assert retmessage == expected_message
assert retmeta == {}
def test_list_dot_removal(self):
msg = {
'source': 'cloudtrail',
'details': {
'responseelements': {
'fileuploadurls': {
'ds_store.': 'astringvalue',
'js/app.js': 'astringvalue'
},
'.randomlist.': [
'weird_case',
'random_case',
],
}
}
}
(retmessage, retmeta) = self.plugin.onMessage(msg, {})
expected_message = {
'source': 'cloudtrail',
'details': {
'responseelements': {
'fileuploadurls': {
'ds_store': 'astringvalue',
'js/app.js': 'astringvalue'
},
'randomlist': [
'weird_case',
'random_case',
],
}
}
}
assert retmessage == expected_message
assert retmeta == {}
|