def setUp(self):
TestAuth.setUp(self)
self.user = User(self.username, self.password, self.email)
auth.g = mock.MagicMock()
with auth.database_session() as session:
session.add(self.user)
self.user = self.user.__class__(self.username, self.password,
self.email)
def setUp(self):
TestCaseWithDatabase.setUp(self)
self.username = '******'
self.password = '******'
self.email = '*****@*****.**'
self.object_to_write = User(self.username, self.password, self.email)
self.session = ContextManagedSession(bind=conf.DATABASE_ENGINE)
def setUp(self):
TestUserView.setUp(self)
self.request_method = self.client.get
self.url = '/users/%s' % self.username
self.headers['Authorization'] = 'Basic %s' % \
b64encode(('%s:%s' % (self.username, self.password)).
encode('ascii')).decode('ascii')
self.bad_user = User('foo', 'bar', '*****@*****.**')
def setUp(self):
TestCaseWithDatabase.setUp(self)
self.endpoint = tokens.Tokens()
self.token_string = str(uuid1())
self.username = '******'
self.password = '******'
self.email = '*****@*****.**'
self.user = User(self.username, self.password, self.email)
self.admin = Administrator(self.username, self.password, self.email)
def setUpClass(cls):
cls.client = app.test_client()
cls.headers = {'content-type': 'application/json'}
cls.owner_name = 'Bilbo Baggins'
cls.owner_password = '******'
cls.owner_email = 'bilbo@bag_end.com'
cls.owner = User(cls.owner_name, cls.owner_password, cls.owner_email)
cls.project_name = 'Test Project'
cls.project_description = 'Project Description with some **Markdown**'
cls.project = Project(cls.project_name, cls.project_description)
def setUp(self):
TestAuth.setUp(self)
self.user = User(self.username, self.password, self.email)
self.token_string = str(uuid1())
self.token = Token(self.token_string, owner=self.user)
with auth.database_session() as session:
session.add(self.user)
session.add(self.token)
auth.g = mock.MagicMock()
def setUp(self):
TestAuth.setUp(self)
self.user = User(self.username, self.password, self.email)
auth.g = mock.MagicMock()
with auth.database_session() as session:
session.add(self.user)
self.user = self.user.__class__(
self.username, self.password, self.email
)
def setUp(self):
self.username = '******'
self.password = '******'
self.email = '*****@*****.**'
self.user = User(self.username, self.password, self.email)
self.user.verify_password = mock.MagicMock(return_value=True)
def setUp(self):
TestUserContainer.setUp(self)
self.request_method = self.client.get
self.url = '/users'
self.user = User(self.username, self.password, self.email)
self.user.verify_password = mock.MagicMock(return_value=True)
def setUp(self):
TestCaseWithDatabase.setUp(self)
self._clear_test_user(database_session)
with database_session() as session:
session.add(User(self.username, self.password, self.email))
class TestVerifyUser(TestAuth):
def setUp(self):
TestAuth.setUp(self)
self.user = User(self.username, self.password, self.email)
auth.g = mock.MagicMock()
with auth.database_session() as session:
session.add(self.user)
self.user = self.user.__class__(self.username, self.password,
self.email)
def tearDown(self):
with auth.database_session() as session:
user = session.query(
self.user.__class__).filter_by(username=self.username).first()
if user is not None:
session.delete(user)
TestAuth.tearDown(self)
def test_auth_token_correct(self):
with mock.patch('sqlalchemy.orm.Query.first',
return_value=self.user) as mock_verify_auth:
self.assertTrue(auth.verify_password(self.username, self.password))
self.assertEqual(mock_verify_auth.call_args, mock.call())
@mock.patch('omicron_server.auth.User.verify_password', return_value=True)
def test_user_query(self, mock_verify_pwd):
self.assertTrue(auth.verify_password(self.username, self.password))
self.assertEqual(mock_verify_pwd.call_args, mock.call(self.password))
@mock.patch('omicron_server.auth.User.verify_password', return_value=True)
def test_user_query_no_user_found(self, mock_verify):
with auth.database_session() as session:
user = session.query(
self.user.__class__).filter_by(username=self.username).first()
session.delete(user)
self.assertFalse(auth.verify_password(self.username, self.password))
self.assertFalse(mock_verify.called)
@mock.patch('omicron_server.auth.User.verify_password', return_value=False)
def test_user_query_bad_password(self, mock_verify):
self.assertFalse(auth.verify_password(self.username, self.password))
self.assertTrue(mock_verify.called)
def test_user_query_token_adds_to_g(self):
self.assertTrue(auth.verify_password(self.username, self.password))
self.assertIsInstance(auth.g.user, self.user.__class__)
@mock.patch('omicron_server.auth.User.verify_auth_token',
return_value=False)
@mock.patch('sqlalchemy.orm.query.Query.first')
@mock.patch('omicron_server.auth.User.verify_password', return_value=True)
def test_user_password_verify_adds_to_g(self, mock_verify_password,
mock_query, mock_verify_token):
mock_query.return_value = self.user
self.assertTrue(auth.verify_password(self.username, self.password))
self.assertEqual(auth.g.user, self.user)
self.assertEqual(mock_query.call_args, mock.call())
self.assertEqual(mock_verify_password.call_args,
mock.call(self.password))
class TestVerifyUser(TestAuth):
def setUp(self):
TestAuth.setUp(self)
self.user = User(self.username, self.password, self.email)
auth.g = mock.MagicMock()
with auth.database_session() as session:
session.add(self.user)
self.user = self.user.__class__(
self.username, self.password, self.email
)
def tearDown(self):
with auth.database_session() as session:
user = session.query(self.user.__class__).filter_by(
username=self.username
).first()
if user is not None:
session.delete(user)
TestAuth.tearDown(self)
def test_auth_token_correct(self):
with mock.patch(
'sqlalchemy.orm.Query.first', return_value=self.user
) as mock_verify_auth:
self.assertTrue(auth.verify_password(
self.username, self.password
)
)
self.assertEqual(
mock_verify_auth.call_args,
mock.call()
)
@mock.patch('omicron_server.auth.User.verify_password', return_value=True)
def test_user_query(self, mock_verify_pwd):
self.assertTrue(auth.verify_password(
self.username, self.password
))
self.assertEqual(
mock_verify_pwd.call_args,
mock.call(self.password)
)
@mock.patch('omicron_server.auth.User.verify_password', return_value=True)
def test_user_query_no_user_found(self, mock_verify):
with auth.database_session() as session:
user = session.query(self.user.__class__).filter_by(
username=self.username
).first()
session.delete(user)
self.assertFalse(auth.verify_password(
self.username, self.password))
self.assertFalse(mock_verify.called)
@mock.patch('omicron_server.auth.User.verify_password', return_value=False)
def test_user_query_bad_password(self, mock_verify):
self.assertFalse(auth.verify_password(
self.username, self.password
))
self.assertTrue(mock_verify.called)
def test_user_query_token_adds_to_g(self):
self.assertTrue(auth.verify_password(
self.username, self.password
))
self.assertIsInstance(auth.g.user, self.user.__class__)
@mock.patch('omicron_server.auth.User.verify_auth_token',
return_value=False)
@mock.patch('sqlalchemy.orm.query.Query.first')
@mock.patch('omicron_server.auth.User.verify_password', return_value=True)
def test_user_password_verify_adds_to_g(
self, mock_verify_password, mock_query, mock_verify_token
):
mock_query.return_value = self.user
self.assertTrue(auth.verify_password(
self.username, self.password
))
self.assertEqual(auth.g.user, self.user)
self.assertEqual(mock_query.call_args, mock.call())
self.assertEqual(mock_verify_password.call_args,
mock.call(self.password))
