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()