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 com.google.common.collect.Lists;
  13 import com.google.common.collect.Sets;
  14 import com.google.common.util.concurrent.ListenableFuture;
  15 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
  16
  17 import java.io.IOException;
  18 import java.util.List;
  19 import java.util.Map;
  20 import java.util.concurrent.ExecutionException;
  21 import java.util.concurrent.ExecutorService;
  22 import java.util.concurrent.Executors;
  23
  24 import junit.framework.Assert;
  25
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.opendaylight.ovsdb.lib.message.MonitorRequest;
  29 import org.opendaylight.ovsdb.lib.message.MonitorRequestBuilder;
  30 import org.opendaylight.ovsdb.lib.message.MonitorSelect;
  31 import org.opendaylight.ovsdb.lib.message.OvsdbRPC;
  32 import org.opendaylight.ovsdb.lib.message.TableUpdates;
  33 import org.opendaylight.ovsdb.lib.message.UpdateNotification;
  34 import org.opendaylight.ovsdb.lib.notation.Column;
  35 import org.opendaylight.ovsdb.lib.notation.Row;
  36 import org.opendaylight.ovsdb.lib.operations.OperationResult;
  37 import org.opendaylight.ovsdb.lib.schema.ColumnSchema;
  38 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
  39 import org.opendaylight.ovsdb.lib.schema.GenericTableSchema;
  40 import org.opendaylight.ovsdb.lib.schema.TableSchema;
  41 import org.slf4j.Logger;
  42 import org.slf4j.LoggerFactory;
  43
  44 import java.util.Set;
  45
  46
  47 public class OvsDBClientTestIT extends OvsdbTestBase {
  48     Logger logger = LoggerFactory.getLogger(OvsDBClientTestIT.class);
  49
  50     OvsDBClientImpl ovs;
  51
  52     @Test
  53     public void testTransact() throws IOException, InterruptedException, ExecutionException {
  54
  55         ListenableFuture<DatabaseSchema> schema = ovs.getSchema(OPEN_VSWITCH_SCHEMA, true);
  56         TableSchema<GenericTableSchema> bridge = schema.get().table("Bridge", GenericTableSchema.class);
  57
  58         for (Map.Entry<String, ColumnSchema> names : bridge.getColumnSchemas().entrySet()) {
  59             System.out.println("names = " + names.getKey());
  60             System.out.println("names.getValue().getType() = " + names.getValue().getType().getBaseType());
  61         }
  62
  63         ColumnSchema<GenericTableSchema, String> name = bridge.column("name", String.class);
  64         ColumnSchema<GenericTableSchema, String> fail_mode = bridge.column("fail_mode", String.class);
  65         ColumnSchema<GenericTableSchema, Set<Integer>> flood_vlans = bridge.multiValuedColumn("flood_vlans", Integer.class);
  66
  67         ListenableFuture<List<OperationResult>> results = ovs.transactBuilder()
  68                 .add(op.insert(bridge)
  69                         .value(name, "br-test")
  70                         .value(flood_vlans, Sets.newHashSet(100, 101, 4001)))
  71                 .add(op.comment("Inserting Bridge br-int"))
  72                 .add(op.update(bridge)
  73                         .set(fail_mode, "secure")
  74                         .where(name.opEqual("br-int"))
  75                         .build())
  76                 .add(op.select(bridge)
  77                         .column(name)
  78                         .where(name.opEqual("br-int"))
  79                         .build())
  80                 .add(op.commit(true))
  81                 .execute();
  82
  83         List<OperationResult> operationResults = results.get();
  84         Assert.assertFalse(operationResults.isEmpty());
  85         System.out.println("Insert & Update operation results = " + operationResults);
  86
  87         results = ovs.transactBuilder()
  88                 .add(op.delete(bridge)
  89                         .where(name.opEqual("br-int"))
  90                         .build())
  91                 .add(op.commit(true))
  92                 .execute();
  93
  94         operationResults = results.get();
  95         Assert.assertFalse(operationResults.isEmpty());
  96         System.out.println("Delete operation results = " + operationResults);
  97
  98         /*
  99          * Adding a separate Abort operation in a transaction. Lets not mix this with other
 100          * valid transactions as above.
 101          */
 102         results = ovs.transactBuilder()
 103                 .add(op.delete(bridge)
 104                         .where(name.opEqual("br-int"))
 105                         .build())
 106                 .add(op.abort())
 107                 .execute();
 108
 109         operationResults = results.get();
 110         Assert.assertFalse(operationResults.isEmpty());
 111         /* Testing for Abort Error */
 112         Assert.assertFalse(operationResults.get(1).getError() == null);
 113         System.out.println("Abort operation results = " + operationResults);
 114
 115         /*
 116          * Adding a separate Abort operation in a transaction. Lets not mix this with other
 117          * valid transactions as above.
 118          */
 119         results = ovs.transactBuilder()
 120                 .add(op.delete(bridge)
 121                         .where(name.opEqual("br-int"))
 122                         .build())
 123                 .add(op.assertion("Assert12345")) // Failing intentionally
 124                 .execute();
 125
 126         operationResults = results.get();
 127         Assert.assertFalse(operationResults.isEmpty());
 128         /* Testing for an Assertion Error */
 129         Assert.assertFalse(operationResults.get(1).getError() == null);
 130         System.out.println("Assert operation results = " + operationResults);
 131
 132     }
 133
 134     @Test
 135     public void testMonitorRequest() throws ExecutionException, InterruptedException {
 136
 137         DatabaseSchema dbSchema = ovs.getSchema(OPEN_VSWITCH_SCHEMA, true).get();
 138         GenericTableSchema bridge = dbSchema.table("Bridge", GenericTableSchema.class);
 139
 140         List<MonitorRequest<GenericTableSchema>> monitorRequests = Lists.newArrayList();
 141         ColumnSchema<GenericTableSchema, Set<Integer>> flood_vlans = bridge.multiValuedColumn("flood_vlans", Integer.class);
 142
 143         monitorRequests.add(
 144                 MonitorRequestBuilder.builder(bridge)
 145                         .addColumn(bridge.column("name"))
 146                         .addColumn(bridge.column("fail_mode", String.class))
 147                         .addColumn(flood_vlans)
 148                         .with(new MonitorSelect(true, true, true, true))
 149                         .build());
 150
 151         final List<Object> results = Lists.newArrayList();
 152
 153         MonitorHandle monitor = ovs.monitor(dbSchema, monitorRequests, new MonitorCallBack() {
 154             @Override
 155             public void update(TableUpdates result) {
 156                 results.add(result);
 157                 System.out.println("result = " + result);
 158             }
 159
 160             @Override
 161             public void exception(Throwable t) {
 162                 results.add(t);
 163                 System.out.println("t = " + t);
 164             }
 165         });
 166
 167         for (int i = 0; i < 5 ; i++) { //wait 5 seconds to get a result
 168             System.out.println("waiting");
 169             Thread.sleep(1000);
 170         }
 171
 172         Assert.assertTrue(!results.isEmpty());
 173         Object result = results.get(0);
 174         Assert.assertTrue(result instanceof TableUpdates);
 175         TableUpdates updates = (TableUpdates) result;
 176         org.opendaylight.ovsdb.lib.message.TableUpdate<GenericTableSchema> update = updates.getUpdate(bridge);
 177         Row<GenericTableSchema> aNew = update.getNew();
 178         for (Column<GenericTableSchema, ?> column: aNew.getColumns()) {
 179             if (column.getSchema().equals(flood_vlans)) {
 180                 Set<Integer> data = column.getData(flood_vlans);
 181                 Assert.assertTrue(!data.isEmpty());
 182             }
 183         }
 184     }
 185
 186     public void testGetDBs() throws ExecutionException, InterruptedException {
 187         ListenableFuture<List<String>> databases = ovs.getDatabases();
 188         List<String> dbNames = databases.get();
 189         Assert.assertNotNull(dbNames);
 190         boolean hasOpenVswitchSchema = false;
 191         for(String dbName : dbNames) {
 192            if (dbName.equals(OPEN_VSWITCH_SCHEMA)) {
 193                 hasOpenVswitchSchema = true;
 194                 break;
 195            }
 196         }
 197         Assert.assertTrue(OPEN_VSWITCH_SCHEMA+" schema is not supported by the switch", hasOpenVswitchSchema);
 198     }
 199
 200     @Before
 201     public  void initalize() throws IOException, ExecutionException, InterruptedException {
 202         if (ovs != null) {
 203             return;
 204         }
 205         OvsdbRPC rpc = getTestConnection();
 206         if (rpc == null) {
 207             System.out.println("Unable to Establish Test Connection");
 208         }
 209         ExecutorService executorService = Executors.newFixedThreadPool(3);
 210         ovs = new OvsDBClientImpl(rpc, executorService);
 211         testGetDBs();
 212     }
 213
 214
 215     @Override
 216     public void update(Object node, UpdateNotification upadateNotification) {
 217         // TODO Auto-generated method stub
 218
 219     }
 220
 221     @Override
 222     public void locked(Object node, List<String> ids) {
 223         // TODO Auto-generated method stub
 224
 225     }
 226     @Override
 227     public void stolen(Object node, List<String> ids) {
 228         // TODO Auto-generated method stub
 229
 230     }
 231 }