simplify test and avoid sst instead of snapstrename/snapstate-snapst

1 files changed, 3 insertions, 8 deletions

diff --git a/overlord/snapstate/snapmgr_test.go b/overlord/snapstate/snapmgr_test.go
index d4a25ca6f8..701f7782bd 100644
--- a/overlord/snapstate/snapmgr_test.go
+++ b/overlord/snapstate/snapmgr_test.go
@@ -3343,15 +3343,10 @@ func (s *snapmgrQuerySuite) TestAll(c *C) {
 
	snapStates, err := snapstate.All(st)
	c.Assert(err, IsNil)
+	c.Assert(snapStates, HasLen, 1)
 
-	c.Check(snapStates, HasLen, 1)
-
-	var snapst *snapstate.SnapState
-
-	for name, sst := range snapStates {
-	c.Assert(name, Equals, "name1")
-	snapst = sst
-	}
+	snapst := snapStates["name1"]
+	c.Assert(snapst, NotNil)
 
	c.Check(snapst.Active, Equals, true)
	c.Check(snapst.CurrentSideInfo(), NotNil)