code revision for Linux

[modules/hydro.git] / src / HYDRO_tests / test_HYDROData_StricklerTable.cxx

diff --git a/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx b/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx
index 4cf579891f65d205661c11874810eea6c2198c79..a7c2da126e81bec5763c0da90480f39290d411fd 100644 (file)
--- a/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx
+++ b/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx
@@ -18,13 +18,135 @@
 
 #include <test_HYDROData_StricklerTable.h>
 #include <HYDROData_Document.h>
+#include <HYDROData_StricklerTable.h>
+#include <HYDROData_Tool.h>
+#include <QStringList>
+#include <QColor>
+#include <QDir>
 
-void test_HYDROData_StricklerTable::testImport()
+const QString DEF_STR_PATH = qgetenv( "HYDRO_ROOT_DIR" ) + "/share/salome/resources/hydro/def_strickler_table.txt";
+
+void test_HYDROData_StricklerTable::test_import()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+
+  Handle(HYDROData_StricklerTable) aTable =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+
+  CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
+
+  QStringList aTypes = aTable->GetTypes();
+  CPPUNIT_ASSERT_EQUAL( 8, aTypes.size() );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable->GetAttrName() );
+
+  QString aType = "Zones de champs cultivé à végétation basse";
+  CPPUNIT_ASSERT_EQUAL( aType, aTypes[3] );
+  CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable->GetColor( aType ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "512" ), aTable->GetAttrValue( aType ) );
+
+  aType = "Zones à forte urbanization (agglomération)";
+  CPPUNIT_ASSERT_EQUAL( aType, aTypes[7] );
+  CPPUNIT_ASSERT_EQUAL( QColor( 18, 52, 86 ), aTable->GetColor( aType ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetAttrValue( aType ) );
+
+  aDoc->Close();
+}
+
+void test_HYDROData_StricklerTable::test_import_export_equivalence()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+
+  Handle(HYDROData_StricklerTable) aTable =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+
+  CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
+  QString aTmpPath = QDir::tempPath() + "/stricker.txt";
+  CPPUNIT_ASSERT_EQUAL( true, aTable->Export( aTmpPath ) );
+
+  QFile aRefFile( DEF_STR_PATH ), aTmpFile( aTmpPath );
+  CPPUNIT_ASSERT_EQUAL( true, aRefFile.open( QFile::ReadOnly ) );
+  CPPUNIT_ASSERT_EQUAL( true, aTmpFile.open( QFile::ReadOnly ) );
+
+  QByteArray aRefContents = aRefFile.readAll();
+  QByteArray aTmpContents = aTmpFile.readAll();
+
+  bool isEqual = aRefContents.size()==aTmpContents.size();
+  CPPUNIT_ASSERT_EQUAL( true, isEqual );
+  for( int i=0, n=aRefContents.size(); isEqual && i<n; i++ )
+    if( aRefContents[i]!=aTmpContents[i] )
+      isEqual = false;
+
+  CPPUNIT_ASSERT_EQUAL( true, isEqual );
+
+  aRefFile.close();
+  aTmpFile.close();
+  aDoc->Close();
+}
+
+void test_HYDROData_StricklerTable::test_type_by_attr()
 {
   Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
 
-  Handle(HYDROData_Entity) anObj = aDoc->CreateObject( KIND_STRICKLER_TABLE );
-  TCollection_AsciiString aFileName = "test_name";
-  
+  Handle(HYDROData_StricklerTable) aTable =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+
+  CPPUNIT_ASSERT_EQUAL( true, aTable->Import( DEF_STR_PATH ) );
+
+  CPPUNIT_ASSERT_EQUAL( QString( "Zones de champs cultivé à végétation basse" ), aTable->GetType( "512" ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "123" ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable->GetType( "" ) );
+
+  aDoc->Close();
+}
+
+void test_HYDROData_StricklerTable::test_unique_attr_name()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+
+  Handle(HYDROData_StricklerTable) aTable1 =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+  Handle(HYDROData_StricklerTable) aTable2 =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+
+  CPPUNIT_ASSERT_EQUAL( true, aTable1->SetAttrName( "CODE_06" ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
+  CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
+
+  CPPUNIT_ASSERT_EQUAL( false, aTable2->SetAttrName( "CODE_06" ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
+  CPPUNIT_ASSERT_EQUAL( QString( "" ), aTable2->GetAttrName() );
+
+  CPPUNIT_ASSERT_EQUAL( true, aTable2->SetAttrName( "CODE_07" ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
+
+  CPPUNIT_ASSERT_EQUAL( false, aTable1->SetAttrName( "CODE_07" ) );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_06" ), aTable1->GetAttrName() );
+  CPPUNIT_ASSERT_EQUAL( QString( "CODE_07" ), aTable2->GetAttrName() );
+
+  aDoc->Close();
+}
+
+void test_HYDROData_StricklerTable::test_colors_sync()
+{
+  Handle(HYDROData_Document) aDoc = HYDROData_Document::Document(1);
+
+  Handle(HYDROData_StricklerTable) aTable1 =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+  Handle(HYDROData_StricklerTable) aTable2 =
+    Handle(HYDROData_StricklerTable)::DownCast( aDoc->CreateObject( KIND_STRICKLER_TABLE ) );
+
+  aTable1->Set( "test1", 10 );
+  aTable1->Set( "test2", 20 );
+  aTable1->SetColor( "test1", QColor( 255, 0, 0 ) );
+  aTable1->SetColor( "test2", QColor( 255, 0, 1 ) );
+  CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 0 ), aTable1->GetColor( "test1" ) );
+  CPPUNIT_ASSERT_EQUAL( QColor( 255, 0, 1 ), aTable1->GetColor( "test2" ) );
+
+  aTable2->Set( "test1", 15 );
+  aTable2->SetColor( "test1", QColor( 255, 255, 0 ) );
+  CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable1->GetColor( "test1" ) );
+  CPPUNIT_ASSERT_EQUAL( QColor( 255, 255, 0 ), aTable2->GetColor( "test1" ) );
+
   aDoc->Close();
 }