test_mongo_service

python 1 week ago
import unittest from unittest.mock import patch, MagicMock from mongoservice import MongoService from mongomock import MongoClient as MockMongoClient class TestMongoService(unittest.TestCase): def setUp(self): self.config = { "mongo-db": { "EVENTHUB": { "MONGO_URL": "mock://localhost:27017", "MONGO_DB": "test_db" } } } self.collection_name = "test_collection" self.sample_data = {"_id": 1, "name": "test"} @patch("mongomock.MongoClient", MockMongoClient) def test_init_regular_mode(self): service = MongoService(self.config, mode="regular") self.assertEqual(service.db.name, "test_db") self.assertEqual(service.db.client.address, ("localhost", 27017)) @patch("mongomock.MongoClient", MockMongoClient) def test_init_mock_mode(self): service = MongoService(self.config, mode="mock") self.assertIsInstance(service.client, MockMongoClient) @patch("mongomock.MongoClient", MockMongoClient) def test_insert_all_query(self): service = MongoService(self.config, mode="mock") data = [{"_id": i, "value": i} for i in range(5)] service.insert_all_query(self.collection_name, data) collection = service.client["test_db"][self.collection_name] self.assertEqual(collection.count_documents({}), 5) @patch("mongomock.MongoClient", MockMongoClient) def test_insert_one_query(self): service = MongoService(self.config, mode="mock") service.insert_one_query(self.collection_name, self.sample_data) collection = service.client["test_db"][self.collection_name] self.assertEqual(collection.count_documents({}), 1) self.assertEqual(collection.find_one({"_id": 1})["name"], "test") @patch("mongomock.MongoClient", MockMongoClient) def test_update_one_query(self): service = MongoService(self.config, mode="mock") collection = service.client["test_db"][self.collection_name] collection.insert_one(self.sample_data) service.update_one_query(self.collection_name, {"_id": 1}, {"$set": {"name": "updated"}}) updated_doc = collection.find_one({"_id": 1}) self.assertEqual(updated_doc["name"], "updated") @patch("mongomock.MongoClient", MockMongoClient) def test_update_many_query(self): service = MongoService(self.config, mode="mock") data = [{"_id": i, "value": i} for i in range(5)] collection = service.client["test_db"][self.collection_name] collection.insert_many(data) service.update_many_query(self.collection_name, {"value": {"$gte": 2}}, {"$set": {"status": "updated"}}) updated_count = collection.count_documents({"status": "updated"}) self.assertEqual(updated_count, 3) @patch("mongomock.MongoClient", MockMongoClient) def test_upsert_update_one(self): service = MongoService(self.config, mode="mock") # Record doesn't exist initially service.update_one_query(self.collection_name, {"_id": 1}, {"$set": {"name": "upserted"}}, upsert=True) collection = service.client["test_db"][self.collection_name] self.assertEqual(collection.count_documents({}), 1) self.assertEqual(collection.find_one({"_id": 1})["name"], "upserted") @patch("mongomock.MongoClient", MockMongoClient) def test_upsert_update_many(self): service = MongoService(self.config, mode="mock") # Record doesn't exist initially service.update_many_query(self.collection_name, {"_id": {"$gte": 1}}, {"$set": {"name": "upserted"}}, upsert=True) collection = service.client["test_db"][self.collection_name] self.assertGreaterEqual(collection.count_documents({}), 1) if __name__ == "__main__": unittest.main()
28
Posted By
Python Script to create AWS beanstalk
#!/usr/bin/python
  
import boto
python aws beanstalk
sandeep sandeep
List all files and folders using python os mo
import os

def list_files_folders(path):
python python-os
kishore_kumar
Get current environment variables in python
import os
env = os.environ

python python-os
kishore_kumar
Get os details using python os
import os
print os.uname()
# Don't use os.system('uname -a'), its j
python python-os
kishore_kumar
Get stats ( lines, words, char count ) of fil
def file_stats(path):
    f = open(path, 'r')
    lines = f.readlines()
python
kishore_kumar
Use map function in python
def get_double(num):
    return num * 2

python
kishore_kumar
Python sample codes for beginners
print "Welcome to python"
python
gaya38 gaya38
Python program for even number checking
a=input("Enter a value:")
if (a%2==0):
    print "The given number is even numb
python
gaya38 gaya38
Python program for prime number check
a=input("Enter a value:")
k=0
b=(a/2)+1
python
gaya38 gaya38
Pass command line arguments in python
import sys
x=len(sys.argv)
a=[]
python
gaya38 gaya38
Python program for the largest number in an a
a = [1,43,98,5]#Dummy data
for l in range(len(a)-1):
        if (a[l]>a[l+1]):
python
gaya38 gaya38
print list of even numbers within a range
n=100
a=[10,20,30,40,50]
b=[60,70,80,90]
python
gaya38 gaya38
generate fibonacci series in python
n=input("Enter the constraint to print n
m=input("Enter the maximum value to prin
a=0
python
gaya38 gaya38
Generate Random number within the range in py
import random
print random.uniform(10,500)
python
gaya38 gaya38
Shuffle list elements in python
import random;
z = [1,90,4,2]
z = random.shuffle(z)
python
gaya38 gaya38
use python requests to get contents of url (
import requests

req = requests.get("https://httpbin.org/
python python-requests
kishore_kumar
how to iterate or get values in python dictio
sample_dict = { "number": 1, "fruits": [

for key in sample_dict:
python
kishore_kumar
create matrix and multiply using numpy in pyt
import numpy as np

matrix = [[1,2,3], [4,5,6], [7,8,9]]
python numpy
kishore_kumar
generate random numbers matrix with numpy pyt
import numpy as np

random_arr = np.random.randint(1,50,9)
python numpy
kishore_kumar
Find min , max and mean for numpy arrays
import numpy as np

random_arr = np.random.randint(1,50,9)
python numpy
kishore_kumar