diff options
author | SerenaFeng <feng.xiaowei@zte.com.cn> | 2016-05-17 21:05:27 +0800 |
---|---|---|
committer | SerenaFeng <feng.xiaowei@zte.com.cn> | 2016-05-17 21:12:20 +0800 |
commit | d719f7428de763767b491b983ce214c329ba37b1 (patch) | |
tree | c5b4e7593d2223eed50581ec278ca20b010c076f /result_collection_api/tests/unit/test_fake_pymongo.py | |
parent | f91356cf045f03f2c318821dd087f05670abb892 (diff) |
add unittest framework for supporting unittest in testAPI
usage is shown in utils/test/result_collection_api/README.md
JIRA: FUNCTEST-251
Change-Id: I788417e296c153cc485f4a4064697bdafc394e5b
Signed-off-by: SerenaFeng <feng.xiaowei@zte.com.cn>
Diffstat (limited to 'result_collection_api/tests/unit/test_fake_pymongo.py')
-rw-r--r-- | result_collection_api/tests/unit/test_fake_pymongo.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/result_collection_api/tests/unit/test_fake_pymongo.py b/result_collection_api/tests/unit/test_fake_pymongo.py new file mode 100644 index 0000000..5ddbf28 --- /dev/null +++ b/result_collection_api/tests/unit/test_fake_pymongo.py @@ -0,0 +1,52 @@ +import unittest +from tornado.web import Application +from tornado import gen +from tornado.testing import AsyncHTTPTestCase, gen_test + +import fake_pymongo + + +class MyTest(AsyncHTTPTestCase): + def setUp(self): + super(MyTest, self).setUp() + self.db = fake_pymongo + self.io_loop.run_sync(self.fixture_setup) + + def get_app(self): + return Application() + + @gen.coroutine + def fixture_setup(self): + self.test1 = {'_id': '1', 'name': 'test1'} + self.test2 = {'name': 'test2'} + yield self.db.pod.insert({'_id': '1', 'name': 'test1'}) + yield self.db.pod.insert({'name': 'test2'}) + + @gen_test + def test_find_one(self): + user = yield self.db.pod.find_one({'name': 'test1'}) + self.assertEqual(user, self.test1) + + @gen_test + def test_find(self): + cursor = self.db.pod.find() + names = [] + while (yield cursor.fetch_next): + ob = cursor.next_object() + names.append(ob.get('name')) + self.assertItemsEqual(names, ['test1', 'test2']) + + @gen_test + def test_update(self): + yield self.db.pod.update({'_id': '1'}, {'name': 'new_test1'}) + user = yield self.db.pod.find_one({'_id': '1'}) + self.assertEqual(user.get('name', None), 'new_test1') + + @gen_test + def test_remove(self): + yield self.db.pod.remove({'_id': '1'}) + user = yield self.db.pod.find_one({'_id': '1'}) + self.assertIsNone(user) + +if __name__ == '__main__': + unittest.main() |