test: add set and get scene user data test

arshamalh · Dec 1, 2023 · e50c7f6 · e50c7f6
1 parent 0ddd6de
commit e50c7f6
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/session/userdata_test.go b/session/userdata_test.go
@@ -59,6 +59,23 @@ func TestUserData(t *testing.T) {
 
 		assert.NotNil(userCurrentImage)
 	})
+
+	t.Run("set and get scene", func(t *testing.T) {
+		s := session.New()
+		userData := s.Get(2)
+
+		userData.SetScene(entities.SceneRenameImage)
+		scene := userData.GetScene()
+
+		assert.NotNil(scene)
+		assert.Equal(userData.CurrentQuestion, entities.QNewImageName)
+
+		userData.SetScene(entities.SceneRenameContainer)
+		scene = userData.GetScene()
+
+		assert.NotNil(scene)
+		assert.Equal(userData.CurrentQuestion, entities.QNewContainerName)
+	})
 }
 
 func newFakeContainers() []*entities.Container {