pulzr 0.3.2

A http load testing tool for performance testing.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236

use anyhow::Result;
use rand::prelude::*;
use std::fs;
use std::path::Path;

pub struct UserAgentManager {
    agents: Vec<String>,
    use_random: bool,
}

impl UserAgentManager {
    pub fn new(custom_ua: Option<String>, random_ua: bool, ua_file: Option<&Path>) -> Result<Self> {
        let agents = if let Some(ua) = custom_ua {
            vec![ua]
        } else if let Some(file_path) = ua_file {
            Self::load_from_file(file_path)?
        } else if random_ua {
            Self::default_user_agents()
        } else {
            vec!["pulzr/0.1.0".to_string()]
        };

        Ok(Self {
            agents,
            use_random: random_ua,
        })
    }

    pub fn get_user_agent(&self) -> &str {
        if self.use_random && self.agents.len() > 1 {
            self.agents.choose(&mut rand::rng()).unwrap()
        } else {
            &self.agents[0]
        }
    }

    fn load_from_file(file_path: &Path) -> Result<Vec<String>> {
        let content = fs::read_to_string(file_path)?;
        Ok(content
            .lines()
            .map(|line| line.trim().to_string())
            .filter(|line| !line.is_empty())
            .collect())
    }

    fn default_user_agents() -> Vec<String> {
        vec![
            // Chrome on Windows
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36".to_string(),
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36".to_string(),
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36".to_string(),

            // Chrome on macOS
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36".to_string(),
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36".to_string(),

            // Firefox on Windows
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:120.0) Gecko/20100101 Firefox/120.0".to_string(),
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:119.0) Gecko/20100101 Firefox/119.0".to_string(),

