X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDRO_tests%2Ftest_HYDROData_StricklerTable.cxx;h=05b2dc4b1c2124508522407161606b75653daacd;hb=9c947f35615e69e9e54a8c4b074dd1f2be13689c;hp=4c0b5f80a3e17f4bcb607bd901e79d839737a635;hpb=f64a3ec454c0826cd41e7bff2d2b64bccebb1cd4;p=modules%2Fhydro.git

diff --git a/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx b/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx
index 4c0b5f80..05b2dc4b 100644
--- a/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx
+++ b/src/HYDRO_tests/test_HYDROData_StricklerTable.cxx
@@ -19,12 +19,12 @@
 #include <test_HYDROData_StricklerTable.h>
 #include <HYDROData_Document.h>
 #include <HYDROData_StricklerTable.h>
+#include <HYDROData_Tool.h>
 #include <QStringList>
 #include <QColor>
 #include <QDir>
-#include <operators.h>
 
-const QString DEF_STR_PATH = qgetenv( "HYDRO_ROOT_DIR" ) + "/share/salome/resources/hydro/def_strickler_table.txt";
+const QString DEF_STR_PATH = qgetenv( "HYDRO_SRC_DIR" ) + "/src/HYDROGUI/resources/def_strickler_table.txt";
 
 void test_HYDROData_StricklerTable::test_import()
 {
@@ -83,3 +83,70 @@ void test_HYDROData_StricklerTable::test_import_export_equivalence()
   aDoc->Close();
 }
 
+void test_HYDROData_StricklerTable::test_type_by_attr()
+{
+  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 ) );
+
+  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();
+}