Skip to content

Commit 5162ddf

Browse files
committed
Add test for dup gist child and remove trash folder exclude iinside the dup gist dirs
1 parent 9133f97 commit 5162ddf

File tree

2 files changed

+14
-7
lines changed

2 files changed

+14
-7
lines changed

desktop/core/src/desktop/api2_tests.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -705,8 +705,15 @@ def test_multiple_gist_dirs_on_gist_create(self):
705705

706706
gist_dir1 = Directory.objects.create(name=Document2.GIST_DIR, owner=self.user, parent_directory=home_dir)
707707
gist_dir2 = Directory.objects.create(name=Document2.GIST_DIR, owner=self.user, parent_directory=home_dir)
708+
gist_child = Document2.objects.create(
709+
name='test_gist_child',
710+
data=json.dumps({'statement': 'SELECT 123'}),
711+
owner=self.user,
712+
type='gist',
713+
parent_directory=gist_dir2,
714+
)
708715

709-
assert_equal(2, Directory.objects.filter(name=Document2.GIST_DIR, type='directory').count())
716+
assert_equal(2, Directory.objects.filter(name=Document2.GIST_DIR, type='directory', owner=self.user).count())
710717

711718
# get_gist_directory merges all duplicate gist directories into one
712719
response = self._create_gist(
@@ -715,11 +722,12 @@ def test_multiple_gist_dirs_on_gist_create(self):
715722
name='test_gist_create',
716723
)
717724
gist_uuid = json.loads(response.content)['uuid']
718-
gist_parent_uuid = Document2.objects.get(uuid=gist_uuid).parent_directory.uuid
725+
gist_home = Document2.objects.get(uuid=gist_uuid).parent_directory
719726

720-
assert_equal(1, Directory.objects.filter(name=Document2.GIST_DIR, type='directory').count())
721-
assert_true(Directory.objects.filter(name=Document2.GIST_DIR, type='directory', uuid=gist_parent_uuid).exists())
722-
assert_equal(gist_dir1.uuid, Directory.objects.get(name=Document2.GIST_DIR, type='directory', parent_directory=home_dir).uuid)
727+
assert_equal(1, Directory.objects.filter(name=Document2.GIST_DIR, type='directory', owner=self.user).count())
728+
assert_true(Directory.objects.filter(name=Document2.GIST_DIR, type='directory', uuid=gist_home.uuid).exists())
729+
assert_equal(gist_dir1.uuid, gist_home.uuid)
730+
assert_equal(Document2.objects.get(name='test_gist_child', type='gist', owner=self.user).parent_directory, gist_home)
723731

724732

725733
def test_get(self):

desktop/core/src/desktop/models.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1079,9 +1079,8 @@ def get_gist_directory(self, user):
10791079
all_gist_dirs = self.filter(owner=user, parent_directory=home_dir, name=Document2.GIST_DIR, type='directory').order_by('-last_modified')
10801080
gist_dir = all_gist_dirs.last()
10811081
gist_dirs_dup = all_gist_dirs.exclude(uuid=gist_dir.uuid)
1082-
10831082
for dir in gist_dirs_dup:
1084-
dir.children.exclude(name='.Trash').update(parent_directory=gist_dir)
1083+
dir.children.update(parent_directory=gist_dir)
10851084

10861085
gist_dirs_dup.delete()
10871086

0 commit comments

Comments
 (0)