Test if account token is used correctly

1 files changed, 55 insertions, 0 deletions

diff --git a/mullvad-tests/tests/account.rs b/mullvad-tests/tests/account.rs
new file mode 100644
index 0000000000..88eda432fc
--- /dev/null
+++ b/mullvad-tests/tests/account.rs
@@ -0,0 +1,55 @@
+#![cfg(all(target_os = "linux", feature = "integration-tests"))]
+
+extern crate mullvad_tests;
+
+use std::fs::File;
+use std::io::{BufRead, BufReader};
+use std::path::Path;
+
+use mullvad_tests::mock_openvpn::search_openvpn_args;
+use mullvad_tests::{DaemonRunner, PathWatcher};
+
+#[test]
+fn uses_account_token() {
+    let mut daemon = DaemonRunner::spawn();
+    let mut rpc_client = daemon.rpc_client().unwrap();
+    let openvpn_args_file = daemon.mock_openvpn_args_file();
+    let mut openvpn_args_file_events = PathWatcher::watch(&openvpn_args_file).unwrap();
+
+    let specified_account = "123456";
+    rpc_client
+        .set_account(Some(specified_account.to_owned()))
+        .unwrap();
+    rpc_client.connect().unwrap();
+
+    openvpn_args_file_events.assert_create_write_close_sequence();
+
+    let account_token_sent_to_plugin = read_account_token(openvpn_args_file).unwrap();
+
+    assert_eq!(account_token_sent_to_plugin, specified_account);
+}
+
+fn read_account_token<P: AsRef<Path>>(openvpn_args_file_path: P) -> Result<String, String> {
+    let account_token_file_path = search_openvpn_args(openvpn_args_file_path, "--auth-user-pass")
+        .skip(1)
+        .next()
+        .ok_or_else(|| "Missing account token file parameter to Talpid OpenVPN plugin".to_owned())?
+        .map_err(|error| {
+            format!(
+                "Failed to read from mock OpenVPN command line file: {}",
+                error
+            )
+        })?;
+
+    let account_token_file = File::open(account_token_file_path)
+        .map_err(|error| format!("Failed to open account token file: {}", error))?;
+
+    let mut reader = BufReader::new(account_token_file);
+    let mut account = String::new();
+
+    reader
+        .read_line(&mut account)
+        .map_err(|error| format!("Failed to read from account token file: {}", error))?;
+
+    Ok(account.trim().to_owned())
+}