nirntests.py

from random import randint
import unittest
from lib import utilities
from lib import mundusfuncs

test_connection = utilities.create_connection("db/neodatabase")


class NirnTest(unittest.TestCase):

    def test_isTokenAvailable(self):
        token = utilities.returnSetting('server_token')

        self.assertTrue(token[1], token[0])

    # The functions within SCRIPT.PY need to be rewritten in order to return
    # values that can be tested using the NirnTest object

    # Function for GetUser
    def test_getUserTest(self):
        # Testing username saved in every database
        username = '@nirnbot'
        # Check if the query is executed successfully
        result = mundusfuncs.getUserInfo(username, test_connection)
        self.assertTrue(result[1], result[0])

    # Function for AddUser
    def test_addUserTest(self):
        # Dummy username
        username = '@anoymous' + str(randint(1, 1000))
        # Get the result
        result = mundusfuncs.addUsertoDb(username, test_connection)
        # Test the result
        self.assertTrue(result[1], result[0])
    # Function for Add Info

    def test_addDataTest(self):
        # Dummy username
        username = '@nirnbot'

        result = mundusfuncs.addData(
            username, ['website', 'https://dummy.org/'], test_connection)

        self.assertTrue(result[1], result[0])

    # Function for UpdateStatus
    def test_updateStatusTest(self):
        # Dummy username
        username = '@nirnbot'

        result = mundusfuncs.updateUserStatus(
            username, 'Testing...', test_connection)

        self.assertTrue(result[1], result[0])

    # Function for Get Status
    def test_getStatusTest(self):
        # Dummy Username
        username = '@nirnbot'

        result = mundusfuncs.getUserStatus(username, test_connection)

        self.assertTrue(result[1], result[0])

    # Function for Get Gas
    def test_getGas(self):
        result = mundusfuncs.returnGas()

        self.assertTrue(result[1], result[0])

    # Function for Display All Users
    def test_firstAdmin(self):
        # Dummy usernamer, should return false
        username = '@nirnbot'
        result = mundusfuncs.displayAll(username, test_connection)

        self.assertFalse(result[1], result[0])


if __name__ == '__main__':
    unittest.main()