예제 #1
0
 def setup(self):
     dummy_db = None
     mapper = None
     self.importer = TupleBasedImporter(None,
                                        dummy_db,
                                        mapper,
                                        None,
                                        dry_run=False)
예제 #2
0
class TestTupleBasedImporter(object):
    def setup(self):
        dummy_db = None
        mapper = None
        self.importer = TupleBasedImporter(None,
                                           dummy_db,
                                           mapper,
                                           None,
                                           dry_run=False)

    def test_insert(self):
        mappings = [
            (('highway', 'secondary'), [defaultmapping.mainroads]),
            (('railway', 'tram'), [defaultmapping.railways]),
            (('landusage', 'grass'), [defaultmapping.landusages]),
        ]
        assert self.importer.insert(
            mappings, 1234, [(0, 0), (1, 0), (1, 1), (0, 0)], {
                'highway': 'secondary',
                'railway': 'tram',
                'oneway': '1',
                'name': 'roundabout',
            })

        mainroads_item = self.importer.db_queue.get()
        eq_(mainroads_item[0], defaultmapping.mainroads)
        eq_(mainroads_item[1], 1234)
        eq_(mainroads_item[3], ['roundabout', 'secondary', 0, 0, 1, None, 5])

        railways_item = self.importer.db_queue.get()
        eq_(railways_item[0], defaultmapping.railways)
        eq_(railways_item[1], 1234)
        eq_(railways_item[3], ['roundabout', 'tram', 0, 0, 7])

        landusages_item = self.importer.db_queue.get()
        eq_(landusages_item[0], defaultmapping.landusages)
        eq_(landusages_item[1], 1234)
        eq_(landusages_item[3], ['roundabout', 'grass', 6195822904.182782, 27])