TestGraphViewMXGraph.java

/*
 * Follow the  Bitcoin
 * Copyright (C) 2014  Danno Ferrin
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * version 2 as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

package com.shemnon.btc.view;

import com.shemnon.btc.blockchaininfo.OfflineData;
import com.shemnon.btc.blockchaininfo.TXInfo;
import com.shemnon.btc.model.ITx;
import com.sun.javafx.application.PlatformImpl;
import javafx.application.Platform;
import javafx.scene.Scene;
import javafx.scene.layout.Pane;
import javafx.stage.Stage;
import org.junit.BeforeClass;
import org.junit.Test;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;

/**
 * 
 * Created by shemnon on 3 Mar 2014.
 */
public class TestGraphViewMXGraph {

    @BeforeClass
    public static void loadOfflineData() {
        OfflineData.loadOfflineData();
    }

    @Test
    public void testSimpleGraph() throws InterruptedException {
        PlatformImpl.startup(() -> {});
        CountDownLatch latch = new CountDownLatch(1);
        Platform.runLater(() ->  {
            GraphViewMXGraph gv = new GraphViewMXGraph((event, tx) -> {});
    
            TXInfo tx42 = TXInfo.query("3a1b9e330d32fef1ee42f8e86420d2be978bbe0dc5862f17da9027cf9e11f8c4");
            gv.addTransaction(tx42);
            List<ITx> moreTXes = new ArrayList<>();
            tx42.getOutputs().forEach(c -> {
                moreTXes.add(c.getTargetTX());
                gv.addCoin(c);
            });
            for (int i = 0; i < 5; i++) {
                List<ITx> thisTX = new ArrayList<>(moreTXes);
                moreTXes.clear();
                thisTX.forEach((t) -> {
                    if (t != null && t.getOutputs() != null) {
                        t.getOutputs().forEach(c -> {
                            moreTXes.add(c.getTargetTX());
                            gv.addCoin(c);
                        });
                        t.getInputs().forEach(gv::addCoin);
                    }
                });
            }
            tx42.getInputs().forEach(gv::addCoin);
            
            gv.layout();
    
            Pane pane = gv.getGraphPane();
            gv.rebuildGraph(false);
            
            pane.setScaleX(0.5);
            pane.setScaleY(0.5);

            ZoomPane zp = new ZoomPane(pane);            
            Scene scene = new Scene(zp);
            scene.getStylesheets().add("com/shemnon/btc/view/btc.css");
            
            Stage stage = new Stage();
            stage.setScene(scene);
            stage.setOnCloseRequest(e -> {stage.close(); latch.countDown();});
            
            stage.show();
        });
        latch.await();
    }
}