import os
import unittest
import tempfile
import nyaa


class NyaaTestCase(unittest.TestCase):

	def setUp(self):
		self.db, nyaa.app.config['DATABASE'] = tempfile.mkstemp()
		nyaa.app.config['TESTING'] = True
		self.app = nyaa.app.test_client()
		with nyaa.app.app_context():
			nyaa.db.create_all()

	def tearDown(self):
		os.close(self.db)
		os.unlink(nyaa.app.config['DATABASE'])

	def test_empty_db(self):
		rv = self.app.get('/')
		assert b'Browse :: Nyaa' in rv.data
		assert b'Guest' in rv.data

	def test_upload_url(self):
		rv = self.app.get('/upload')
		assert b'Upload Torrent' in rv.data
		assert b'You are not logged in, and are uploading anonymously.' in rv.data

	def test_rules_url(self):
		rv = self.app.get('/rules')
		assert b'Site Rules' in rv.data

	def test_help_url(self):
		rv = self.app.get('/help')
		assert b'Using the Site' in rv.data

	def test_rss_url(self):
		rv = self.app.get('/?page=rss')
		assert b'/xmlns/nyaa' in rv.data

	def test_login_url(self):
		rv = self.app.get('/login')
		assert b'Username or email address' in rv.data

	def test_registry(self):
		rv = self.app.get('/register')
		assert b'Username' in rv.data
		assert b'Password' in rv.data


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