Пример #1
0
 def test002_list_error(self):
     sample = Exception("test")
     db_mock = MagicMock()
     db_mock.query.side_effect = sample
     with patch("frontend.api.controllers.scans.process_error") as mock:
         api_scans.list(db_mock)
     self.assertTrue(db_mock.query.called)
     self.assertEqual(db_mock.query.call_args, ((Scan,),))
     self.assertTrue(mock.called)
     self.assertEqual(mock.call_args, ((sample,),))
Пример #2
0
 def test004_list_custom_request(self):
     db_mock = MagicMock()
     scan_schema_mock = MagicMock()
     offset, limit = randint(1, 100), randint(1, 100)
     expected = {"total": db_mock.query().count(),
                 "offset": offset,
                 "limit": limit,
                 "data": scan_schema_mock.dump().data}
     api_scans.request.query.offset = offset
     api_scans.request.query.limit = limit
     scan_schema = "frontend.api.controllers.scans.scan_schema"
     with patch(scan_schema, scan_schema_mock):
         result = api_scans.list(db_mock)
     self.assertEqual(result, expected)
     self.assertTrue(db_mock.query().count.called)
Пример #3
0
 def test003_list_default(self):
     db_mock = MagicMock()
     default_offset, default_limit = 0, 5
     scan_schema_mock = MagicMock()
     expected = {"total": len(db_mock.query().limit().offset().all()),
                 "offset": default_offset,
                 "limit": default_limit,
                 "data": scan_schema_mock.dump().data}
     scan_schema = "frontend.api.controllers.scans.scan_schema"
     with patch(scan_schema, scan_schema_mock):
         result = api_scans.list(db_mock)
     self.assertEqual(result, expected)
     self.assertEqual(api_scans.response.content_type,
                      "application/json; charset=UTF-8")
     self.assertEqual(db_mock.query.call_args, ((Scan,),))
     self.assertEqual(db_mock.query().limit.call_args, ((default_limit,),))
     self.assertEqual(db_mock.query().limit().offset.call_args,
                      ((default_offset,),))
     self.assertFalse(db_mock.query().count.called)