2 * Copyright (C) 2014 EBay Software Foundation
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
8 * Authors : Ashwin Raveendran
10 package org.opendaylight.ovsdb.lib;
12 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
14 import java.io.IOException;
15 import java.util.List;
17 import java.util.concurrent.ExecutionException;
18 import java.util.concurrent.ExecutorService;
19 import java.util.concurrent.Executors;
21 import junit.framework.Assert;
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.notation.OvsDBSet;
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;
41 import com.google.common.collect.Lists;
42 import com.google.common.collect.Sets;
43 import com.google.common.util.concurrent.ListenableFuture;
46 public class OvsDBClientTestIT extends OvsdbTestBase {
47 Logger logger = LoggerFactory.getLogger(OvsDBClientTestIT.class);
52 public void testTransact() throws IOException, InterruptedException, ExecutionException {
54 ListenableFuture<DatabaseSchema> schema = ovs.getSchema(OvsDBClient.OPEN_VSWITCH_SCHEMA, true);
55 TableSchema<GenericTableSchema> bridge = schema.get().table("Bridge", GenericTableSchema.class);
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());
62 ColumnSchema<GenericTableSchema, String> name = bridge.column("name", String.class);
63 ColumnSchema<GenericTableSchema, String> fail_mode = bridge.column("fail_mode", String.class);
66 * Adding a Test around non-atomic columns, since Set/Map columns requires special
67 * serialization/deserailzation for OVSDB Bidirection JSON.
68 * TODO : Replace this with regular Set and hide the ugliness of OvsDBSet inside the API.
70 ColumnSchema<GenericTableSchema, OvsDBSet> flood_vlans = bridge.column("flood_vlans", OvsDBSet.class);
71 OvsDBSet vlans = new OvsDBSet();
72 vlans.addAll(Sets.newHashSet(100, 200));
73 ListenableFuture<List<OperationResult>> results = ovs.transactBuilder()
74 .add(op.insert(bridge)
75 .value(name, "br-int")
76 .value(flood_vlans, vlans))
77 .add(op.update(bridge)
78 .set(fail_mode, "secure")
79 .where(name.opEqual("br-int"))
80 //.and(name.opEqual("br-int"))
82 .add(op.select(bridge)
84 .where(name.opEqual("br-int"))
89 List<OperationResult> operationResults = results.get();
90 Assert.assertFalse(operationResults.isEmpty());
91 System.out.println("Insert & Update operation results = " + operationResults);
93 results = ovs.transactBuilder()
94 .add(op.delete(bridge)
95 .where(name.opEqual("br-int"))
100 operationResults = results.get();
101 Assert.assertFalse(operationResults.isEmpty());
102 System.out.println("Delete operation results = " + operationResults);
105 * Adding a separate Abort operation in a transaction. Lets not mix this with other
106 * valid transactions as above.
108 results = ovs.transactBuilder()
109 .add(op.delete(bridge)
110 .where(name.opEqual("br-int"))
115 operationResults = results.get();
116 Assert.assertFalse(operationResults.isEmpty());
117 System.out.println("Abort operation results = " + operationResults);
122 public void testMonitorRequest() throws ExecutionException, InterruptedException {
124 DatabaseSchema dbSchema = ovs.getSchema(OvsDBClient.OPEN_VSWITCH_SCHEMA, true).get();
125 GenericTableSchema bridge = dbSchema.table("Bridge", GenericTableSchema.class);
127 List<MonitorRequest<GenericTableSchema>> monitorRequests = Lists.newArrayList();
129 MonitorRequestBuilder.builder(bridge)
130 .addColumn(bridge.column("name"))
131 .addColumn(bridge.column("fail_mode", String.class))
132 .addColumn(bridge.column("flood_vlans"))
133 .with(new MonitorSelect(true, true, true, true))
136 final List<Object> results = Lists.newArrayList();
138 MonitorHandle monitor = ovs.monitor(dbSchema, monitorRequests, new MonitorCallBack() {
140 public void update(TableUpdates result) {
142 System.out.println("result = " + result);
146 public void exception(Throwable t) {
148 System.out.println("t = " + t);
152 //for (int i = 0; i < 5 && results.isEmpty(); i++) { //wait 5 seconds to get a result
153 for (int i = 0; i < 5 ; i++) { //wait 5 seconds to get a result
154 System.out.println("waiting");
158 Assert.assertTrue(!results.isEmpty());
159 Object result = results.get(0);
160 Assert.assertTrue(result instanceof TableUpdates);
161 TableUpdate bridgeUpdate = ((TableUpdates) result).getUpdate(bridge);
162 Assert.assertNotNull(bridgeUpdate);
166 public void testGetDBs() throws ExecutionException, InterruptedException {
167 ListenableFuture<List<String>> databases = ovs.getDatabases();
168 List<String> dbNames = databases.get();
169 Assert.assertNotNull(dbNames);
170 Assert.assertTrue(dbNames.size() > 0);
174 public void initalize() throws IOException {
178 OvsdbRPC rpc = getTestConnection();
180 System.out.println("Unable to Establish Test Connection");
182 ExecutorService executorService = Executors.newFixedThreadPool(3);
183 ovs = new OvsDBClientImpl(rpc, executorService);
188 public void update(Object node, UpdateNotification upadateNotification) {
189 // TODO Auto-generated method stub
194 public void locked(Object node, List<String> ids) {
195 // TODO Auto-generated method stub
199 public void stolen(Object node, List<String> ids) {
200 // TODO Auto-generated method stub