test_api API documentation

def setUp(self): View Source

 9    def setUp(self):
10        self.app = api
11        self.app.config['TESTING'] = True
12        setup_mongo_client(self.app)  # Set up the mongo client after changing the TESTING flag
13        self.client = self.app.test_client()
14        print("Using MongoDB client:", type(self.app.mongo_client))

Hook method for setting up the test fixture before exercising it.

def test_get_events(self): View Source

17    def test_get_events(self):
18        # Create a mock collection
19        db = self.app.mongo_client['test']
20        collection = db['events']
21
22        # Replace the collection's find method with a Mock object
23        mock_find = Mock()
24        collection.find = mock_find
25        mock_find.return_value = [
26            {"_id": "Event 1"},
27            {"_id": "Event 2"},
28        ]
29
30        response = self.client.get('/events')
31        self.assertEqual(response.status_code, 200)

@patch('pymongo.collection.Collection.update_one')

def test_register_success(self, mock_update_one): View Source

34    @patch("pymongo.collection.Collection.update_one")
35    def test_register_success(self, mock_update_one):
36        app_client = api.test_client()  # Create a test client for this test case
37
38        # Mock the update_one method to simulate a successful registration
39        mock_update_one.return_value = Mock(upserted_id=123)
40
41        test_data = {
42            'email': 'test_user',
43            'password': 'test_password',
44            'firstName': 'Test',
45            'lastName': 'User'
46        }
47
48        response = app_client.post('/register', json=test_data)
49
50        self.assertEqual(response.status_code, 200)
51
52        response_data = response.get_json()
53        self.assertEqual(response_data['msg'], "register successful")

def test_unauthorized_get_user_registered_events(self): View Source

55    def test_unauthorized_get_user_registered_events(self):
56        # Mock the database query result
57        app_client = api.test_client()
58
59        db = app_client.application.mongo_client['test']  # Access the app's Mongo client
60        collection = db['user']
61        mock_find = Mock()
62
63        collection.find = mock_find
64        mock_find.return_value = [
65            {"eventTitle": "Yoga"},
66            {"eventTitle": "Swimming"}
67        ]
68
69        with patch("flask_jwt_extended.get_jwt_identity", return_value="test_user"):
70            response = app_client.get('/usersEvents')
71
72        self.assertEqual(response.status_code, 401)

@patch('base.request')

@patch('base.jwt_required')

@patch('base.mongo')

def test_unauthorized_enrolled_true(self, mock_mongo, mock_jwt_required, mock_request): View Source

75    @patch('base.request')
76    @patch('base.jwt_required')
77    @patch('base.mongo')
78    def test_unauthorized_enrolled_true(self, mock_mongo, mock_jwt_required, mock_request):
79        app_client = api.test_client()
80        # Mock request.json() method to return test data
81        mock_request.json.return_value = {'eventTitle': 'Event Name'}
82        
83        # Mock get_jwt_identity() to return a test user identity
84        mock_jwt_required.return_value = lambda f: f
85
86        # Mock the find_one method to return an enrollment
87        mock_mongo.user.find_one.return_value = {'email': 'test@example.com', 'eventTitle': 'Event Name'}
88
89        response = app_client.post('/is-enrolled')
90        data = json.loads(response.get_data(as_text=True))
91
92        self.assertEqual(response.status_code, 401)