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