github.com/david-imola/snapd@v0.0.0-20210611180407-2de8ddeece6d/overlord/patch/patch6_3_test.go (about) 1 // -*- Mode: Go; indent-tabs-mode: t -*- 2 3 /* 4 * Copyright (C) 2019 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 patch_test 21 22 import ( 23 "bytes" 24 25 . "gopkg.in/check.v1" 26 27 "github.com/snapcore/snapd/dirs" 28 "github.com/snapcore/snapd/overlord/patch" 29 "github.com/snapcore/snapd/overlord/snapstate" 30 "github.com/snapcore/snapd/overlord/state" 31 "github.com/snapcore/snapd/snap" 32 ) 33 34 type patch63Suite struct{} 35 36 var _ = Suite(&patch63Suite{}) 37 38 var statePatch6_3JSON = []byte(` 39 { 40 "data": { 41 "patch-level": 6, 42 "snaps": { 43 "local-install": { 44 "type": "app", 45 "sequence": [ 46 { 47 "name": "local-install", 48 "snap-id": "", 49 "revision": "x1" 50 }, 51 { 52 "name": "local-install", 53 "snap-id": "", 54 "revision": "x2" 55 } 56 ], 57 "active": true, 58 "current": "x2" 59 }, 60 "prefix-postfix-slashes": { 61 "type": "app", 62 "sequence": [ 63 { 64 "name": "prefix-postfix-slashes", 65 "snap-id": "Hswp9oOzj3b4mw8gcC00XtxWnKH9QiCQ", 66 "revision": "30", 67 "channel": "edge", 68 "title": "some-title" 69 }, 70 { 71 "name": "prefix-postfix-slashes", 72 "snap-id": "Hswp9oOzj3b4mw8gcC00XtxWnKH9QiCQ", 73 "revision": "32", 74 "channel": "edge", 75 "title": "some-title" 76 } 77 ], 78 "active": true, 79 "current": "32", 80 "channel": "//edge//", 81 "user-id": 1 82 }, 83 "one-prefix-slash": { 84 "type": "app", 85 "sequence": [ 86 { 87 "name": "white", 88 "snap-id": "one-prefix-slash-id", 89 "revision": "2" 90 } 91 ], 92 "active": true, 93 "current": "2", 94 "channel": "/stable" 95 }, 96 "track-with-risk": { 97 "type": "app", 98 "sequence": [ 99 { 100 "name": "track-with-risk", 101 "snap-id": "track-with-snapid-id", 102 "revision": "3" 103 } 104 ], 105 "active": true, 106 "current": "3", 107 "channel": "latest/stable" 108 }, 109 "track-with-risk-branch": { 110 "type": "app", 111 "sequence": [ 112 { 113 "name": "red", 114 "snap-id": "track-with-risk-branch-snapid-id", 115 "revision": "3" 116 } 117 ], 118 "active": true, 119 "current": "3", 120 "channel": "1.0/stable/branch" 121 } 122 } 123 }, 124 "changes": { 125 "6": { 126 "id": "6", 127 "kind": "auto-refresh", 128 "summary": "...", 129 "status": 0, 130 "clean": true, 131 "task-ids": ["1", "8"] 132 }, 133 "9": { 134 "id": "9", 135 "kind": "install", 136 "summary": "...", 137 "status": 4, 138 "clean": true, 139 "task-ids": ["10"] 140 }, 141 "99": { 142 "id": "99", 143 "kind": "install", 144 "summary": "...", 145 "status": 0, 146 "clean": true, 147 "task-ids": ["99"] 148 } 149 }, 150 "tasks": { 151 "1": { 152 "id": "1", 153 "kind": "download-snap", 154 "summary": "...", 155 "status": 2, 156 "clean": true, 157 "data": { 158 "snap-setup": { 159 "channel": "/stable", 160 "type": "app", 161 "is-auto-refresh": true, 162 "snap-path": "/path", 163 "download-info": { 164 "download-url": "foo", 165 "size": 1234, 166 "sha3-384": "123456", 167 "deltas": [ 168 { 169 "from-revision": 10934, 170 "to-revision": 10972, 171 "format": "xdelta3", 172 "download-url": "foo", 173 "size": 16431136, 174 "sha3-384": "1" 175 } 176 ] 177 }, 178 "side-info": { 179 "name": "other-snap", 180 "snap-id": "other-snap-id", 181 "revision": "1", 182 "channel": "stable", 183 "title": "other-snap" 184 }, 185 "media": [ 186 { 187 "type": "icon", 188 "url": "a" 189 }, 190 { 191 "type": "screenshot", 192 "url": "2" 193 }, 194 { 195 "type": "screenshot", 196 "url": "3" 197 }, 198 { 199 "type": "screenshot", 200 "url": "4" 201 }, 202 { 203 "type": "video", 204 "url": "5" 205 } 206 ] 207 } 208 }, 209 "change": "6" 210 }, 211 "8": { 212 "id": "8", 213 "kind": "link-snap", 214 "summary": "", 215 "status": 2, 216 "data": { 217 "old-candidate-index": -1, 218 "old-channel": "/18/edge//", 219 "snap-setup-task": "1" 220 }, 221 "change": "6" 222 }, 223 "10": { 224 "id": "10", 225 "kind": "other", 226 "summary": "", 227 "status": 4, 228 "data": { 229 "old-channel": "/edge", 230 "snap-setup": { 231 "channel": "/stable", 232 "type": "app", 233 "snap-path": "/path", 234 "side-info": { 235 "name": "some-snap", 236 "snap-id": "some-snap-id", 237 "revision": "1", 238 "channel": "stable", 239 "title": "snapd" 240 } 241 } 242 }, 243 "change": "9" 244 }, 245 "99": { 246 "id": "99", 247 "kind": "prepare-snap", 248 "summary": "", 249 "status": 0, 250 "data": { 251 "snap-setup": { 252 "type": "app", 253 "snap-path": "/path", 254 "side-info": { 255 "name": "local-snap", 256 "snap-id": "", 257 "revision": "unset" 258 } 259 } 260 }, 261 "change": "99" 262 } 263 } 264 }`) 265 266 func (s *patch63Suite) SetUpTest(c *C) { 267 dirs.SetRootDir(c.MkDir()) 268 snap.MockSanitizePlugsSlots(func(*snap.Info) {}) 269 } 270 271 func (s *patch63Suite) TestPatch63(c *C) { 272 restore1 := patch.MockLevel(6, 3) 273 defer restore1() 274 275 r := bytes.NewReader(statePatch6_3JSON) 276 st, err := state.ReadState(nil, r) 277 c.Assert(err, IsNil) 278 279 c.Assert(patch.Apply(st), IsNil) 280 st.Lock() 281 defer st.Unlock() 282 283 all, err := snapstate.All(st) 284 c.Assert(err, IsNil) 285 // our mocks are ok 286 c.Check(all, HasLen, 5) 287 c.Check(all["prefix-postfix-slashes"], NotNil) 288 // our patch changed this 289 c.Check(all["prefix-postfix-slashes"].TrackingChannel, Equals, "latest/edge") 290 // none of the other information has changed 291 c.Check(all["prefix-postfix-slashes"], DeepEquals, &snapstate.SnapState{ 292 SnapType: "app", 293 Sequence: []*snap.SideInfo{ 294 { 295 RealName: "prefix-postfix-slashes", 296 SnapID: "Hswp9oOzj3b4mw8gcC00XtxWnKH9QiCQ", 297 Revision: snap.R(30), 298 Channel: "edge", 299 EditedTitle: "some-title", 300 }, { 301 RealName: "prefix-postfix-slashes", 302 SnapID: "Hswp9oOzj3b4mw8gcC00XtxWnKH9QiCQ", 303 Revision: snap.R(32), 304 Channel: "edge", 305 EditedTitle: "some-title", 306 }, 307 }, 308 Active: true, 309 Current: snap.R(32), 310 TrackingChannel: "latest/edge", 311 UserID: 1, 312 }) 313 // another transition 314 c.Check(all["one-prefix-slash"].TrackingChannel, Equals, "latest/stable") 315 // full 316 c.Check(all["track-with-risk"].TrackingChannel, Equals, "latest/stable") 317 // unchanged 318 c.Check(all["track-with-risk-branch"].TrackingChannel, Equals, "1.0/stable/branch") 319 // also unchanged 320 c.Check(all["local-install"].TrackingChannel, Equals, "") 321 322 // check tasks 323 task := st.Task("1") 324 c.Assert(task, NotNil) 325 // this was converted 326 var snapsup snapstate.SnapSetup 327 err = task.Get("snap-setup", &snapsup) 328 c.Assert(err, IsNil) 329 c.Check(snapsup.Channel, Equals, "latest/stable") 330 331 // sanity check that old stuff is untouched 332 c.Check(snapsup.Flags.IsAutoRefresh, Equals, true) 333 c.Assert(snapsup.Media, HasLen, 5) 334 c.Check(snapsup.Media[0].URL, Equals, "a") 335 c.Assert(snapsup.DownloadInfo, NotNil) 336 c.Check(snapsup.DownloadInfo.DownloadURL, Equals, "foo") 337 c.Check(snapsup.DownloadInfo.Deltas, HasLen, 1) 338 339 // old-channel data got updated 340 task = st.Task("8") 341 c.Assert(task, NotNil) 342 var oldCh string 343 c.Assert(task.Get("old-channel", &oldCh), IsNil) 344 c.Check(oldCh, Equals, "18/edge") 345 346 // task 10 not updated because the change is ready 347 task = st.Task("10") 348 c.Assert(task, NotNil) 349 c.Assert(task.Get("snap-setup", &snapsup), IsNil) 350 c.Check(snapsup.Channel, Equals, "/stable") 351 err = task.Get("old-channel", &oldCh) 352 c.Assert(err, IsNil) 353 c.Check(oldCh, Equals, "/edge") 354 }