Examples with BundleBuildingServiceImpl org.onebusaway.admin.service.bundle.impl.BundleBuildingServiceImpl used on opensource projects

Example 1 with BundleBuildingServiceImpl

use of org.onebusaway.admin.service.bundle.impl.BundleBuildingServiceImpl in project onebusaway-application-modules by camsys.

the class BundleBuildingServiceImplTest method setup.

public void setup() {
    _service = new BundleBuildingServiceImpl() {

        @Override
        public String getDefaultAgencyId() {
            return null;
        }
    };
    _service.setDebug(true);
    FileService fileService;
    fileService = new S3FileServiceImpl() {

        @Override
        public void setup() {
        }

        @Override
        public boolean bundleDirectoryExists(String filename) {
            return !"noSuchDirectory".equals(filename);
        }

        @Override
        public boolean createBundleDirectory(String filename) {
            return true;
        }

        @Override
        public List<String[]> listBundleDirectories(int maxResults) {
            ArrayList<String[]> list = new ArrayList<String[]>();
            String[] columns0 = { "2012April", "", "" + System.currentTimeMillis() };
            list.add(columns0);
            String[] columns1 = { "2012Jan", "", "" + System.currentTimeMillis() };
            list.add(columns1);
            String[] columns2 = { "2011April", "", "" + System.currentTimeMillis() };
            list.add(columns2);
            String[] columns3 = { "2011Jan", "", "" + System.currentTimeMillis() };
            list.add(columns3);
            String[] columns4 = { "2010April", "", "" + System.currentTimeMillis() };
            list.add(columns4);
            String[] columns5 = { "2010Jan", "", "" + System.currentTimeMillis() };
            list.add(columns5);
            return list;
        }

        @Override
        public List<String> list(String directory, int maxResults) {
            _log.debug("list called with " + directory);
            ArrayList<String> list = new ArrayList<String>();
            if (directory.equals("test/gtfs_latest")) {
                list.add("gtfs-m34.zip");
            } else if (directory.equals("test/aux_latest")) {
                if ("true".equals(_service.getAuxConfig())) {
                    list.add(CT_GIS_ZIP);
                    list.add(CT_SCHEDULE_ZIP);
                } else {
                    list.add("stif-m34.zip");
                }
            } else if (directory.equals("test/config")) {
            // do nothing
            } else {
                throw new RuntimeException("file not found for dir=" + directory);
            }
            return list;
        }

        @Override
        public String get(String key, String tmpDir) {
            _log.debug("get called with " + key);
            InputStream source = null;
            if (key.equals("gtfs-m34.zip")) {
                source = this.getClass().getResourceAsStream("gtfs-m34.zip");
            } else if (key.equals("stif-m34.zip")) {
                source = this.getClass().getResourceAsStream("stif-m34.zip");
            } else if (key.equals(CT_GIS_ZIP)) {
                source = this.getClass().getResourceAsStream(CT_GIS_ZIP);
            } else if (key.equals(CT_SCHEDULE_ZIP)) {
                source = this.getClass().getResourceAsStream(CT_SCHEDULE_ZIP);
            } else {
                throw new RuntimeException("unmatched key=" + key + " for tmpDir=" + tmpDir);
            }
            String filename = tmpDir + File.separator + key;
            _log.info("copying " + source + " to " + filename);
            new NYCFileUtils().copy(source, filename);
            return filename;
        }

        @Override
        public String put(String key, String file) {
            // do nothing
            return null;
        }
    };
    fileService.setBucketName("obanyc-bundle-data");
    fileService.setGtfsPath("gtfs_latest");
    fileService.setAuxPath("aux_latest");
    fileService.setBuildPath("builds");
    fileService.setConfigPath("config");
    fileService.setup();
    // uncomment for s3
    // fileService = new FileServiceImpl();
    // fileService.setBucketName("obanyc-bundle-data");
    // fileService.setGtfsPath("gtfs_latest");
    // fileService.setStifPath("stif_latest");
    // fileService.setBuildPath("builds");
    // fileService.setConfigPath("config");
    // fileService.setup();
    _service.setFileService(fileService);
    _service.setup();
}