@Test public void testUpdateData() { String[] names = new String[]{ "name_0" }; Object[] values = new Object[]{ "value_0" }; StaticDataRow dataRow = new StaticDataRow( names, values ); Object[] newValues = new Object[]{ "new_value_0" }; dataRow.updateData( newValues ); assertThat( dataRow.getColumnNames(), is( equalTo( new String[]{ "name_0" } ) ) ); assertThat( (String) dataRow.get( "name_0" ), is( equalTo( "new_value_0" ) ) ); }
@Test public void testCreation() { StaticDataRow dataRow = new StaticDataRow(); assertThat( dataRow.getColumnNames(), is( equalTo( new String[]{ } ) ) ); StaticDataRow staticDataRowParam = mock( StaticDataRow.class ); dataRow = new StaticDataRow( staticDataRowParam ); assertThat( dataRow.getColumnNames(), is( equalTo( new String[]{ } ) ) ); DataRow dataRowParam = mock( DataRow.class ); doReturn( new String[]{ "test_name" } ).when( dataRowParam ).getColumnNames(); doReturn( "test_val" ).when( dataRowParam ).get( "test_name" ); dataRow = new StaticDataRow( dataRowParam ); assertThat( dataRow.getColumnNames(), is( equalTo( new String[]{ "test_name" } ) ) ); assertThat( (String) dataRow.get( "test_name" ), is( equalTo( "test_val" ) ) ); String[] names = new String[]{ "name_0", "name_1" }; Object[] values = new Object[]{ "value_0" }; dataRow = new StaticDataRow( names, values ); assertThat( dataRow.getColumnNames(), is( equalTo( new String[]{ "name_0" } ) ) ); assertThat( (String) dataRow.get( "name_0" ), is( equalTo( "value_0" ) ) ); Map<String, Object> parameterValues = new HashMap<String, Object>(); parameterValues.put( "name_0", "value_0" ); dataRow = new StaticDataRow( parameterValues ); assertThat( dataRow.getColumnNames(), is( equalTo( new String[]{ "name_0" } ) ) ); assertThat( (String) dataRow.get( "name_0" ), is( equalTo( "value_0" ) ) ); }