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
|
#
# Copyright (c) 2015 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
# We use this script to simulate the login of multiple concurrent users, with
# one login per second.
#
# To use it, you need:
# -ruby
# -pre-created users
#
# This can be run with `ruby concurrent_logins.rb <x>`
# where x is the number of users you want to login.
#
# It was created to measure login times internally on the application with
# varying number of users
require 'fileutils'
USER_PATTERN = "loadtest%d"
PASSWORD_PATTERN = "password_%d"
COUNT = ARGV[0].to_i
def header_options
"--header 'X-Requested-With: XMLHttpRequest' --header 'X-XSRF-TOKEN: blablabla'"
end
def cookies
"--cookie 'XSRF-TOKEN: blablabla'"
end
def curl_command
"curl --silent"
end
def login_command(user_index, username, password)
FileUtils.rm "/tmp/user#{user_index}.cookie", force: true
"#{curl_command} -X POST --data 'username=#{username}&password=#{password}' --cookie-jar /tmp/user#{user_index}.cookie -w '%{http_code}|%{time_total}' --output '/dev/null' #{cookies} #{header_options} http://localhost:3333/login"
end
def check_inbox(user_index)
"#{curl_command} --cookie /tmp/user#{user_index}.cookie http://localhost:3333"
end
def complete_login(user_index, user, password)
start = Time.now
login_response = `#{login_command(user_index, user, password)}`
status_code, total_time = login_response.split("|")
puts "Login request #{total_time}"
if status_code.to_i == 200
interstitial = Time.now
begin
sleep 0.05
inbox = `#{check_inbox(user_index)}`
end until /compose-trigger/.match(inbox)
puts "Login loading #{sprintf('%.3f',Time.now - interstitial)}"
puts "Login total #{sprintf('%.3f',Time.now - start)}"
else
puts "Login failed with #{status_code} #{sprintf('%.3f',Time.now - start)}"
end
end
def multi_login
ts = (1...(1+COUNT)).map do |user_index|
t = Thread.new do
username = USER_PATTERN % [user_index]
password = PASSWORD_PATTERN % [user_index]
complete_login(user_index, username, password)
end
sleep 1
t
end
ts.each(&:join)
end
multi_login
|
