216 lines
10 KiB
Rust
216 lines
10 KiB
Rust
use chrono::{TimeZone, Utc};
|
|
use merge_request::*;
|
|
use rusqlite::Connection;
|
|
use std::sync::{Arc, Mutex};
|
|
#[derive(Clone)]
|
|
struct Assignments(Arc<Mutex<CurrentAssignment>>);
|
|
impl AssignmentSource for Assignments {
|
|
fn current_assignment(&self, _: &str, _: &str) -> Result<Option<CurrentAssignment>, String> {
|
|
Ok(Some(self.0.lock().unwrap().clone()))
|
|
}
|
|
}
|
|
struct Repositories;
|
|
impl RepositorySource for Repositories {
|
|
fn repository_belongs_to_workspace(&self, w: &str, r: &str) -> Result<bool, String> {
|
|
Ok(w == "W" && r == "R")
|
|
}
|
|
}
|
|
fn at(s: u32) -> chrono::DateTime<Utc> {
|
|
Utc.with_ymd_and_hms(2026, 7, 26, 12, 0, s)
|
|
.single()
|
|
.unwrap()
|
|
}
|
|
fn auth() -> MergeRequestAuth {
|
|
MergeRequestAuth {
|
|
workspace_id: "W".into(),
|
|
repository_id: "R".into(),
|
|
runtime_id: "runtime".into(),
|
|
worker_id: "coder".into(),
|
|
assignment_id: "A".into(),
|
|
}
|
|
}
|
|
fn fixture() -> (tempfile::TempDir, MergeRequestStore) {
|
|
let d = tempfile::tempdir().unwrap();
|
|
let p = d.path().join("db");
|
|
let c = Connection::open(&p).unwrap();
|
|
c.execute_batch("CREATE TABLE workspaces(workspace_id TEXT PRIMARY KEY);CREATE TABLE repositories(workspace_id TEXT,repository_id TEXT,PRIMARY KEY(workspace_id,repository_id));CREATE TABLE typed_tickets(workspace_id TEXT,ticket_id TEXT,workflow_state TEXT,workflow_state_explicit INTEGER,updated_at TEXT,PRIMARY KEY(workspace_id,ticket_id));CREATE TABLE typed_ticket_events(workspace_id TEXT,ticket_id TEXT,event_index INTEGER,kind TEXT,author TEXT,at TEXT,from_state TEXT,to_state TEXT,heading TEXT,body TEXT,PRIMARY KEY(workspace_id,ticket_id,event_index));CREATE TABLE typed_ticket_event_attributes(workspace_id TEXT,ticket_id TEXT,event_index INTEGER,key TEXT,value TEXT,PRIMARY KEY(workspace_id,ticket_id,event_index,key));INSERT INTO workspaces VALUES('W');INSERT INTO repositories VALUES('W','R');INSERT INTO typed_tickets VALUES('W','T','inprogress',1,'t');").unwrap();
|
|
drop(c);
|
|
let a = Assignments(Arc::new(Mutex::new(CurrentAssignment {
|
|
assignment_id: "A".into(),
|
|
ticket_id: "T".into(),
|
|
runtime_id: "runtime".into(),
|
|
worker_id: "coder".into(),
|
|
})));
|
|
let s = MergeRequestStore::open(&p, Arc::new(a), Arc::new(Repositories)).unwrap();
|
|
(d, s)
|
|
}
|
|
fn open(s: &MergeRequestStore) {
|
|
s.open_merge_request(OpenMergeRequest {
|
|
merge_request_id: "MR".into(),
|
|
ticket_id: "T".into(),
|
|
repository_id: "R".into(),
|
|
selector_from: "work/t".into(),
|
|
selector_to: "develop".into(),
|
|
summary: "summary".into(),
|
|
auth: auth(),
|
|
now: at(1),
|
|
})
|
|
.unwrap();
|
|
}
|
|
fn request(s: &MergeRequestStore, subject: &str, token: &str) -> ReviewRequestedEvent {
|
|
s.register_reviewer_child_session(RegisterReviewerChildSession {
|
|
workspace_id: "W".into(),
|
|
parent_runtime_id: "runtime".into(),
|
|
parent_worker_id: "coder".into(),
|
|
child_session_id: format!("child-{token}"),
|
|
reviewer_profile: "builtin:reviewer".into(),
|
|
now: at(2),
|
|
})
|
|
.unwrap();
|
|
s.request_review(RequestMergeRequestReview {
|
|
ticket_id: "T".into(),
|
|
subject_ref: subject.into(),
|
|
child_session_id: format!("child-{token}"),
|
|
capability_token: token.into(),
|
|
auth: auth(),
|
|
now: at(3),
|
|
})
|
|
.unwrap()
|
|
.request_event
|
|
}
|
|
fn approve(s: &MergeRequestStore, subject: &str, token: &str) -> ReviewEvent {
|
|
request(s, subject, token);
|
|
s.submit_review(SubmitMergeRequestReview {
|
|
ticket_id: "T".into(),
|
|
current_subject_ref: subject.into(),
|
|
capability_token: token.into(),
|
|
decision: ReviewDecision::Approve,
|
|
body: "approved".into(),
|
|
findings: vec![],
|
|
now: at(4),
|
|
})
|
|
.unwrap()
|
|
}
|
|
#[test]
|
|
fn selectors_thread_and_completion_have_no_revision_or_commit_api() {
|
|
let (_d, s) = fixture();
|
|
open(&s);
|
|
let review = approve(&s, "opaque-source-ref", "token");
|
|
let ready = s
|
|
.readiness(ReadinessCheck {
|
|
ticket_id: "T".into(),
|
|
current_subject_ref: Some("opaque-source-ref".into()),
|
|
auth: auth(),
|
|
})
|
|
.unwrap();
|
|
assert!(ready.ready);
|
|
let merged = s
|
|
.complete(CompleteMergeRequest {
|
|
ticket_id: "T".into(),
|
|
operation_id: "op".into(),
|
|
approval_event_id: review.event_id,
|
|
current_subject_ref: "opaque-source-ref".into(),
|
|
target_ref_before: "old-target-ref".into(),
|
|
target_ref_after: "new-target-ref".into(),
|
|
strategy: MergeStrategy::FastForward,
|
|
resolution: ConflictResolution::None,
|
|
auth: auth(),
|
|
now: at(5),
|
|
})
|
|
.unwrap();
|
|
assert_eq!(merged.approved_source_ref, "opaque-source-ref");
|
|
let mr = s.get("W", "T").unwrap();
|
|
assert_eq!(mr.selector_from.as_deref(), Some("work/t"));
|
|
assert_eq!(mr.state, MergeRequestState::Merged);
|
|
let json = serde_json::to_string(&mr).unwrap();
|
|
for banned in [
|
|
"revision_id",
|
|
"attempt_id",
|
|
"base_commit",
|
|
"head_commit",
|
|
"source_commit",
|
|
"result_commit",
|
|
"current_revision",
|
|
] {
|
|
assert!(!json.contains(banned), "{banned} in {json}")
|
|
}
|
|
}
|
|
#[test]
|
|
fn source_move_cancels_submission_and_old_approval_is_reusable_when_source_returns() {
|
|
let (_d, s) = fixture();
|
|
open(&s);
|
|
let approved = approve(&s, "source-a", "one");
|
|
request(&s, "source-b", "two");
|
|
assert!(
|
|
s.submit_review(SubmitMergeRequestReview {
|
|
ticket_id: "T".into(),
|
|
current_subject_ref: "source-c".into(),
|
|
capability_token: "two".into(),
|
|
decision: ReviewDecision::Approve,
|
|
body: "stale".into(),
|
|
findings: vec![],
|
|
now: at(6)
|
|
})
|
|
.is_err()
|
|
);
|
|
let mr = s.get("W", "T").unwrap();
|
|
assert!(
|
|
mr.thread
|
|
.iter()
|
|
.any(|e| matches!(e, MergeRequestThreadEvent::ReviewCancelled(_)))
|
|
);
|
|
assert_eq!(
|
|
mr.effective_review("source-a").map(|r| &r.event_id),
|
|
Some(&approved.event_id)
|
|
);
|
|
}
|
|
#[test]
|
|
fn review_revocation_invalidates_readiness() {
|
|
let (_d, s) = fixture();
|
|
open(&s);
|
|
let review = approve(&s, "source", "one");
|
|
s.revoke_review(RevokeMergeRequestReview {
|
|
ticket_id: "T".into(),
|
|
review_event_id: review.event_id,
|
|
reason: "bad evidence".into(),
|
|
auth: auth(),
|
|
now: at(7),
|
|
})
|
|
.unwrap();
|
|
let r = s
|
|
.readiness(ReadinessCheck {
|
|
ticket_id: "T".into(),
|
|
current_subject_ref: Some("source".into()),
|
|
auth: auth(),
|
|
})
|
|
.unwrap();
|
|
assert!(!r.ready);
|
|
}
|
|
|
|
#[test]
|
|
fn v11_migration_preserves_review_events_and_requires_selector_repair() {
|
|
let c = Connection::open_in_memory().unwrap();
|
|
c.execute_batch("CREATE TABLE repositories(workspace_id TEXT,repository_id TEXT,PRIMARY KEY(workspace_id,repository_id));CREATE TABLE typed_tickets(workspace_id TEXT,ticket_id TEXT,PRIMARY KEY(workspace_id,ticket_id));INSERT INTO repositories VALUES('W','R');INSERT INTO typed_tickets VALUES('W','T');CREATE TABLE merge_request_schema(singleton INTEGER PRIMARY KEY,version INTEGER);INSERT INTO merge_request_schema VALUES(1,11);CREATE TABLE merge_requests(workspace_id TEXT,merge_request_id TEXT,repository_id TEXT,state TEXT,target_ref_selector TEXT,current_revision_ordinal INTEGER,current_revision_id TEXT,created_at TEXT,updated_at TEXT,merged_revision_id TEXT,merged_at TEXT);CREATE TABLE merge_request_ticket_relations(workspace_id TEXT,merge_request_id TEXT,ticket_id TEXT,relation_kind TEXT,created_at TEXT);CREATE TABLE merge_request_revisions(workspace_id TEXT,merge_request_id TEXT,revision_id TEXT,ordinal INTEGER,base_commit TEXT,head_commit TEXT,diff_digest TEXT,summary TEXT,assignment_id TEXT,created_at TEXT);CREATE TABLE merge_request_revision_paths(workspace_id TEXT,merge_request_id TEXT,revision_id TEXT,ordinal INTEGER,path TEXT);CREATE TABLE merge_request_reviewer_child_sessions(workspace_id TEXT,child_session_id TEXT,parent_runtime_id TEXT,parent_worker_id TEXT,reviewer_profile TEXT,registered_at TEXT);CREATE TABLE merge_request_review_attempts(workspace_id TEXT,attempt_id TEXT,merge_request_id TEXT,ticket_id TEXT,revision_id TEXT,revision_ordinal INTEGER,parent_assignment_id TEXT,parent_runtime_id TEXT,parent_worker_id TEXT,child_session_id TEXT,reviewer_effective_profile TEXT,capability_token TEXT,status TEXT,created_at TEXT,consumed_at TEXT);CREATE TABLE merge_request_reviews(workspace_id TEXT,attempt_id TEXT,merge_request_id TEXT,revision_id TEXT,decision TEXT,body TEXT,submitted_at TEXT);CREATE TABLE merge_request_review_findings(workspace_id TEXT,attempt_id TEXT,ordinal INTEGER,severity TEXT,code TEXT,path TEXT,line INTEGER,body TEXT);CREATE TABLE merge_request_completion_operations(workspace_id TEXT,operation_id TEXT,ticket_id TEXT,revision_id TEXT,authority_kind TEXT,implementation_assignment_id TEXT,completion_actor_runtime_id TEXT,completion_actor_worker_id TEXT,target_commit TEXT,source_commit TEXT,result_commit TEXT,strategy TEXT,resolution TEXT,fingerprint TEXT,status TEXT,result_ticket_state TEXT,created_at TEXT,updated_at TEXT);INSERT INTO merge_requests VALUES('W','MR','R','open','develop',1,'V','2026-07-26T12:00:00Z','2026-07-26T12:00:00Z',NULL,NULL);INSERT INTO merge_request_ticket_relations VALUES('W','MR','T','implements','2026-07-26T12:00:00Z');INSERT INTO merge_request_revisions VALUES('W','MR','V',1,'base','subject','digest','summary','A','2026-07-26T12:00:00Z');INSERT INTO merge_request_review_attempts VALUES('W','AT','MR','T','V',1,'A','runtime','coder','child','builtin:reviewer','token','submitted','2026-07-26T12:00:00Z','2026-07-26T12:00:01Z');INSERT INTO merge_request_reviews VALUES('W','AT','MR','V','approve','approved','2026-07-26T12:00:01Z');").unwrap();
|
|
merge_request::migrate(&c).unwrap();
|
|
let selector: Option<String> = c
|
|
.query_row("SELECT selector_from FROM merge_requests", [], |r| r.get(0))
|
|
.unwrap();
|
|
assert!(selector.is_none());
|
|
let kinds: String = c
|
|
.query_row(
|
|
"SELECT group_concat(kind,',') FROM merge_request_thread_events ORDER BY sequence",
|
|
[],
|
|
|r| r.get(0),
|
|
)
|
|
.unwrap();
|
|
assert_eq!(kinds, "review_requested,review");
|
|
let old: bool = c
|
|
.query_row(
|
|
"SELECT EXISTS(SELECT 1 FROM sqlite_master WHERE name='merge_request_revisions')",
|
|
[],
|
|
|r| r.get(0),
|
|
)
|
|
.unwrap();
|
|
assert!(!old);
|
|
}
|