рефакторинг метода Скопировать, с тестом

Author: Mr-Rm

src/OneScript.StandardLibrary/Collections/ValueTable/ValueTable.cs

@@ -546,59 +546,43 @@ public ValueTable CopyColumns(string columnNames = null)
         [ContextMethod("Скопировать", "Copy")]
         public ValueTable Copy(IValue rows = null, string columnNames = null)
         {
-            var Result = CopyColumns(columnNames);
+            var result = CopyColumns(columnNames);
             var columns = GetProcessingColumnList(columnNames);
 
-            IEnumerable<ValueTableRow> requestedRows;
-            if (rows == null)
-            {
-                requestedRows = _rows;
-            }
-            else
-            {
-                if (rows is StructureImpl structure)
-                    requestedRows = FindRows(structure).Select(x => x as ValueTableRow);
-                else if (rows is ArrayImpl array)
-                    requestedRows = GetRowsEnumByArray(array);
-                else
-                    throw RuntimeException.InvalidArgumentType();
-            }
+            IEnumerable<ValueTableRow> requestedRows = rows switch
+            {
+                null => _rows,
+                StructureImpl structure => FindRows(structure).Select(x => x as ValueTableRow),
+                ArrayImpl array => GetRowsEnumByArray(array),
+                _ => throw RuntimeException.InvalidArgumentType(),
+            };
 
             var columnMap = new Dictionary<ValueTableColumn, ValueTableColumn>();
             foreach (var column in columns)
             {
-                var destinationColumn = Result.Columns.FindColumnByName(column.Name);
+                var destinationColumn = result.Columns.FindColumnByName(column.Name);
                 columnMap.Add(column, destinationColumn);
             }
 
             foreach (var row in requestedRows)
             {
-                var new_row = Result.Add();
+                var new_row = result.Add();
                 foreach (var Column in columns)
                 {
                     new_row.Set(columnMap[Column], row.Get(Column));
                 }
             }
 
-            return Result;
+            return result;
         }
 
-        private IEnumerable<ValueTableRow> GetRowsEnumByArray(IValue rows)
-        {
-            IEnumerable<ValueTableRow> requestedRows;
-            var rowsArray = rows as ArrayImpl;
+        private IEnumerable<ValueTableRow> GetRowsEnumByArray(ArrayImpl rowsArray)
+        {
             if (rowsArray == null)
                 throw RuntimeException.InvalidArgumentType();
 
-            requestedRows = rowsArray.Select(x =>
-            {
-                var vtr = x as ValueTableRow;
-                if (vtr == null || vtr.Owner() != this)
-                    throw RuntimeException.InvalidArgumentValue();
-
-                return vtr;
-            });
-            return requestedRows;
+            return rowsArray.Select(x => x is ValueTableRow vtr && vtr.Owner() == this ? vtr
+                : throw RuntimeException.InvalidArgumentValue());
         }
 
         private struct ValueTableSortRule

tests/valuetable.os

@@ -30,7 +30,9 @@
 	ВсеТесты.Добавить("ТестДолжен_НайтиНесколькоСтрокВТаблице");
 	ВсеТесты.Добавить("ТестДолжен_СкопироватьТаблицуПолностью");
 	ВсеТесты.Добавить("ТестДолжен_СкопироватьТаблицуПоМассивуСтрок");
+	ВсеТесты.Добавить("ТестДолжен_ПроверитьИсключениеПриКопированииТаблицыПоМассивуСтрок");
 	ВсеТесты.Добавить("ТестДолжен_СкопироватьТаблицуНесколькоКолонок");
+	ВсеТесты.Добавить("ТестДолжен_СкопироватьТаблицуПоОтбору");
 	ВсеТесты.Добавить("ТестДолжен_ПроверитьСверткуБезУказанияКолонок");
 
 	ВсеТесты.Добавить("ТестДолжен_ЗагрузитьКолонку");
@@ -537,6 +539,44 @@
 
 КонецПроцедуры
 
+Процедура ТестДолжен_СкопироватьТаблицуПоОтбору() Экспорт
+
+	Т = СоздатьТаблицуСДанными();
+	Т.Добавить().Значение = 4;
+
+	Отбор = Новый Структура("Значение", 4);
+	Т2 = Т.Скопировать(Отбор,"Ключ");
+
+	юТест.ПроверитьНеравенство(Т, Т2);
+	юТест.ПроверитьРавенство(2, Т2.Количество());
+	юТест.ПроверитьРавенство(1, Т2.Колонки.Количество());
+	юТест.ПроверитьРавенство("Ключ4", Т2[0].Ключ);
+	юТест.ПроверитьРавенство(Неопределено, Т2[1].Ключ);
+
+КонецПроцедуры
+
+Процедура ТестДолжен_ПроверитьИсключениеПриКопированииТаблицыПоМассивуСтрок() Экспорт
+
+	Т = СоздатьТаблицуСДанными();
+
+	Т0 = Новый ТаблицаЗначений();
+	Т0.Колонки.Добавить("Тест");
+	Т0.Добавить().Тест = 1;
+
+	МассивСтрок = Новый Массив;
+	МассивСтрок.Добавить(Т0[0]);
+
+	Попытка
+		Т2 = Т.Скопировать(МассивСтрок);
+	Исключение
+		Возврат;
+	КонецПопытки;
+	
+	ВызватьИсключение "Ожидали исключение, но его не было";
+
+КонецПроцедуры
+
+
 Процедура ТестДолжен_ПроверитьСверткуБезУказанияКолонок() Экспорт
 
 	Перем Т;