Python Link.drop_collection示例

示例#1

文件： test_resources.py 项目： tcufer/nyc_traffic_speed_api

    def test_link(self):
        # get all links
        rv, json = self.client.get('/trafficLink/')
        self.assertTrue(rv.status_code == 200)
        assert isinstance(json.pop('trafficLinkList'), list)


        #get a link by linkId
        rv, json = self.client.get('/trafficLink/4616337')
        self.assertTrue(rv.status_code == 200)
        self.assertTrue('borough' and
                        'encodedPolyLine' and
                        'encodedPolyLineLvls' and
                        'linkId' and
                        'linkName' and
                        'linkPoints' and
                        'owner' in json['Link'][0])

        #invalid id
        rv, json = self.client.get('/trafficLink/123456789')
        self.assertTrue(rv.status_code == 404)

        #no link data
        from common.models import Link
        Link.drop_collection()
        rv, json = self.client.get('/trafficLink/')
        self.assertTrue(rv.status_code == 404)

示例#2

文件： test_data_ingestion.py 项目： tcufer/nyc_traffic_speed_api

 def setUp(self):
     app.config['TESTING'] = True
     self.app = app
     Sensor.drop_collection()
     Link.drop_collection()