A couple more ProjectRepositoryEntity Tests

Wavesonics · Jul 3, 2024 · e9646aa · e9646aa
1 parent c49a347
commit e9646aa
Show file tree

Hide file tree

Showing 2 changed files with 75 additions and 3 deletions.
diff --git a/server/src/main/kotlin/com/darkrockstudios/apps/hammer/project/ProjectRepository.kt b/server/src/main/kotlin/com/darkrockstudios/apps/hammer/project/ProjectRepository.kt
@@ -161,9 +161,6 @@ class ProjectRepository(
 		if (validateSyncId(userId, projectDef, syncId).not())
 			return SResult.failure("Invalid SyncId", exception = InvalidSyncIdException())
 
-		// TODO how to move this to Datasource?
-		//ensureEntityDir(userId, projectDef)
-
 		val result = when (entity) {
 			is ApiProjectEntity.SceneEntity -> sceneSynchronizer.saveEntity(
 				userId,

diff --git a/.../kotlin/com/darkrockstudios/apps/hammer/project/repository/ProjectRepositoryEntityTest.kt b/.../kotlin/com/darkrockstudios/apps/hammer/project/repository/ProjectRepositoryEntityTest.kt
@@ -9,9 +9,11 @@ import com.darkrockstudios.apps.hammer.utilities.SResult
 import com.darkrockstudios.apps.hammer.utilities.isFailure
 import com.darkrockstudios.apps.hammer.utilities.isSuccess
 import io.mockk.coEvery
+import io.mockk.coVerify
 import io.mockk.every
 import kotlinx.coroutines.test.runTest
 import org.junit.Test
+import kotlin.test.assertEquals
 import kotlin.test.assertTrue
 import kotlin.time.Duration.Companion.minutes
 
@@ -65,6 +67,79 @@ class ProjectRepositoryEntityTest : ProjectRepositoryBaseTest() {
 		}
 	}
 
+	@Test
+	fun `Load Entity - Scene Entity`() = runTest {
+		val syncId = "sync-id"
+		val entityId = 1
+		val sceneEntity = createSceneEntity(1)
+
+		mockCreateSession(syncId)
+
+		coEvery { projectsSessionManager.hasActiveSyncSession(any()) } returns false
+		coEvery { projectSessionManager.hasActiveSyncSession(any()) } returns true
+		coEvery { projectSessionManager.validateSyncId(any(), any(), any()) } returns true
+
+		coEvery {
+			projectDatasource.findEntityType(
+				entityId,
+				userId,
+				projectDefinition
+			)
+		} returns ApiProjectEntity.Type.SCENE
+
+		coEvery {
+			sceneSynchronizer.loadEntity(
+				userId,
+				projectDefinition,
+				entityId,
+			)
+		} returns SResult.success(sceneEntity)
+
+		createProjectRepository().apply {
+			val result = loadEntity(userId, projectDefinition, entityId, syncId)
+			assertTrue(isSuccess(result))
+			assertEquals(sceneEntity, result.data)
+		}
+	}
+
+	@Test
+	fun `Save Entity - Scene Entity`() = runTest {
+		val syncId = "sync-id"
+		val sceneEntity = createSceneEntity(1)
+
+		mockCreateSession(syncId)
+
+		coEvery { projectsSessionManager.hasActiveSyncSession(any()) } returns false
+		coEvery { projectSessionManager.hasActiveSyncSession(any()) } returns true
+		coEvery { projectSessionManager.validateSyncId(any(), any(), any()) } returns true
+
+		coEvery {
+			sceneSynchronizer.saveEntity(
+				userId,
+				projectDefinition,
+				sceneEntity,
+				null,
+				false
+			)
+		} returns SResult.success(true)
+
+		createProjectRepository().apply {
+			val result =
+				saveEntity(userId, projectDefinition, createSceneEntity(1), null, syncId, false)
+			assertTrue { result.isSuccess }
+		}
+
+		coVerify {
+			sceneSynchronizer.saveEntity(
+				userId,
+				projectDefinition,
+				sceneEntity,
+				null,
+				false
+			)
+		}
+	}
+
 	private fun createSceneEntity(entityId: Int): ApiProjectEntity.SceneEntity {
 		return ApiProjectEntity.SceneEntity(
 			id = entityId,