test: add set and get current image user data test

arshamalh · Dec 1, 2023 · 0ddd6de · 0ddd6de
1 parent a565a67
commit 0ddd6de
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/session/userdata_test.go b/session/userdata_test.go
@@ -48,6 +48,17 @@ func TestUserData(t *testing.T) {
 			assert.NotNil(userImage)
 		}
 	})
+
+	t.Run("set and get current image", func(t *testing.T) {
+		s := session.New()
+		userData := s.Get(2)
+		fakeImages := newFakeImages()
+
+		userData.SetCurrentImage(fakeImages[1])
+		userCurrentImage := userData.GetCurrentImage()
+
+		assert.NotNil(userCurrentImage)
+	})
 }
 
 func newFakeContainers() []*entities.Container {