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)