DataSet mockDataSet = mock(DataSet.class);
/* keyspace */
KeyspaceModel keyspace = new KeyspaceModel();
keyspace.setName("beautifulKeyspaceName");
List<ColumnFamilyModel> columnFamilies = new ArrayList<ColumnFamilyModel>();
ColumnFamilyModel columnFamily = new ColumnFamilyModel();
columnFamily.setName("beautifulColumnFamilyName");
columnFamily.setType(ColumnType.SUPER);
List<RowModel> rows = new ArrayList<RowModel>();
RowModel row1 = new RowModel();
row1.setKey(new GenericType("01", GenericTypeEnum.BYTES_TYPE));
List<SuperColumnModel> superColumns1 = new ArrayList<SuperColumnModel>();
SuperColumnModel superColumn11 = new SuperColumnModel();
superColumn11.setName(new GenericType("11", GenericTypeEnum.BYTES_TYPE));
List<ColumnModel> columns11 = new ArrayList<ColumnModel>();
columns11.add(constructDefaultColumnForMock("1110", "1110"));
columns11.add(constructDefaultColumnForMock("1120", "1120"));
superColumn11.setColumns(columns11);
superColumns1.add(superColumn11);
SuperColumnModel superColumn12 = new SuperColumnModel();
superColumn12.setName(new GenericType("12", GenericTypeEnum.BYTES_TYPE));
List<ColumnModel> columns12 = new ArrayList<ColumnModel>();
columns12.add(constructDefaultColumnForMock("1210", "1210"));
columns12.add(constructDefaultColumnForMock("1220", "1220"));
superColumn12.setColumns(columns12);
superColumns1.add(superColumn12);
row1.setSuperColumns(superColumns1);
rows.add(row1);
RowModel row2 = new RowModel();
row2.setKey(new GenericType("02", GenericTypeEnum.BYTES_TYPE));
List<SuperColumnModel> superColumns2 = new ArrayList<SuperColumnModel>();
SuperColumnModel superColumn21 = new SuperColumnModel();
superColumn21.setName(new GenericType("21", GenericTypeEnum.BYTES_TYPE));
List<ColumnModel> columns21 = new ArrayList<ColumnModel>();
columns21.add(constructDefaultColumnForMock("2110", "2110"));
columns21.add(constructDefaultColumnForMock("2120", "2120"));
superColumn21.setColumns(columns21);
superColumns2.add(superColumn21);
row2.setSuperColumns(superColumns2);
rows.add(row2);
columnFamily.setRows(rows);
columnFamilies.add(columnFamily);
keyspace.setColumnFamilies(columnFamilies);
when(mockDataSet.getKeyspace()).thenReturn(keyspace);
when(mockDataSet.getColumnFamilies()).thenReturn(keyspace.getColumnFamilies());