async def test_false(self):
check = await self.commands_db.delete("is_false")
self.assertFalse(check)
check = await self.commands_db.delete_all(MockServer(id="54321"))
self.assertFalse(check)
# Inserting again should return false
check = await self.commands_db.insert("testing", "this is a test", MockServer())
self.assertFalse(check)
async def test_get_all(self):
await self.config_db.insert(MockServer(), "key2", "anothervalue")
await self.config_db.insert(MockServer(), "awesome",
"wow this is cool")
result = await self.config_db.get_all(MockServer())
self.assertIn(("mykey", "mytestvalue"), result)
self.assertIn(("key2", "anothervalue"), result)
self.assertIn(("awesome", "wow this is cool"), result)
async def asyncSetUp(self):
await self.commands_db.create_table()
await self.commands_db.insert("testing", "this is a test", MockServer())
self.cursor.execute("SELECT name, command, server_id from commands")
result = self.cursor.fetchone()
self.assertTupleEqual(("testing", "this is a test", "12345"), result)
async def asyncSetUp(self):
await self.config_db.create_table()
await self.config_db.insert(MockServer(), "mykey", "mytestvalue")
self.cursor.execute("SELECT key,value,server_id from config")
result = self.cursor.fetchone()
self.assertTupleEqual(("mykey", "mytestvalue", "12345"), result)
async def test_delete_all(self):
server = MockServer()
await self.commands_db.insert("another", "another command", server)
await self.commands_db.insert("test", "asdasdasd", server)
await self.commands_db.insert("타이니 봇", "안녕", server)
await self.commands_db.delete_all(server)
self.cursor.execute("SELECT name, command from commands")
result = self.cursor.fetchone()
self.assertIsNone(result)
def setUp(self):
self.server = MockServer()
# Make our own connection to test things.
# noinspection PyArgumentList
self.connection = sqlite3.connect("file::memory:?cache=shared",
uri=True)
self.cursor = self.connection.cursor()
self.cursor.execute("DROP TABLE IF EXISTS roles")
self.connection.commit()
self.roles_db = RolesDB("file::memory:?cache=shared", uri=True)
def setUpClass(cls) -> None:
super().setUpClass()
cls._server = MockServer()
cls._server_port = cls._server.port
target_server = threading.Thread(target=cls._server.create_server,
daemon=True)
target_server.start()
cls._request = WebRequests(cls._proxy_host, cls._server_port)
cls._answer = WebAnswers(cls._proxy_host, cls._server_port)
cls._proxy = ProxyServer(cls._proxy_host, 0, *['' for _ in range(4)])
cls._proxy_port = cls._proxy.port
cls._proxy._authenticate = MockClass()
cls._proxy._banner = MockClass()
cls._proxy._block = MockClass()
target_proxy = threading.Thread(target=cls._proxy.run, daemon=True)
target_proxy.start()
async def test_get_all(self):
server = MockServer()
await self.commands_db.insert("one", "testing", server)
await self.commands_db.insert("two", "testing", server)
await self.commands_db.insert("three", "another", server)
await self.commands_db.insert("four", "testing", server)
await self.commands_db.insert("five", "another", server)
for x in range(0, 2): # Loop twice to check cache for code coverage
result = await self.commands_db.get_all()
keys = result.keys()
self.assertIn("one", keys)
self.assertIn("two", keys)
self.assertIn("three", keys)
self.assertIn("four", keys)
self.assertIn("five", keys)
self.assertEqual(result["one"], "testing")
self.assertEqual(result["two"], "testing")
self.assertEqual(result["three"], "another")
self.assertEqual(result["four"], "testing")
self.assertEqual(result["five"], "another")
async def test_get(self):
result = await self.config_db.get(MockServer(), "mykey")
self.assertEqual("mytestvalue", result)
result = await self.config_db.get(MockServer(), "does_not_exist")
self.assertIsNone(result)
async def test_delete(self):
await self.config_db.delete(MockServer(), "mykey")
self.cursor.execute("SELECT key,value,server_id from config")
result = self.cursor.fetchone()
self.assertIsNone(result)
async def test_update(self):
await self.config_db.update(MockServer(), "mykey", "my new test value")
self.cursor.execute("SELECT key,value,server_id from config")
result = self.cursor.fetchone()
self.assertTupleEqual(("mykey", "my new test value", "12345"), result)