gryf/slack-backup
1
0
Fork 0
mirror of https://github.com/gryf/slack-backup.git synced 2025-12-17 11:30:25 +01:00
Code Issues Projects Releases Wiki Activity

Adjusting test data/tests for new file attachment way.

Browse Source
This commit is contained in:
gryf
2019-01-05 17:30:22 +01:00
parent 4a3bb067f4
commit fe1e2dd230
1 changed files with 147 additions and 163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
tests/test_client.py
View File

@@ -258,21 +258,22 @@ MSGS = {'messages': [{"type": "message",
{"display_as_bot": False, {"display_as_bot": False,
"subtype": "file_share", "subtype": "file_share",
"username": "<@UCCCCCCCC|name2>", "username": "<@UCCCCCCCC|name2>",
"file": {"thumb_960": "https://files.slack.com/files-tmb" "files": [{"thumb_960": "https://files.slack.com/files-"
"/hash/screenshot_960.png", "tmb/hash/screenshot_960.png",
"user": "UCCCCCCCC", "user": "UCCCCCCCC",
"size": 77222, "size": 77222,
"thumb_1024_h": 754, "thumb_1024_h": 754,
"timestamp": 1479407569, "timestamp": 1479407569,
"url_private_download": "https://files.slack.co" "url_private_download": "https://files.slack."
"m/files-pri/hsh/downlo" "com/files-pri/hsh/do"
"ad/screenshot.png", "wnload/screenshot.pn"
"thumb_360": "https://files.slack.com/files-tmb" "g",
"/hash/screenshot_360.png", "thumb_360": "https://files.slack.com/files-"
"tmb/hash/screenshot_360.png",
"username": "", "username": "",
"external_type": "", "external_type": "",
"thumb_64": "https://files.slack.com/files-tmb/" "thumb_64": "https://files.slack.com/files-"
"hash/screenshot_64.png", "tmb/hash/screenshot_64.png",
"created": 1479407569, "created": 1479407569,
"ims": [], "ims": [],
"groups": [], "groups": [],
@@ -290,34 +291,34 @@ MSGS = {'messages': [{"type": "message",
"pretty_type": "PNG", "pretty_type": "PNG",
"editable": False, "editable": False,
"thumb_960_w": 960, "thumb_960_w": 960,
"thumb_80": "https://files.slack.com/files-tmb/" "thumb_80": "https://files.slack.com/files-"
"hash/screenshot_80.png", "tmb/hash/screenshot_80.png",
"comments_count": 0, "comments_count": 0,
"image_exif_rotation": 1, "image_exif_rotation": 1,
"thumb_160": "https://files.slack.com/files-tmb" "thumb_160": "https://files.slack.com/files-"
"/hash/screenshot_160.png", "tmb/hash/screenshot_160.png",
"thumb_480_w": 480, "thumb_480_w": 480,
"is_external": False, "is_external": False,
"display_as_bot": False, "display_as_bot": False,
"thumb_720_h": 530, "thumb_720_h": 530,
"channels": ["C00000001"], "channels": ["C00000001"],
"title": "Screenshot.png", "title": "Screenshot.png",
"thumb_480": "https://files.slack.com/files-tmb" "thumb_480": "https://files.slack.com/files-"
"/hash/screenshot_480.png", "tmb/hash/screenshot_480.png",
"url_private": "https://files.slack.com/files-" "url_private": "https://files.slack.com/files"
"pri/hsh/screenshot.png", "-pri/hsh/screenshot.png",
"mode": "hosted", "mode": "hosted",
"thumb_1024_w": 1024, "thumb_1024_w": 1024,
"permalink": "https://esm64.slack.com/files/" "permalink": "https://esm64.slack.com/files/"
"name2/F3405RRB5/screenshot.png", "name2/F3405RRB5/screenshot.png",
"thumb_480_h": 353, "thumb_480_h": 353,
"public_url_shared": False, "public_url_shared": False,
"thumb_720": "https://files.slack.com/files-tmb" "thumb_720": "https://files.slack.com/files-"
"/hash/screenshot_720.png", "tmb/hash/screenshot_720.png",
"thumb_360_w": 360, "thumb_360_w": 360,
"permalink_public": "https://slack-files.com/" "permalink_public": "https://slack-files.com/"
"hsh-7dbb96b758", "hsh-7dbb96b758",
"thumb_720_w": 720}, "thumb_720_w": 720}],
"type": "message", "type": "message",
"user": "UCCCCCCCC", "user": "UCCCCCCCC",
"bot_id": None, "bot_id": None,
@@ -360,7 +361,7 @@ SHARED = {"type": "message",
"text": "<@UAAAAAAAA> shared a file: <https://bla." "text": "<@UAAAAAAAA> shared a file: <https://bla."
"slack.com/files/name%20lastname/F7ARMB4JU/17|" "slack.com/files/name%20lastname/F7ARMB4JU/17|"
"some_spreadsheet>", "some_spreadsheet>",
"file": { "files": [{
"id": "F6ABMB0Ja", "id": "F6ABMB0Ja",
"created": 1479147929, "created": 1479147929,
"timestamp": 1479147929, "timestamp": 1479147929,
@@ -394,7 +395,7 @@ SHARED = {"type": "message",
"ims": [], "ims": [],
"comments_count": 0, "comments_count": 0,
"has_rich_preview": True "has_rich_preview": True
}, }],
"user": 'UAAAAAAAA', "user": 'UAAAAAAAA',
"upload": False, "upload": False,
"display_as_bot": False, "display_as_bot": False,
@@ -423,7 +424,7 @@ PINNED = {'attachments': [{'fallback': 'blah',
EXTERNAL_DATA = {"bot_id": None, EXTERNAL_DATA = {"bot_id": None,
"display_as_bot": False, "display_as_bot": False,
"file": { "files": [{
"channels": [ "channels": [
"xxx" "xxx"
], ],
@@ -487,14 +488,14 @@ EXTERNAL_DATA = {"bot_id": None,
"groups": [], "groups": [],
"ims": [], "ims": [],
"comments_count": 0, "comments_count": 0,
"has_rich_preview": True}, "has_rich_preview": True}],
"user": "xxx", "user": "xxx",
"upload": False, "upload": False,
"username": "xxx"} "username": "xxx"}
INTERNAL_DATA = {"bot_id": None, INTERNAL_DATA = {"bot_id": None,
"display_as_bot": False, "display_as_bot": False,
"file": {"channels": ["zzz"], "files": [{"channels": ["zzz"],
"comments_count": 1, "comments_count": 1,
"created": 1524724681, "created": 1524724681,
"display_as_bot": False, "display_as_bot": False,
@@ -566,7 +567,7 @@ INTERNAL_DATA = {"bot_id": None,
"files-pri/TXXXXXX-id/" "files-pri/TXXXXXX-id/"
"download/img.jpg", "download/img.jpg",
"user": "UAAAAAAAA", "user": "UAAAAAAAA",
"username": ""}, "username": ""}],
"subtype": "file_share", "subtype": "file_share",
"text": "<@UAAAAAAAA> uploaded a file: <https://fake.slack." "text": "<@UAAAAAAAA> uploaded a file: <https://fake.slack."
"com/files/UAAAAAAAA/id/img.jpg|img.jpg> and " "com/files/UAAAAAAAA/id/img.jpg|img.jpg> and "
@@ -749,22 +750,7 @@ class TestCreateMessage(unittest.TestCase):
self.assertTrue('shared a file' in msg.text) self.assertTrue('shared a file' in msg.text)
self.assertFalse(msg.is_starred) self.assertFalse(msg.is_starred)
self.assertEqual(msg.type, 'file_share') self.assertEqual(msg.type, 'file_share')
fd.assert_called_once_with(msg, SHARED['file']) fd.assert_called_once_with(msg, SHARED['files'][0])
@mock.patch('slack_backup.client.Client._get_user')
def test_external_file_upload_as_attachment(self, gu):
cl = client.Client(FakeArgs())
cl.downloader._download = mock.MagicMock(return_value='aa')
cl.session = mock.MagicMock()
cl._url_file_to_attachment = True
channel = o.Channel({'name': 'test', 'id': 'C00000001'})
cl._create_message(SHARED, channel)
msg = cl.session.add.call_args[0][0]
self.assertEqual(len(msg.attachments), 1)
self.assertTrue('shared a file' in msg.text)
self.assertFalse(msg.is_starred)
@mock.patch('slack_backup.client.Client._file_data') @mock.patch('slack_backup.client.Client._file_data')
@mock.patch('slack_backup.client.Client._get_user') @mock.patch('slack_backup.client.Client._get_user')
@@ -793,17 +779,17 @@ class TestFileShare(unittest.TestCase):
def test_file_data(self, md, dl): def test_file_data(self, md, dl):
dl.side_effect = ['some_path'] dl.side_effect = ['some_path']
url = INTERNAL_DATA['file']['url_private_download'] url = INTERNAL_DATA['files'][0]['url_private_download']
cl = client.Client(FakeArgs()) cl = client.Client(FakeArgs())
cl.downloader._download = mock.MagicMock(return_value=url) cl.downloader._download = mock.MagicMock(return_value=url)
cl.session = mock.MagicMock() cl.session = mock.MagicMock()
msg = o.Message() msg = o.Message()
cl._file_data(msg, INTERNAL_DATA['file']) cl._file_data(msg, INTERNAL_DATA['files'][0])
self.assertIsNotNone(msg.file) self.assertIsNotNone(msg.files)
self.assertEqual(msg.file.filepath, 'some_path') self.assertEqual(msg.files[0].filepath, 'some_path')
@mock.patch('slack_backup.download.Download.download') @mock.patch('slack_backup.download.Download.download')
@mock.patch('slack_backup.utils.makedirs') @mock.patch('slack_backup.utils.makedirs')
@@ -811,15 +797,15 @@ class TestFileShare(unittest.TestCase):
dl.side_effect = ['some_path'] dl.side_effect = ['some_path']
data = copy.deepcopy(INTERNAL_DATA) data = copy.deepcopy(INTERNAL_DATA)
data['file']['is_starred'] = True data['files'][0]['is_starred'] = True
url = data['file']['url_private_download'] url = data['files'][0]['url_private_download']
cl = client.Client(FakeArgs()) cl = client.Client(FakeArgs())
cl.downloader._download = mock.MagicMock(return_value=url) cl.downloader._download = mock.MagicMock(return_value=url)
cl.session = mock.MagicMock() cl.session = mock.MagicMock()
msg = o.Message() msg = o.Message()
cl._file_data(msg, data['file']) cl._file_data(msg, data['files'][0])
self.assertTrue(msg.is_starred) self.assertTrue(msg.is_starred)
@@ -830,8 +816,6 @@ class TestFileShare(unittest.TestCase):
uuid.side_effect = ['aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee'] uuid.side_effect = ['aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee']
dl.side_effect = ['some_path'] dl.side_effect = ['some_path']
# url = EXTERNAL_DATA['file']['url_private']
cl = client.Client(FakeArgs()) cl = client.Client(FakeArgs())
cl.session = mock.MagicMock() cl.session = mock.MagicMock()
@@ -843,7 +827,7 @@ class TestFileShare(unittest.TestCase):
'usp=drivesdk --> ' 'usp=drivesdk --> '
'assets/files/aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee\n') 'assets/files/aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee\n')
cl._file_data(msg, EXTERNAL_DATA['file']) cl._file_data(msg, EXTERNAL_DATA['files'][0])
file_ = cl.session.add.call_args[0][0] file_ = cl.session.add.call_args[0][0]
self.assertEqual(cl._dldata, [expexted_line]) self.assertEqual(cl._dldata, [expexted_line])
self.assertEqual(file_.filepath, self.assertEqual(file_.filepath,