1 files changed, 60 insertions, 0 deletions
diff --git a/yaksh/tests/load_test.py b/yaksh/tests/load_test.py
new file mode 100644
index 0000000..fd90b11
--- /dev/null
+++ b/yaksh/tests/load_test.py
@@ -0,0 +1,60 @@
+import os
+import signal
+import subprocess
+from datetime import datetime
+import pytz
+from threading import Thread
+from selenium.webdriver.firefox.webdriver import WebDriver
+
+from django.contrib.staticfiles.testing import StaticLiveServerTestCase
+
+from yaksh.models import User, Profile, Question, Quiz, Course, QuestionPaper, TestCase
+from selenium_test import SeleniumTest
+
+from yaksh.code_server import ServerPool #SERVER_POOL_PORT, SERVER_PORTS
+from yaksh import settings
+# from django.core.management import call_command
+from yaksh.management.commands.create_demo_course import create_demo_course
+
+
+CUR_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+class MySeleniumTests(StaticLiveServerTestCase):
+    @classmethod
+    def setUpClass(cls):
+        super(MySeleniumTests, cls).setUpClass()
+        # setup a demo code server
+        settings.code_evaluators['python']['standardtestcase'] = \
+            "yaksh.python_assertion_evaluator.PythonAssertionEvaluator"
+        settings.code_evaluators['c']['standardtestcase'] = \
+            "yaksh.cpp_code_evaluator.CppCodeEvaluator"
+        settings.code_evaluators['bash']['standardtestcase'] = \
+            "yaksh.bash_code_evaluator.BashCodeEvaluator"
+        settings.SERVER_POOL_PORT = 53578
+        code_server_pool = ServerPool(ports=settings.SERVER_PORTS, pool_port=settings.SERVER_POOL_PORT)
+        cls.code_server_pool = code_server_pool
+        cls.code_server_thread = t = Thread(target=code_server_pool.run)
+        t.start()
+
+        demo_course_setup = create_demo_course()
+
+    @classmethod
+    def tearDownClass(cls):
+        # User.objects.all().delete()
+        # Question.objects.all().delete()
+        # Quiz.objects.all().delete()
+        # Course.objects.all().delete()
+
+        settings.SERVER_POOL_PORT = 53579
+
+        cls.code_server_pool.stop()
+        cls.code_server_thread.join()
+
+        super(MySeleniumTests, cls).tearDownClass()
+
+    def test_load(self):
+        url = '%s%s' % (self.live_server_url, '/exam/login/')
+        quiz_name = "Demo_quiz"
+        selenium_test = SeleniumTest(url=url, quiz_name=quiz_name)
+        selenium_test.run_load_test(url=url, username='demo_student', password='demo_student')