mirror of
https://github.com/torproject/gettor.git
synced 2025-02-12 22:38:57 +00:00
390120a0f9
Instead of processing link and help requests separately, we can process them together and use the command to determine the different subject and body messages. Removed unused code and simplified database command.
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
import pytest
|
|
import pytest_twisted
|
|
from datetime import datetime
|
|
from twisted.trial import unittest
|
|
|
|
from . import conftests
|
|
|
|
class DatabaseTests(unittest.TestCase):
|
|
|
|
# Fail any tests which take longer than 15 seconds.
|
|
timeout = 15
|
|
def setUp(self):
|
|
self.settings = conftests.options.parse_settings("en","tests/test.conf.json")
|
|
self.locales = conftests.strings.get_locales()
|
|
|
|
self.conn = conftests.SQLite3(self.settings.get("dbname"))
|
|
|
|
def tearDown(self):
|
|
print("tearDown()")
|
|
del self.conn
|
|
|
|
@pytest_twisted.inlineCallbacks
|
|
def add_dummy_requests(self, command, num):
|
|
now_str = datetime.now().strftime("%Y%m%d")
|
|
for i in range(0, num):
|
|
yield self.conn.new_request(
|
|
id='testid',
|
|
command=command,
|
|
platform='linux',
|
|
language='en',
|
|
service='email',
|
|
date=now_str,
|
|
status="ONHOLD",
|
|
)
|
|
|
|
@pytest_twisted.inlineCallbacks
|
|
def test_stored_locales(self):
|
|
locales = []
|
|
ls = yield self.conn.get_locales()
|
|
for l in ls:
|
|
locales.append(l[0])
|
|
self.assertIn('en-US', locales)
|
|
|
|
@pytest_twisted.inlineCallbacks
|
|
def test_requests(self):
|
|
now_str = datetime.now().strftime("%Y%m%d")
|
|
yield self.add_dummy_requests("links", 2)
|
|
yield self.add_dummy_requests("help", 1)
|
|
num = yield self.conn.get_num_requests("testid", "email")
|
|
self.assertEqual(num[0][0], 3)
|
|
|
|
requests = yield self.conn.get_requests("ONHOLD", "email")
|
|
for request in requests:
|
|
self.assertEqual(request[4], "email")
|
|
self.assertEqual(request[5], now_str)
|
|
self.assertEqual(request[6], "ONHOLD")
|
|
self.assertEqual(len(requests), 3)
|
|
|
|
yield self.conn.remove_request("testid", "email", now_str)
|
|
num = yield self.conn.get_num_requests("testid", "email")
|
|
self.assertEqual(num[0][0], 0)
|
|
|
|
@pytest_twisted.inlineCallbacks
|
|
def test_links(self):
|
|
links = yield self.conn.get_links("linux", "en-US", "ACTIVE")
|
|
|
|
for link in links:
|
|
self.assertEqual(link[1], "linux")
|
|
self.assertEqual(link[2], "en-US")
|
|
self.assertEqual(link[6], "ACTIVE")
|
|
self.assertIn(link[5], ["github", "gitlab"])
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|