Fix delete button

src/lib/y2partitioner/widgets/delete_disk_partition_button.rb

@@ -114,22 +114,22 @@ def validate_partition_table
 
       # Deletes the indicated device
       #
-      # @note When the device is a partitionable, all its partitions are deleted.
+      # @note When the device has a partition table, all its partitions are deleted.
       #
       # @note Shadowing for BtrFS subvolumes is always refreshed.
       # @see Y2Storage::Filesystems::Btrfs.refresh_subvolumes_shadowing
       #
       # @param device [Y2Storage::BlkDevice]
       def delete_device(device)
-        if device.is?(:partitionable)
-          log.info "deleting partitions for #{device}"
-          device.partition_table.delete_all_partitions unless device.partition_table.nil?
-          UIState.instance.select_row(device)
-        else
+        if device.is?(:partition)
           log.info "deleting partition #{device}"
           partitionable = device.partitionable
           partitionable.partition_table.delete_partition(device)
           UIState.instance.select_row(partitionable)
+        else
+          log.info "deleting partitions for #{device}"
+          device.partition_table.delete_all_partitions unless device.partition_table.nil?
+          UIState.instance.select_row(device)
         end
 
         device_graph = DeviceGraphs.instance.current

test/y2partitioner/widgets/delete_disk_partition_button_test.rb

@@ -115,9 +115,27 @@
       context "when the confirm message is accepted" do
         let(:accept) { true }
 
-        it "deletes the device" do
-          subject.handle
-          expect(Y2Storage::BlkDevice.find_by_name(device_graph, device_name)).to be_nil
+        context "and the device is a partition" do
+          let(:device_name) { "/dev/sda2" }
+
+          it "deletes the partition" do
+            subject.handle
+            expect(Y2Storage::BlkDevice.find_by_name(device_graph, device_name)).to be_nil
+          end
+        end
+
+        context "and the device is not a partition" do
+          let(:device_name) { "/dev/sda" }
+
+          before do
+            allow(Yast::UI).to receive(:UserInput).and_return(:yes)
+          end
+
+          it "deletes all its partitions" do
+            expect(device.partitions).to_not be_empty
+            subject.handle
+            expect(device.partitions).to be_empty
+          end
         end
 
         it "refresh btrfs subvolumes shadowing" do