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
|
# (C) Copyright 2005-2025 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
from email.mime.multipart import MIMEMultipart
import io
import os
import shutil
import tempfile
import unittest
from apptools.logger.agent.attachments import Attachments
class AttachmentsTestCase(unittest.TestCase):
def setUp(self):
self.tmpdir = tempfile.mkdtemp()
self.addCleanup(shutil.rmtree, self.tmpdir)
self.tmpfile = os.path.join(self.tmpdir, "dummy_file.txt")
with io.open(self.tmpfile, "w", encoding="utf8") as filehandle:
filehandle.write("Dummy data in dummy file for dummies")
def test_attaching_workspace(self):
class DummyWorkspace(object):
path = self.tmpdir
class MockedApplication(object):
tmpdir = self.tmpdir
def get_service(self, service_id):
return DummyWorkspace()
attachments = Attachments(
application=MockedApplication(), message=MIMEMultipart()
)
attachments.package_workspace()
message = attachments.message
self.assertTrue(message.is_multipart())
payload = message.get_payload()
self.assertEqual(len(payload), 1)
def test_attaching_single_project(self):
class DummySingleProject(object):
location = self.tmpdir
class MockedApplication(object):
tmpdir = self.tmpdir
def get_service(self, service_id):
return DummySingleProject()
attachments = Attachments(
application=MockedApplication(), message=MIMEMultipart()
)
attachments.package_single_project()
message = attachments.message
self.assertTrue(message.is_multipart())
payload = message.get_payload()
self.assertEqual(len(payload), 1)
|
