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
|
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import WebDriverException
import multiprocessing
import argparse
class SeleniumTest():
def __init__(self, url, quiz_name):
self.driver = webdriver.Firefox()
self.quiz_name = quiz_name
self.url = url
def run_load_test(self, url, username, password):
try:
self.driver.delete_all_cookies()
self.driver.get(self.url)
self.login(username, password)
self.open_quiz()
self.close_quiz()
self.logout()
self.driver.close()
except Exception as e:
with open("/tmp/yaksh_load_test_log.txt", "ab") as f:
f.write('Username: {0}\nError: {1}\n'.format(username, e))
self.driver.close()
def login(self, username, password):
# get the username, password and submit form elements
username_elem = self.driver.find_element_by_id("id_username")
password_elem = self.driver.find_element_by_id("id_password")
submit_login_elem = self.driver.find_element_by_css_selector('button.btn')
# Type in the username, password and submit form
username_elem.send_keys(username)
password_elem.send_keys(password)
submit_login_elem.click()
def submit_answer(self, question_label, answer, loop_count=1):
self.driver.implicitly_wait(2)
for count in range(loop_count):
self.driver.find_element_by_link_text(question_label).click()
submit_answer_elem = self.driver.find_element_by_id("check")
self.driver.execute_script('editor.setValue({})'.format(answer))
submit_answer_elem.click()
def test_c_question(self, question_label):
# Incorrect Answer
loop_count = 10
answer = '\"int add(int a, int b, int c)\\n{return;}\"'
self.submit_answer(question_label, answer, loop_count)
# Infinite Loop
loop_count = 3
answer = '\"int add(int a, int b, int c)\\n{while(1){}}\"'
self.submit_answer(question_label, answer, loop_count)
# Correct Answer
loop_count = 1
answer = '\"int add(int a, int b, int c)\\n{return a + b + c;}\"'
self.submit_answer(question_label, answer, loop_count)
def test_python_question(self, question_label):
# Incorrect Answer
loop_count = 10
answer = '\"def is_palindrome(s):\\n return s\"'
self.submit_answer(question_label, answer, loop_count)
# Infinite Loop
loop_count = 3
answer = '\"while True:\\n pass"'
self.submit_answer(question_label, answer, loop_count)
# Correct Answer
loop_count = 1
answer = '\"def is_palindrome(s):\\n return s[::-1] == s\"'
self.submit_answer(question_label, answer, loop_count)
def test_bash_question(self, question_label):
# Incorrect Answer
loop_count = 10
answer = '\"#!/bin/bash\\nls\"'
self.submit_answer(question_label, answer, loop_count)
# Infinite Loop
loop_count = 3
answer = '\"#!/bin/bash\\nwhile [ 1 ]; do : ; done\"'
self.submit_answer(question_label, answer, loop_count)
# Correct Answer
loop_count = 1
answer = '\"#!/bin/bash\\ncat $1 | cut -d: -f2 | paste -d: $3 - $2\"'
self.submit_answer(question_label, answer, loop_count)
def open_quiz(self):
# open quiz link
quiz_link_elem = self.driver.find_element_by_link_text(self.quiz_name).click()
# Get page elements
start_exam_elem = WebDriverWait(self.driver, 5).until(
EC.presence_of_element_located((By.NAME, "start"))
)
start_exam_elem.click()
self.test_c_question(question_label=1)
self.test_python_question(question_label=3)
self.test_bash_question(question_label=2)
def close_quiz(self):
quit_link_elem = WebDriverWait(self.driver, 5).until(
EC.presence_of_element_located((By.ID, "login_again"))
)
quit_link_elem.click()
def logout(self):
logout_link_elem = WebDriverWait(self.driver, 5).until(
EC.presence_of_element_located((By.ID, "logout"))
)
logout_link_elem.click()
def user_gen(url, ids):
return [(url, 'User%d'%x, 'User%d'%x) for x in ids]
def wrap_run_load_test(args):
url = "http://yaksh.fossee.aero.iitb.ac.in/exam/"
selenium_test = SeleniumTest(url=url, quiz_name=quiz_name)
return selenium_test.run_load_test(*args)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('url', type=str, help="url of the website being tested")
parser.add_argument('start', type=int, help="Starting user id")
parser.add_argument("-n", "--number", type=int, default=10, help="number of users")
opts = parser.parse_args()
quiz_name = "Demo quiz"
selenium_test = SeleniumTest(url=opts.url, quiz_name=quiz_name)
pool = multiprocessing.Pool(opts.number)
pool.map(wrap_run_load_test, user_gen(opts.url, range(opts.start, opts.start + opts.number)))
pool.close()
pool.join()
|
