src/HYDRO_tests/test_HYDROData_StricklerTable.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #include <test_HYDROData_StricklerTable.h>
  20 #include <HYDROData_Document.h>
  21 #include <HYDROData_StricklerTable.h>
  22 #include <HYDROData_Tool.h>
  23 #include <QStringList>
  24 #include <QColor>
  25 #include <QDir>
  26
  27 const QString DEF_STR_PATH = qgetenv( "HYDRO_SRC_DIR" ) + "/src/HYDROGUI/resources/def_strickler_table.txt";
  28
  29 void test_HYDROData_StricklerTable::test_import()
  30 {
  31   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  32
  33   Handle(HYDROData_StricklerTable) aTable =
  34     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  35
  36   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  37
  38   QStringList aTypes = aTable->GetTypes();
  39   CPPUNIT_ASSERT_EQUAL( 8, aTypes.size() );
  40   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable->GetAttrName() );
  41
  42   QString aType = "Zones de champs cultivé à végétation basse";
  43   CPPUNIT_ASSERT_EQUAL( aType, aTypes[3] );
  44   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable->GetColor( aType ) );
  45   CPPUNIT_ASSERT_EQUAL( QString( "512" ), aTable->GetAttrValue( aType ) );
  46
  47   aType = "Zones à forte urbanization (agglomération)";
  48   CPPUNIT_ASSERT_EQUAL( aType, aTypes[7] );
  49   CPPUNIT_ASSERT_EQUAL( QColor( 18, 52, 86 ), aTable->GetColor( aType ) );
  50   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetAttrValue( aType ) );
  51
  52   aDoc->Close();
  53 }
  54
  55 void test_HYDROData_StricklerTable::test_import_export_equivalence()
  56 {
  57   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  58
  59   Handle(HYDROData_StricklerTable) aTable =
  60     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  61
  62   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  63   QString aTmpPath = QDir::tempPath() + "/stricker.txt";
  64   CPPUNIT_ASSERT_EQUAL( true, aTable->Export( aTmpPath ) );
  65
  66   QFile aRefFile( DEF_STR_PATH ), aTmpFile( aTmpPath );
  67   CPPUNIT_ASSERT_EQUAL( true, aRefFile.open( QFile::ReadOnly ) );
  68   CPPUNIT_ASSERT_EQUAL( true, aTmpFile.open( QFile::ReadOnly ) );
  69
  70   QByteArray aRefContents = aRefFile.readAll();
  71   QByteArray aTmpContents = aTmpFile.readAll();
  72
  73   bool isEqual = aRefContents.size()==aTmpContents.size();
  74   CPPUNIT_ASSERT_EQUAL( true, isEqual );
  75   for( int i=0, n=aRefContents.size(); isEqual && i<n; i++ )
  76     if( aRefContents[i]!=aTmpContents[i] )
  77       isEqual = false;
  78
  79   CPPUNIT_ASSERT_EQUAL( true, isEqual );
  80
  81   aRefFile.close();
  82   aTmpFile.close();
  83   aDoc->Close();
  84 }
  85
  86 void test_HYDROData_StricklerTable::test_type_by_attr()
  87 {
  88   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
  89
  90   Handle(HYDROData_StricklerTable) aTable =
  91     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
  92
  93   CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
  94
  95   CPPUNIT_ASSERT_EQUAL( QString( "Zones de champs cultivé à végétation basse" ), aTable->GetType( "512" ) );
  96   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "123" ) );
  97   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "" ) );
  98
  99   aDoc->Close();
 100 }
 101
 102 void test_HYDROData_StricklerTable::test_unique_attr_name()
 103 {
 104   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 105
 106   Handle(HYDROData_StricklerTable) aTable1 =
 107     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 108   Handle(HYDROData_StricklerTable) aTable2 =
 109     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 110
 111   CPPUNIT_ASSERT_EQUAL( true, aTable1->SetAttrName( "CODE_06" ) );
 112   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 113   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
 114
 115   CPPUNIT_ASSERT_EQUAL( false, aTable2->SetAttrName( "CODE_06" ) );
 116   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 117   CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
 118
 119   CPPUNIT_ASSERT_EQUAL( true, aTable2->SetAttrName( "CODE_07" ) );
 120   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 121   CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
 122
 123   CPPUNIT_ASSERT_EQUAL( false, aTable1->SetAttrName( "CODE_07" ) );
 124   CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
 125   CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
 126
 127   aDoc->Close();
 128 }
 129
 130 void test_HYDROData_StricklerTable::test_colors_sync()
 131 {
 132   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 133
 134   Handle(HYDROData_StricklerTable) aTable1 =
 135     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 136   Handle(HYDROData_StricklerTable) aTable2 =
 137     Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
 138
 139   aTable1->Set( "test1", 10 );
 140   aTable1->Set( "test2", 20 );
 141   aTable1->SetColor( "test1", QColor( 255, 0, 0 ) );
 142   aTable1->SetColor( "test2", QColor( 255, 0, 1 ) );
 143   CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 0 ), aTable1->GetColor( "test1" ) );
 144   CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 1 ), aTable1->GetColor( "test2" ) );
 145
 146   aTable2->Set( "test1", 15 );
 147   aTable2->SetColor( "test1", QColor( 255, 255, 0 ) );
 148   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable1->GetColor( "test1" ) );
 149   CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable2->GetColor( "test1" ) );
 150
 151   aDoc->Close();
 152 }