From 687fedfffa402e049c59668f2a6248ad2ba17910 Mon Sep 17 00:00:00 2001 From: Christopher Baklid Date: Sun, 20 May 2018 21:11:18 +0200 Subject: Tests directory (#73) moves all tests into a testing directory and splits the tests into separate files --- tests/test_database.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 tests/test_database.py (limited to 'tests/test_database.py') diff --git a/tests/test_database.py b/tests/test_database.py new file mode 100644 index 00000000..237cd68d --- /dev/null +++ b/tests/test_database.py @@ -0,0 +1,31 @@ +from tests import SiteTest + +class DatabaseTests(SiteTest): + """ Test cases for the database module """ + def test_table_actions(self): + import string + import secrets + from pysite.database import RethinkDB + + alphabet = string.ascii_letters + generated_table_name = ''.join(secrets.choice(alphabet) for i in range(8)) + + rdb = RethinkDB() + # Create table name and expect it to work + result = rdb.create_table(generated_table_name) + self.assertEqual(result, True) + + # Create the same table name and expect it to already exist + result = rdb.create_table(generated_table_name) + self.assertEqual(result, False) + + # Drop table and expect it to work + result = rdb.drop_table(generated_table_name) + self.assertEqual(result, True) + + # Drop the same table and expect it to already be gone + result = rdb.drop_table(generated_table_name) + self.assertEqual(result, False) + + # This is to get some more code coverage + self.assertEqual(rdb.teardown_request('_'), None) -- cgit v1.2.3