            // Firefox on macOS
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:120.0) Gecko/20100101 Firefox/120.0".to_string(),
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:119.0) Gecko/20100101 Firefox/119.0".to_string(),

            // Safari on macOS
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.1 Safari/605.1.15".to_string(),
            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Safari/605.1.15".to_string(),

            // Edge on Windows
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36 Edg/120.0.0.0".to_string(),
            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36 Edg/119.0.0.0".to_string(),

            // Chrome on Linux
            "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36".to_string(),
            "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36".to_string(),

            // Firefox on Linux
            "Mozilla/5.0 (X11; Linux x86_64; rv:120.0) Gecko/20100101 Firefox/120.0".to_string(),
            "Mozilla/5.0 (X11; Linux x86_64; rv:119.0) Gecko/20100101 Firefox/119.0".to_string(),

            // Mobile Chrome on Android
            "Mozilla/5.0 (Linux; Android 14; SM-G991B) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Mobile Safari/537.36".to_string(),
            "Mozilla/5.0 (Linux; Android 13; SM-G991B) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Mobile Safari/537.36".to_string(),

            // Mobile Safari on iOS
            "Mozilla/5.0 (iPhone; CPU iPhone OS 17_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.1 Mobile/15E148 Safari/604.1".to_string(),
            "Mozilla/5.0 (iPhone; CPU iPhone OS 17_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Mobile/15E148 Safari/604.1".to_string(),

            // iPad Safari
            "Mozilla/5.0 (iPad; CPU OS 17_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.1 Mobile/15E148 Safari/604.1".to_string(),
            "Mozilla/5.0 (iPad; CPU OS 17_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.0 Mobile/15E148 Safari/604.1".to_string(),

            // Samsung Internet
            "Mozilla/5.0 (Linux; Android 14; SM-G991B) AppleWebKit/537.36 (KHTML, like Gecko) SamsungBrowser/23.0 Chrome/115.0.0.0 Mobile Safari/537.36".to_string(),
            "Mozilla/5.0 (Linux; Android 13; SM-G991B) AppleWebKit/537.36 (KHTML, like Gecko) SamsungBrowser/22.0 Chrome/114.0.0.0 Mobile Safari/537.36".to_string(),
        ]
    }

    pub fn get_agent_count(&self) -> usize {
        self.agents.len()
    }

    pub fn is_random(&self) -> bool {
        self.use_random
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::io::Write;
    use tempfile::NamedTempFile;

    #[test]
    fn test_default_user_agent() {
        let manager = UserAgentManager::new(None, false, None).unwrap();
        assert_eq!(manager.get_user_agent(), "pulzr/0.1.0");
        assert!(!manager.is_random());
        assert_eq!(manager.get_agent_count(), 1);
    }

    #[test]
    fn test_custom_user_agent() {
        let custom_ua = "MyCustomAgent/1.0".to_string();
        let manager = UserAgentManager::new(Some(custom_ua.clone()), false, None).unwrap();
        assert_eq!(manager.get_user_agent(), custom_ua);
        assert!(!manager.is_random());
        assert_eq!(manager.get_agent_count(), 1);
    }

    #[test]
    fn test_random_user_agents() {
        let manager = UserAgentManager::new(None, true, None).unwrap();
        assert!(manager.is_random());
        assert!(manager.get_agent_count() > 20); // Should have many default agents

        // Get multiple user agents and verify they can vary
        let mut agents = std::collections::HashSet::new();
        for _ in 0..50 {
            agents.insert(manager.get_user_agent().to_string());
        }

        // With random selection, we should get some variety (though not guaranteed)
        // This test might occasionally fail due to randomness, but very unlikely
        assert!(!agents.is_empty());
    }

    #[test]
    fn test_user_agent_from_file() -> Result<()> {
        let mut temp_file = NamedTempFile::new()?;
        writeln!(temp_file, "Agent1/1.0")?;
        writeln!(temp_file, "Agent2/2.0")?;
        writeln!(temp_file)?; // Empty line should be filtered out
        writeln!(temp_file, "Agent3/3.0")?;
        temp_file.flush()?;

        let manager = UserAgentManager::new(None, false, Some(temp_file.path()))?;
        assert_eq!(manager.get_agent_count(), 3);
        assert_eq!(manager.get_user_agent(), "Agent1/1.0");
        assert!(!manager.is_random());

        Ok(())
    }

    #[test]
    fn test_random_user_agent_from_file() -> Result<()> {
        let mut temp_file = NamedTempFile::new()?;
        writeln!(temp_file, "Agent1/1.0")?;
        writeln!(temp_file, "Agent2/2.0")?;
        writeln!(temp_file, "Agent3/3.0")?;
        temp_file.flush()?;

        let manager = UserAgentManager::new(None, true, Some(temp_file.path()))?;
        assert_eq!(manager.get_agent_count(), 3);
        assert!(manager.is_random());

        Ok(())
    }

    #[test]
    fn test_file_not_found() {
        let result = UserAgentManager::new(
            None,
            false,
            Some(std::path::Path::new("/nonexistent/file.txt")),
        );
        assert!(result.is_err());
    }

    #[test]
    fn test_empty_file() -> Result<()> {
        let temp_file = NamedTempFile::new()?;
        // Don't write anything to the file

        let manager = UserAgentManager::new(None, false, Some(temp_file.path()))?;
        assert_eq!(manager.get_agent_count(), 0);

        Ok(())
    }

    #[test]
    fn test_custom_user_agent_overrides_random() {
        let custom_ua = "CustomAgent/1.0".to_string();
        let manager = UserAgentManager::new(Some(custom_ua.clone()), true, None).unwrap();
        assert_eq!(manager.get_user_agent(), custom_ua);
        assert!(manager.is_random()); // Random flag is preserved, but only one agent so no randomness
        assert_eq!(manager.get_agent_count(), 1);
    }

    #[test]
    fn test_default_user_agents_count() {
        let agents = UserAgentManager::default_user_agents();
        assert!(agents.len() >= 20); // Should have at least 20 default agents

        // Verify all agents are non-empty
        for agent in &agents {
            assert!(!agent.is_empty());
            assert!(agent.len() > 10); // Reasonable minimum length
        }
    }

    #[test]
    fn test_file_with_whitespace() -> Result<()> {
        let mut temp_file = NamedTempFile::new()?;
        writeln!(temp_file, "  Agent1/1.0  ")?; // Leading/trailing whitespace
        writeln!(temp_file, "\t\tAgent2/2.0\t\t")?; // Tabs
        writeln!(temp_file, "   ")?; // Only whitespace - should be filtered
        temp_file.flush()?;

        let manager = UserAgentManager::new(None, false, Some(temp_file.path()))?;
        assert_eq!(manager.get_agent_count(), 2);
        assert_eq!(manager.get_user_agent(), "Agent1/1.0");

        Ok(())
    }
}