github.com/Lephar/snapd@v0.0.0-20210825215435-c7fba9cef4d2/cmd/snap-update-ns/sorting_test.go (about)

     1  // -*- Mode: Go; indent-tabs-mode: t -*-
     2  
     3  /*
     4   * Copyright (C) 2017 Canonical Ltd
     5   *
     6   * This program is free software: you can redistribute it and/or modify
     7   * it under the terms of the GNU General Public License version 3 as
     8   * published by the Free Software Foundation.
     9   *
    10   * This program is distributed in the hope that it will be useful,
    11   * but WITHOUT ANY WARRANTY; without even the implied warranty of
    12   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13   * GNU General Public License for more details.
    14   *
    15   * You should have received a copy of the GNU General Public License
    16   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    17   *
    18   */
    19  
    20  package main
    21  
    22  import (
    23  	"sort"
    24  
    25  	. "gopkg.in/check.v1"
    26  
    27  	"github.com/snapcore/snapd/osutil"
    28  )
    29  
    30  type sortSuite struct{}
    31  
    32  var _ = Suite(&sortSuite{})
    33  
    34  func (s *sortSuite) TestTrailingSlashesComparison(c *C) {
    35  	// Naively sorted entries.
    36  	entries := []osutil.MountEntry{
    37  		{Dir: "/a/b"},
    38  		{Dir: "/a/b-1"},
    39  		{Dir: "/a/b-1/3"},
    40  		{Dir: "/a/b/c"},
    41  	}
    42  	sort.Sort(byOriginAndMagicDir(entries))
    43  	// Entries sorted as if they had a trailing slash.
    44  	c.Assert(entries, DeepEquals, []osutil.MountEntry{
    45  		{Dir: "/a/b-1"},
    46  		{Dir: "/a/b-1/3"},
    47  		{Dir: "/a/b"},
    48  		{Dir: "/a/b/c"},
    49  	})
    50  }
    51  
    52  func (s *sortSuite) TestParallelInstancesAndSimple(c *C) {
    53  	// Naively sorted entries.
    54  	entries := []osutil.MountEntry{
    55  		{Dir: "/a/b-1"},
    56  		{Dir: "/a/b", Options: []string{osutil.XSnapdOriginLayout()}},
    57  		{Dir: "/snap/bar/baz", Options: []string{osutil.XSnapdOriginLayout()}},
    58  		{Dir: "/snap/bar", Options: []string{osutil.XSnapdOriginOvername()}},
    59  		{Dir: "/a/b-1/3"},
    60  		{Dir: "/var/snap/bar", Options: []string{osutil.XSnapdOriginOvername()}},
    61  		{Dir: "/a/b/c"},
    62  	}
    63  	sort.Sort(byOriginAndMagicDir(entries))
    64  	// Entries sorted as if they had a trailing slash.
    65  	c.Assert(entries, DeepEquals, []osutil.MountEntry{
    66  		{Dir: "/snap/bar", Options: []string{osutil.XSnapdOriginOvername()}},
    67  		{Dir: "/var/snap/bar", Options: []string{osutil.XSnapdOriginOvername()}},
    68  		{Dir: "/a/b-1"},
    69  		{Dir: "/a/b-1/3"},
    70  		{Dir: "/a/b", Options: []string{osutil.XSnapdOriginLayout()}},
    71  		{Dir: "/a/b/c"},
    72  		{Dir: "/snap/bar/baz", Options: []string{osutil.XSnapdOriginLayout()}},
    73  	})
    74  }
    75  
    76  func (s *sortSuite) TestOvernameOrder(c *C) {
    77  	expected := []osutil.MountEntry{
    78  		{Dir: "/a/b/2", Options: []string{osutil.XSnapdOriginOvername()}},
    79  		{Dir: "/a/b"},
    80  	}
    81  	entries := []osutil.MountEntry{
    82  		{Dir: "/a/b"},
    83  		{Dir: "/a/b/2", Options: []string{osutil.XSnapdOriginOvername()}},
    84  	}
    85  	entriesRev := []osutil.MountEntry{
    86  		{Dir: "/a/b/2", Options: []string{osutil.XSnapdOriginOvername()}},
    87  		{Dir: "/a/b"},
    88  	}
    89  	sort.Sort(byOriginAndMagicDir(entries))
    90  	c.Assert(entries, DeepEquals, expected)
    91  	sort.Sort(byOriginAndMagicDir(entriesRev))
    92  	c.Assert(entriesRev, DeepEquals, expected)
    93  }
    94  
    95  func (s *sortSuite) TestParallelInstancesAlmostSorted(c *C) {
    96  	// use a mount profile that was seen to be broken in the wild
    97  	entries := []osutil.MountEntry{
    98  		{Dir: "/snap/foo", Options: []string{osutil.XSnapdOriginOvername()}},
    99  		{Dir: "/var/snap/foo", Options: []string{osutil.XSnapdOriginOvername()}},
   100  		{Dir: "/snap/foo/44/foo/certs", Options: []string{osutil.XSnapdOriginLayout()}},
   101  		{Dir: "/snap/foo/44/foo/config", Options: []string{osutil.XSnapdOriginLayout()}},
   102  		{Dir: "/snap/foo/44/usr/bin/python", Options: []string{osutil.XSnapdOriginLayout()}},
   103  		{Dir: "/snap/foo/44/usr/bin/python3", Options: []string{osutil.XSnapdOriginLayout()}},
   104  		{Dir: "/usr/bin/java", Options: []string{osutil.XSnapdOriginLayout()}},
   105  		{Dir: "/usr/bin/java8", Options: []string{osutil.XSnapdOriginLayout()}},
   106  		{Dir: "/usr/bin/node", Options: []string{osutil.XSnapdOriginLayout()}},
   107  		{Dir: "/usr/bin/nodejs12x", Options: []string{osutil.XSnapdOriginLayout()}},
   108  		{Dir: "/usr/bin/python2.7", Options: []string{osutil.XSnapdOriginLayout()}},
   109  		{Dir: "/usr/bin/python3.7", Options: []string{osutil.XSnapdOriginLayout()}},
   110  		{Dir: "/usr/bin/python3.8", Options: []string{osutil.XSnapdOriginLayout()}},
   111  	}
   112  	sort.Sort(byOriginAndMagicDir(entries))
   113  	// overname entries are always first
   114  	c.Assert(entries, DeepEquals, []osutil.MountEntry{
   115  		{Dir: "/snap/foo", Options: []string{osutil.XSnapdOriginOvername()}},
   116  		{Dir: "/var/snap/foo", Options: []string{osutil.XSnapdOriginOvername()}},
   117  		{Dir: "/snap/foo/44/foo/certs", Options: []string{osutil.XSnapdOriginLayout()}},
   118  		{Dir: "/snap/foo/44/foo/config", Options: []string{osutil.XSnapdOriginLayout()}},
   119  		{Dir: "/snap/foo/44/usr/bin/python", Options: []string{osutil.XSnapdOriginLayout()}},
   120  		{Dir: "/snap/foo/44/usr/bin/python3", Options: []string{osutil.XSnapdOriginLayout()}},
   121  		{Dir: "/usr/bin/java", Options: []string{osutil.XSnapdOriginLayout()}},
   122  		{Dir: "/usr/bin/java8", Options: []string{osutil.XSnapdOriginLayout()}},
   123  		{Dir: "/usr/bin/node", Options: []string{osutil.XSnapdOriginLayout()}},
   124  		{Dir: "/usr/bin/nodejs12x", Options: []string{osutil.XSnapdOriginLayout()}},
   125  		{Dir: "/usr/bin/python2.7", Options: []string{osutil.XSnapdOriginLayout()}},
   126  		{Dir: "/usr/bin/python3.7", Options: []string{osutil.XSnapdOriginLayout()}},
   127  		{Dir: "/usr/bin/python3.8", Options: []string{osutil.XSnapdOriginLayout()}},
   128  	})
   129  }