feat(backend): persist tasks and generated message history
- add EntityTask domain and CRUD API backed by PostgreSQL - relate generated messages directly to tasks and delete on task removal - move preview generation to backend Llama endpoint - migrate frontend task APIs from localStorage to backend endpoints - update tests and CLAUDE rules for backend-owned LLM/persistence
This commit is contained in:
@@ -0,0 +1,141 @@
|
||||
package com.condado.newsletter.controller
|
||||
|
||||
import com.condado.newsletter.model.GeneratedMessageHistory
|
||||
import com.condado.newsletter.model.EntityTask
|
||||
import com.condado.newsletter.model.VirtualEntity
|
||||
import com.condado.newsletter.repository.EntityTaskRepository
|
||||
import com.condado.newsletter.repository.GeneratedMessageHistoryRepository
|
||||
import com.condado.newsletter.repository.VirtualEntityRepository
|
||||
import com.condado.newsletter.scheduler.EntityScheduler
|
||||
import com.condado.newsletter.service.JwtService
|
||||
import com.condado.newsletter.service.LlamaPreviewService
|
||||
import com.ninjasquad.springmockk.MockkBean
|
||||
import io.mockk.every
|
||||
import jakarta.servlet.http.Cookie
|
||||
import org.junit.jupiter.api.AfterEach
|
||||
import org.junit.jupiter.api.Test
|
||||
import org.springframework.beans.factory.annotation.Autowired
|
||||
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc
|
||||
import org.springframework.boot.test.context.SpringBootTest
|
||||
import org.springframework.http.MediaType
|
||||
import org.springframework.test.web.servlet.MockMvc
|
||||
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete
|
||||
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get
|
||||
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post
|
||||
import org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath
|
||||
import org.springframework.test.web.servlet.result.MockMvcResultMatchers.status
|
||||
import java.util.UUID
|
||||
|
||||
@SpringBootTest
|
||||
@AutoConfigureMockMvc
|
||||
class TaskGeneratedMessageControllerTest {
|
||||
|
||||
@Autowired lateinit var mockMvc: MockMvc
|
||||
@Autowired lateinit var jwtService: JwtService
|
||||
@Autowired lateinit var generatedMessageHistoryRepository: GeneratedMessageHistoryRepository
|
||||
@Autowired lateinit var virtualEntityRepository: VirtualEntityRepository
|
||||
@Autowired lateinit var entityTaskRepository: EntityTaskRepository
|
||||
|
||||
@MockkBean lateinit var entityScheduler: EntityScheduler
|
||||
@MockkBean lateinit var llamaPreviewService: LlamaPreviewService
|
||||
|
||||
private fun authCookie() = Cookie("jwt", jwtService.generateToken())
|
||||
|
||||
@AfterEach
|
||||
fun cleanUp() {
|
||||
generatedMessageHistoryRepository.deleteAll()
|
||||
entityTaskRepository.deleteAll()
|
||||
virtualEntityRepository.deleteAll()
|
||||
}
|
||||
|
||||
private fun createTask(taskName: String = "Task"): EntityTask {
|
||||
val entity = virtualEntityRepository.save(
|
||||
VirtualEntity(name = "Entity A", email = "entity-a@condado.com", jobTitle = "Ops")
|
||||
)
|
||||
return entityTaskRepository.save(
|
||||
EntityTask(
|
||||
virtualEntity = entity,
|
||||
name = taskName,
|
||||
prompt = "Prompt",
|
||||
scheduleCron = "0 9 * * 1",
|
||||
emailLookback = "last_week"
|
||||
)
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun should_returnPersistedHistory_when_getByTaskId() {
|
||||
val task = createTask()
|
||||
generatedMessageHistoryRepository.save(
|
||||
GeneratedMessageHistory(
|
||||
task = task,
|
||||
label = "Message #1",
|
||||
content = "SUBJECT: A\\nBODY:\\nHello"
|
||||
)
|
||||
)
|
||||
|
||||
mockMvc.perform(get("/api/v1/tasks/${task.id}/generated-messages").cookie(authCookie()))
|
||||
.andExpect(status().isOk)
|
||||
.andExpect(jsonPath("$[0].taskId").value(task.id.toString()))
|
||||
.andExpect(jsonPath("$[0].label").value("Message #1"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun should_generateAndPersistMessage_when_generateEndpointCalled() {
|
||||
val task = createTask("Morning Blast")
|
||||
every { llamaPreviewService.generate(any()) } returns "SUBJECT: Generated\\nBODY:\\nFrom backend"
|
||||
|
||||
val payload = """
|
||||
{
|
||||
"entity": {
|
||||
"id": "${UUID.randomUUID()}",
|
||||
"name": "Entity A",
|
||||
"email": "entity@example.com",
|
||||
"jobTitle": "Ops",
|
||||
"personality": "Formal",
|
||||
"scheduleCron": "0 9 * * 1",
|
||||
"contextWindowDays": 3,
|
||||
"active": true
|
||||
},
|
||||
"task": {
|
||||
"entityId": "${UUID.randomUUID()}",
|
||||
"name": "Morning Blast",
|
||||
"prompt": "Talk about coffee",
|
||||
"scheduleCron": "0 8 * * 1-5",
|
||||
"emailLookback": "last_week"
|
||||
}
|
||||
}
|
||||
""".trimIndent()
|
||||
|
||||
mockMvc.perform(
|
||||
post("/api/v1/tasks/${task.id}/generated-messages/generate")
|
||||
.cookie(authCookie())
|
||||
.contentType(MediaType.APPLICATION_JSON)
|
||||
.content(payload)
|
||||
)
|
||||
.andExpect(status().isCreated)
|
||||
.andExpect(jsonPath("$.taskId").value(task.id.toString()))
|
||||
.andExpect(jsonPath("$.content").value("SUBJECT: Generated\\nBODY:\\nFrom backend"))
|
||||
.andExpect(jsonPath("$.label").value("Message #1"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun should_deleteHistoryItem_when_deleteEndpointCalled() {
|
||||
val task = createTask()
|
||||
val saved = generatedMessageHistoryRepository.save(
|
||||
GeneratedMessageHistory(
|
||||
task = task,
|
||||
label = "Message #1",
|
||||
content = "SUBJECT: A\\nBODY:\\nHello"
|
||||
)
|
||||
)
|
||||
|
||||
mockMvc.perform(delete("/api/v1/tasks/${task.id}/generated-messages/${saved.id}").cookie(authCookie()))
|
||||
.andExpect(status().isNoContent)
|
||||
|
||||
mockMvc.perform(get("/api/v1/tasks/${task.id}/generated-messages").cookie(authCookie()))
|
||||
.andExpect(status().isOk)
|
||||
.andExpect(jsonPath("$").isArray)
|
||||
.andExpect(jsonPath("$.length()").value(0))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user