tests.py

import os
import app
import unittest
import tempfile

""" Just modified the Flask example unit test, need to write orderbook tests."""
class ExchangeTestCase(unittest.TestCase):
    def setUp(self):
        self.db_fd, app.app.config['DATABASE'] = tempfile.mkstemp()
        app.app.config['TESTING'] = True
        self.app = app.app.test_client()
        app.init_db()

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

    def login(self, email, password):
    	return self.app.post('/login', data=dict(
        email=email,
        password=password
        ), follow_redirects=True)

    def logout(self):
    	return self.app.get('/logout', follow_redirects=True)


    def test_login_logout(self):
	    rv = self.login('sam@b.net', 'shit')
	    assert 'Logged in!' in rv.data
	    rv = self.logout()
	    assert 'Successfully logged out!' in rv.data
	    rv = self.login('adminx', 'default')
	    assert 'Please check your email and username.' in rv.data
	    rv = self.login('admin', 'defaultx')
	    assert 'Please check your email and username.' in rv.data

def logout(self):
    return self.app.get('/logout', follow_redirects=True)

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