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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
# -*- coding: utf-8 -*-
#
# Authors: Guillermo Gonzalez <guillermo.gonzalez@canonical.com>
# Manuel de la Pena <manuel@canonical.com>
# Natalia B. Bidart <natalia.bidart@canonical.com>
#
# Copyright 2010-2011 Canonical Ltd.
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 3, as published
# by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranties of
# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
# PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.
"""Test for the VolumeManager helper."""
import os
import uuid
from tests.syncdaemon.test_vm import BaseVolumeManagerTests
from contrib.testing.testcase import BaseTwistedTestCase
from ubuntuone.platform import expand_user, os_helper
from ubuntuone.syncdaemon import vm_helper
from ubuntuone.syncdaemon.vm_helper import (
create_shares_link,
get_share_dir_name,
get_udf_path,
get_udf_suggested_path,
)
class VMHelperTest(BaseVolumeManagerTests):
"""Test the vm_helper methods."""
def _test_get_udf_path(self, suggested_path):
"""Assert that the resulting udf path is correct."""
assert isinstance(suggested_path, unicode)
assert suggested_path.startswith(u'~')
path = get_udf_path(suggested_path)
expected = suggested_path.replace(u'/', os.path.sep)
expected = expand_user(expected.encode('utf8'))
self.assertEqual(path, expected)
def test_get_udf_path(self):
"""A bytes sequence is returned."""
self._test_get_udf_path(suggested_path=u'~/Documents')
def test_get_udf_path_non_ascii(self):
"""A bytes sequence is returned."""
self._test_get_udf_path(suggested_path=u'~/Documents/Ñoño ñandú')
def test_get_udf_path_funny_chars(self):
"""A bytes sequence is returned."""
self._test_get_udf_path(suggested_path=u'~/Documents/Nr 1: really?')
def test_get_udf_suggested_path(self):
"""Test for get_udf_suggested_path."""
in_home = os.path.join(self.home_dir, 'foo')
self.assertEqual(u'~/foo', get_udf_suggested_path(in_home))
def test_get_udf_suggested_path_long_path(self):
"""Test for get_udf_suggested_path."""
deep_in_home = os.path.join(self.home_dir, 'docs', 'foo', 'bar')
actual = get_udf_suggested_path(deep_in_home)
self.assertEqual(u'~/docs/foo/bar', actual)
def test_get_udf_suggested_path_value_error(self):
"""Test for get_udf_suggested_path."""
outside_home = os.path.join(self.home_dir, os.path.pardir,
'bar', 'foo')
relative_home = os.path.join(os.path.pardir, os.path.pardir, 'foo')
self.assertRaises(ValueError, get_udf_suggested_path, outside_home)
self.assertRaises(ValueError, get_udf_suggested_path, None)
self.assertRaises(ValueError, get_udf_suggested_path, relative_home)
class VMHelperLinkTestCase(BaseTwistedTestCase):
"""Tests for the VM Helper symlinks."""
def test_create_shares_link_exists(self):
"""create_shares_link is noop when there's something with that name."""
base = self.mktemp("test_create_shares_link_exists")
source_path = os.path.join(base, "source")
dest_path = os.path.join(base, "dest")
os_helper.make_dir(dest_path)
self.assertFalse(create_shares_link(source_path, dest_path))
def test_create_shares_link_existing_destiny_with_lnk_extension(self):
"""Add the lnk extension to the end of the file like windows needs."""
base = self.mktemp("test_create_shares_link_exists")
source_path = os.path.join(base, "source")
dest_path = os.path.join(base, "dest.lnk")
os_helper.make_dir(dest_path)
self.assertFalse(create_shares_link(source_path, dest_path))
def test_create_shares_link_makes_the_link(self):
"""create_shares_link makes the link as expected."""
base = self.mktemp("test_create_shares_link_makes_the_link")
source_path = os.path.join(base, "source")
dest_path = os.path.join(base, "dest")
os_helper.make_dir(source_path)
self.assertTrue(create_shares_link(source_path, dest_path))
self.assertTrue(vm_helper.is_link(dest_path))
def test_create_shares_link_existing(self):
"""create_shares_link on an existing path does nothing."""
base = self.mktemp("test_create_shares_link_makes_the_link")
source_path = os.path.join(base, "source")
dest_path = os.path.join(base, "dest")
os_helper.make_dir(source_path)
self.assertTrue(create_shares_link(source_path, dest_path))
self.assertFalse(create_shares_link(source_path, dest_path))
def test_create_shares_link_existing_source_with_lnk_extension(self):
"""Add the lnk extension to the end of the file like windows needs."""
base = self.mktemp("test_create_shares_link_makes_the_link")
source_path = os.path.join(base, "source")
dest_path = os.path.join(base, "dest.lnk")
os_helper.make_dir(source_path)
self.assertTrue(create_shares_link(source_path, dest_path))
self.assertFalse(create_shares_link(source_path, dest_path))
class GetShareDirNameTests(BaseVolumeManagerTests):
share_id = uuid.uuid4()
name = u'The little pretty share (♥)'
def test_get_share_dir_name(self):
"""Test for get_share_dir_name."""
other_name = u'Dorian Grey'
share = self._create_share_volume(volume_id=self.share_id,
name=self.name,
other_visible_name=other_name)
result = get_share_dir_name(share)
expected = u'%s (%s, %s)' % (self.name, other_name, self.share_id)
self.assertEqual(result, expected.encode('utf8'))
def test_get_share_dir_name_visible_name_empty(self):
"""Test for get_share_dir_name."""
other_name = u''
share = self._create_share_volume(volume_id=self.share_id,
name=self.name,
other_visible_name=other_name)
result = get_share_dir_name(share)
expected = u'%s (%s)' % (self.name, self.share_id)
self.assertEqual(result, expected.encode('utf8'))
|