def test_loads_exported_array_data(self, open_mock): documents = [ { '_id': 1, 'msg': 'my test string' }, { '_id': 2, 'message': 'my other string' }, { '_id': 3, 'foo': 'bar', 'greeting': 'hi there' }, ] file_contents = dumps(documents) self.mock_open(open_mock, file_contents) with self.real_app.app_context(): db = get_db() load_data_from_mongoexport('myresid.', 'my/file/location', 'mycoll') open_mock.assert_called_with('my/file/location') collection_contents = list(db['myresid.mycoll'].find()) self.assertItemsEqual(collection_contents, documents) db['myresid.mycoll'].drop()
def test_loads_exported_data(self, open_mock): documents = [ {'_id': 1, 'msg': 'my test string'}, {'_id': 2, 'message': 'my other string'}, {'_id': 3, 'foo': 'bar', 'greeting': 'hi there'}, ] file_contents = '\n'.join([dumps(doc) for doc in documents]) self.mock_open(open_mock, file_contents) with self.real_app.app_context(): db = get_db() # Test normally (keeping the _id) load_data_from_mongoexport('myresid.', 'my/file/location', 'mycoll') open_mock.assert_called_with('my/file/location') collection_contents = list(db['myresid.mycoll'].find()) self.assertItemsEqual(collection_contents, documents) db['myresid.mycoll'].drop() # Test removing the _id load_data_from_mongoexport('myresid.', 'my/file/location', 'mycoll', True) collection_contents = list(db['myresid.mycoll'].find()) for doc in collection_contents: # Should not be any of the given _id's self.assertNotIn(doc['_id'], (1, 2, 3)) db['myresid.mycoll'].drop()
def test_loads_exported_array_data(self, open_mock): documents = [ {'_id': 1, 'msg': 'my test string'}, {'_id': 2, 'message': 'my other string'}, {'_id': 3, 'foo': 'bar', 'greeting': 'hi there'}, ] file_contents = dumps(documents) self.mock_open(open_mock, file_contents) with self.real_app.app_context(): db = get_db() load_data_from_mongoexport('myresid.', 'my/file/location', 'mycoll') open_mock.assert_called_with('my/file/location') collection_contents = list(db['myresid.mycoll'].find()) self.assertItemsEqual(collection_contents, documents) db['myresid.mycoll'].drop()
def test_loads_exported_data(self, open_mock): documents = [ { '_id': 1, 'msg': 'my test string' }, { '_id': 2, 'message': 'my other string' }, { '_id': 3, 'foo': 'bar', 'greeting': 'hi there' }, ] file_contents = '\n'.join([dumps(doc) for doc in documents]) self.mock_open(open_mock, file_contents) with self.real_app.app_context(): db = get_db() # Test normally (keeping the _id) load_data_from_mongoexport('myresid.', 'my/file/location', 'mycoll') open_mock.assert_called_with('my/file/location') collection_contents = list(db['myresid.mycoll'].find()) self.assertItemsEqual(collection_contents, documents) db['myresid.mycoll'].drop() # Test removing the _id load_data_from_mongoexport('myresid.', 'my/file/location', 'mycoll', True) collection_contents = list(db['myresid.mycoll'].find()) for doc in collection_contents: # Should not be any of the given _id's self.assertNotIn(doc['_id'], (1, 2, 3)) db['myresid.mycoll'].drop()