mirror of
https://github.com/different-ai/openwork
synced 2026-04-25 17:15:34 +02:00
fix(desktop): remove stray token store tests (#1037)
This commit is contained in:
@@ -318,114 +318,3 @@ pub fn start_openwork_server(
|
||||
|
||||
Ok(OpenworkServerManager::snapshot_locked(&mut state))
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::{
|
||||
generate_token, load_openwork_server_token_store, load_or_create_workspace_tokens_at_path,
|
||||
persist_workspace_owner_token_at_path, save_openwork_server_token_store,
|
||||
PersistedOpenworkServerTokenStore, PersistedOpenworkServerTokens,
|
||||
OPENWORK_SERVER_TOKEN_STORE_VERSION,
|
||||
};
|
||||
use std::collections::HashMap;
|
||||
use std::fs;
|
||||
|
||||
#[test]
|
||||
fn round_trips_workspace_tokens() {
|
||||
let temp_dir = std::env::temp_dir().join(format!(
|
||||
"openwork-server-token-store-test-{}",
|
||||
uuid::Uuid::new_v4()
|
||||
));
|
||||
fs::create_dir_all(&temp_dir).expect("create temp dir");
|
||||
let path = temp_dir.join("openwork-server-tokens.json");
|
||||
|
||||
let mut workspaces = HashMap::new();
|
||||
workspaces.insert(
|
||||
"/tmp/workspace-a".to_string(),
|
||||
PersistedOpenworkServerTokens {
|
||||
client_token: "client-token".to_string(),
|
||||
host_token: "host-token".to_string(),
|
||||
owner_token: Some("owner-token".to_string()),
|
||||
updated_at: 123,
|
||||
},
|
||||
);
|
||||
let store = PersistedOpenworkServerTokenStore {
|
||||
version: OPENWORK_SERVER_TOKEN_STORE_VERSION,
|
||||
workspaces,
|
||||
};
|
||||
|
||||
save_openwork_server_token_store(&path, &store).expect("save token store");
|
||||
let loaded = load_openwork_server_token_store(&path).expect("load token store");
|
||||
|
||||
assert_eq!(loaded.version, OPENWORK_SERVER_TOKEN_STORE_VERSION);
|
||||
assert_eq!(loaded.workspaces.len(), 1);
|
||||
let tokens = loaded
|
||||
.workspaces
|
||||
.get("/tmp/workspace-a")
|
||||
.expect("workspace tokens present");
|
||||
assert_eq!(tokens.client_token, "client-token");
|
||||
assert_eq!(tokens.host_token, "host-token");
|
||||
assert_eq!(tokens.owner_token.as_deref(), Some("owner-token"));
|
||||
|
||||
let _ = fs::remove_file(&path);
|
||||
let _ = fs::remove_dir_all(&temp_dir);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn legacy_unpersisted_tokens_change_between_starts() {
|
||||
let first_client = generate_token();
|
||||
let second_client = generate_token();
|
||||
let first_owner = generate_token();
|
||||
let second_owner = generate_token();
|
||||
|
||||
println!("before collaborator launch1={first_client}");
|
||||
println!("before collaborator launch2={second_client}");
|
||||
println!("before collaborator same={}", first_client == second_client);
|
||||
println!("before owner launch1={first_owner}");
|
||||
println!("before owner launch2={second_owner}");
|
||||
println!("before owner same={}", first_owner == second_owner);
|
||||
|
||||
assert_ne!(first_client, second_client);
|
||||
assert_ne!(first_owner, second_owner);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn reuses_tokens_for_the_same_workspace_after_restart() {
|
||||
let temp_dir = std::env::temp_dir().join(format!(
|
||||
"openwork-server-token-store-test-{}",
|
||||
uuid::Uuid::new_v4()
|
||||
));
|
||||
fs::create_dir_all(&temp_dir).expect("create temp dir");
|
||||
let path = temp_dir.join("openwork-server-tokens.json");
|
||||
|
||||
let first = load_or_create_workspace_tokens_at_path(&path, "/tmp/workspace-a")
|
||||
.expect("create first token set");
|
||||
persist_workspace_owner_token_at_path(&path, "/tmp/workspace-a", "owner-token")
|
||||
.expect("persist owner token");
|
||||
let second = load_or_create_workspace_tokens_at_path(&path, "/tmp/workspace-a")
|
||||
.expect("load second token set");
|
||||
|
||||
println!("after collaborator launch1={}", first.client_token);
|
||||
println!("after collaborator launch2={}", second.client_token);
|
||||
println!(
|
||||
"after collaborator same={}",
|
||||
first.client_token == second.client_token
|
||||
);
|
||||
println!("after owner launch1={}", "owner-token");
|
||||
println!(
|
||||
"after owner launch2={}",
|
||||
second.owner_token.as_deref().unwrap_or("")
|
||||
);
|
||||
println!(
|
||||
"after owner same={}",
|
||||
second.owner_token.as_deref() == Some("owner-token")
|
||||
);
|
||||
|
||||
assert_eq!(first.client_token, second.client_token);
|
||||
assert_eq!(first.host_token, second.host_token);
|
||||
assert_eq!(second.owner_token.as_deref(), Some("owner-token"));
|
||||
|
||||
let _ = fs::remove_file(&path);
|
||||
let _ = fs::remove_dir_all(&temp_dir);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user