From eda3f2462fd056e76cffb98cbc523a026227e877 Mon Sep 17 00:00:00 2001
From: Partha Aji <paji@redhat.com>
Date: Thu, 20 Feb 2020 16:58:32 -0500
Subject: [PATCH] Refs #29104 -  Adding a drpm unit type

---
 lib/runcible/extensions/drpm.rb | 24 ++++++++++++++++++++++++
 test/extensions/drpm_test.rb    | 26 ++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 lib/runcible/extensions/drpm.rb
 create mode 100644 test/extensions/drpm_test.rb

diff --git a/lib/runcible/extensions/drpm.rb b/lib/runcible/extensions/drpm.rb
new file mode 100644
index 00000000..5ad875d9
--- /dev/null
+++ b/lib/runcible/extensions/drpm.rb
@@ -0,0 +1,24 @@
+module Runcible
+  module Extensions
+    class Drpm < Runcible::Extensions::Unit
+      def self.content_type
+        'drpm'
+      end
+
+      # This function is not implemented for RPMs since they do not have content IDs
+      def find
+        fail NotImplementedError
+      end
+
+      # This function is not implemented for RPMs since they do not have content IDs
+      def find_all
+        fail NotImplementedError
+      end
+
+      # This function is not implemented for RPMs since they do not have content IDs
+      def unassociate_ids_from_repo(repo_id, ids)
+        fail NotImplementedError
+      end
+    end
+  end
+end
diff --git a/test/extensions/drpm_test.rb b/test/extensions/drpm_test.rb
new file mode 100644
index 00000000..27431bc7
--- /dev/null
+++ b/test/extensions/drpm_test.rb
@@ -0,0 +1,26 @@
+require 'rubygems'
+require 'minitest/autorun'
+
+require './lib/runcible'
+require './test/extensions/unit_base'
+require './test/support/repository_support'
+
+module Extensions
+  class TestDrpm < MiniTest::Unit::TestCase
+    def self.before_suite
+      @@extension = TestRuncible.server.extensions.drpm
+    end
+
+    def test_content_type
+      assert_equal 'drpm', @@extension.content_type
+    end
+
+    def test_find
+      assert_raises(NotImplementedError) { @@extension.find }
+    end
+
+    def test_find_all
+      assert_raises(NotImplementedError) { @@extension.find_all }
+    end
+  end
+end