82 lines
3.2 KiB
Python
82 lines
3.2 KiB
Python
from django.utils import unittest
|
|
import json
|
|
from django.test.client import Client
|
|
from django.contrib.auth.models import User
|
|
from minecraft.models import MinecraftProfile, Server, PlayerSession
|
|
|
|
class MOTDTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.client = Client()
|
|
|
|
def testUnregisteredUser(self):
|
|
response = json.loads(self.client.get('/api/motd/NewUser').content)
|
|
self.assertIsInstance(response['motd'], list)
|
|
|
|
class BalanceTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.client = Client()
|
|
self.user = User.objects.create_user('ValidUsername', 'test@example.com')
|
|
self.user.set_password('password')
|
|
self.user.save()
|
|
self.user.minecraftprofile.mc_username = "ValidUsername"
|
|
self.user.minecraftprofile.save()
|
|
self.user.minecraftprofile.currencyaccount.balance = 1000
|
|
self.user.minecraftprofile.currencyaccount.save()
|
|
|
|
def tearDown(self):
|
|
self.user.delete()
|
|
|
|
def testWithoutLogin(self):
|
|
result = self.client.get('/api/balance').content
|
|
self.assertEqual(result, "")
|
|
|
|
def testWithLogin(self):
|
|
self.client.login(username=self.user.username, password='password')
|
|
response = json.loads(self.client.get('/api/balance').content)
|
|
self.assertEqual(response['balance'], 1000)
|
|
|
|
|
|
class WhitelistTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.client = Client()
|
|
self.user = User.objects.create_user('ValidUsername', 'test@example.com')
|
|
self.user.minecraftprofile.mc_username = "ValidUsername"
|
|
self.user.minecraftprofile.save()
|
|
|
|
def tearDown(self):
|
|
self.user.delete()
|
|
|
|
def testValidProfile(self):
|
|
response = json.loads(self.client.get('/api/validate/ValidUsername').content)
|
|
self.assertEqual(response['valid'], True)
|
|
|
|
def testInvalidProfile(self):
|
|
response = json.loads(self.client.get('/api/validate/InvalidUser').content)
|
|
self.assertEqual(response['valid'], False)
|
|
|
|
class SessionTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.client = Client()
|
|
self.user = User.objects.create_user('ValidUsername', 'test@example.com')
|
|
self.user.minecraftprofile.mc_username = "ValidUsername"
|
|
self.user.minecraftprofile.save()
|
|
self.server = Server.objects.create(hostname='localhost')
|
|
|
|
def tearDown(self):
|
|
self.user.delete()
|
|
self.server.delete()
|
|
|
|
def testSessionStart(self):
|
|
resp = self.client.post('/api/session', {'hostname':self.server.hostname, 'player':self.user.minecraftprofile.mc_username, 'ip': '127.0.0.1'})
|
|
self.assertEqual(resp.status_code, 200)
|
|
session = json.loads(resp.content)
|
|
sessionObj = PlayerSession.objects.get(id__exact=session['session'])
|
|
|
|
def testSessionEnd(self):
|
|
resp = self.client.post('/api/session', {'hostname':self.server.hostname, 'player':self.user.minecraftprofile.mc_username, 'ip': '127.0.0.1'})
|
|
session = json.loads(resp.content)
|
|
resp = self.client.put('/api/session', {'session':session['session']})
|
|
self.assertEqual(resp.status_code, 200)
|
|
sessionObj = PlayerSession.objects.get(id__exact=session['session'])
|
|
self.assertNotEqual(sessionObj.end, None)
|