library/src/test/java/org/opendaylight/ovsdb/lib/OvsDBClientTestIT.java

   1 /*
   2  * Copyright (C) 2014 EBay Software Foundation
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  *
   8  * Authors : Ashwin Raveendran
   9  */
  10 package org.opendaylight.ovsdb.lib;
  11
  12 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
  13
  14 import java.io.IOException;
  15 import java.util.List;
  16 import java.util.Map;
  17 import java.util.concurrent.ExecutionException;
  18 import java.util.concurrent.ExecutorService;
  19 import java.util.concurrent.Executors;
  20
  21 import junit.framework.Assert;
  22
  23 import org.junit.Before;
  24 import org.junit.Test;
  25 import org.opendaylight.ovsdb.lib.message.MonitorRequest;
  26 import org.opendaylight.ovsdb.lib.message.MonitorRequestBuilder;
  27 import org.opendaylight.ovsdb.lib.message.MonitorSelect;
  28 import org.opendaylight.ovsdb.lib.message.OvsdbRPC;
  29 import org.opendaylight.ovsdb.lib.message.TableUpdate;
  30 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  31 import org.opendaylight.ovsdb.lib.message.UpdateNotification;
  32 import org.opendaylight.ovsdb.lib.operations.OperationResult;
  33 import org.opendaylight.ovsdb.lib.schema.ColumnSchema;
  34 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  35 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  36 import org.opendaylight.ovsdb.lib.schema.TableSchema;
  37 import org.slf4j.Logger;
  38 import org.slf4j.LoggerFactory;
  39
  40 import com.google.common.collect.Lists;
  41 import com.google.common.util.concurrent.ListenableFuture;
  42
  43
  44 public class OvsDBClientTestIT extends OvsdbTestBase {
  45     Logger logger = LoggerFactory.getLogger(OvsDBClientTestIT.class);
  46
  47     OvsDBClientImpl ovs;
  48
  49     @Test
  50     public void testTransact() throws IOException, InterruptedException, ExecutionException {
  51
  52         ListenableFuture<DatabaseSchema> schema = ovs.getSchema(OvsDBClient.OPEN_VSWITCH_SCHEMA, true);
  53         TableSchema<GenericTableSchema> bridge = schema.get().table("Bridge", GenericTableSchema.class);
  54
  55         for (Map.Entry<String, ColumnSchema> names : bridge.getColumnSchemas().entrySet()) {
  56             System.out.println("names = " + names.getKey());
  57             System.out.println("names.getValue().getType() = " + names.getValue().getType().getBaseType());
  58         }
  59
  60         ColumnSchema<GenericTableSchema, String> name = bridge.column("name", String.class);
  61         ColumnSchema<GenericTableSchema, String> fail_mode = bridge.column("fail_mode", String.class);
  62
  63         ListenableFuture<List<OperationResult>> results = ovs.transactBuilder()
  64                 .add(op.insert(bridge).value(name, "br-int"))
  65                 .add(op.update(bridge)
  66                         .set(fail_mode, "secure")
  67                         .where(name.opEqual("br-int"))
  68                         //.and(name.opEqual("br-int"))
  69                         .operation())
  70                 .add(op.commit(true))
  71                 .execute();
  72
  73         List<OperationResult> operationResults = results.get();
  74         Assert.assertFalse(operationResults.isEmpty());
  75         System.out.println("Insert & Update operation results = " + operationResults);
  76
  77         results = ovs.transactBuilder()
  78                 .add(op.delete(bridge)
  79                         .where(name.opEqual("br-int"))
  80                         .operation())
  81                 .add(op.commit(true))
  82                 .execute();
  83
  84         operationResults = results.get();
  85         Assert.assertFalse(operationResults.isEmpty());
  86         System.out.println("Delete operation results = " + operationResults);
  87
  88         /*
  89          * Adding a separate Abort operation in a transaction. Lets not mix this with other
  90          * valid transactions as above.
  91          */
  92         results = ovs.transactBuilder()
  93                 .add(op.delete(bridge)
  94                         .where(name.opEqual("br-int"))
  95                         .operation())
  96                 .add(op.abort())
  97                 .execute();
  98
  99         operationResults = results.get();
 100         Assert.assertFalse(operationResults.isEmpty());
 101         System.out.println("Abort operation results = " + operationResults);
 102
 103     }
 104
 105     @Test
 106     public void testMonitorRequest() throws ExecutionException, InterruptedException {
 107
 108         DatabaseSchema dbSchema = ovs.getSchema(OvsDBClient.OPEN_VSWITCH_SCHEMA, true).get();
 109         GenericTableSchema bridge = dbSchema.table("Bridge", GenericTableSchema.class);
 110
 111         List<MonitorRequest<GenericTableSchema>> monitorRequests = Lists.newArrayList();
 112         monitorRequests.add(
 113                 MonitorRequestBuilder.builder(bridge)
 114                         .addColumn(bridge.column("name"))
 115                         .addColumn(bridge.column("fail_mode", String.class))
 116                         .with(new MonitorSelect(true, true, true, true))
 117                         .build());
 118
 119         final List<Object> results = Lists.newArrayList();
 120
 121         MonitorHandle monitor = ovs.monitor(dbSchema, monitorRequests, new MonitorCallBack() {
 122             @Override
 123             public void update(TableUpdates result) {
 124                 results.add(result);
 125                 System.out.println("result = " + result);
 126             }
 127
 128             @Override
 129             public void exception(Throwable t) {
 130                 results.add(t);
 131                 System.out.println("t = " + t);
 132             }
 133         });
 134
 135         //for (int i = 0; i < 5 && results.isEmpty(); i++) { //wait 5 seconds to get a result
 136         for (int i = 0; i < 5 ; i++) { //wait 5 seconds to get a result
 137             System.out.println("waiting");
 138             Thread.sleep(1000);
 139         }
 140
 141         Assert.assertTrue(!results.isEmpty());
 142         Object result = results.get(0);
 143         Assert.assertTrue(result instanceof TableUpdates);
 144         TableUpdate bridgeUpdate = ((TableUpdates) result).getUpdate(bridge);
 145         Assert.assertNotNull(bridgeUpdate);
 146     }
 147
 148     @Test
 149     public void testGetDBs() throws ExecutionException, InterruptedException {
 150         ListenableFuture<List<String>> databases = ovs.getDatabases();
 151         List<String> dbNames = databases.get();
 152         Assert.assertNotNull(dbNames);
 153         Assert.assertTrue(dbNames.size() > 0);
 154     }
 155
 156     @Before
 157     public  void initalize() throws IOException {
 158         if (ovs != null) {
 159             return;
 160         }
 161         OvsdbRPC rpc = getTestConnection();
 162         if (rpc == null) {
 163             System.out.println("Unable to Establish Test Connection");
 164         }
 165         ExecutorService executorService = Executors.newFixedThreadPool(3);
 166         ovs = new OvsDBClientImpl(rpc, executorService);
 167     }
 168
 169
 170     @Override
 171     public void update(Object node, UpdateNotification upadateNotification) {
 172         // TODO Auto-generated method stub
 173
 174     }
 175
 176     @Override
 177     public void locked(Object node, List<String> ids) {
 178         // TODO Auto-generated method stub
 179
 180     }
 181     @Override
 182     public void stolen(Object node, List<String> ids) {
 183         // TODO Auto-generated method stub
 184
 185     }
 186 }