tests shared storage

custom-pallets/shared-storage/src/mock.rs

@@ -16,7 +16,7 @@ frame_support::construct_runtime!(
     pub enum Test
     {
         System: frame_system,
-        TemplateModule: pallet_template,
+        SharedStorage: pallet_template,
     }
 );
 

custom-pallets/shared-storage/src/tests.rs

@@ -7,9 +7,9 @@ fn it_works_for_default_value() {
         // Go past genesis block so events get deposited
         System::set_block_number(1);
         // Dispatch a signed extrinsic.
-        assert_ok!(TemplateModule::do_something(RuntimeOrigin::signed(1), 42));
+        assert_ok!(SharedStorage::do_something(RuntimeOrigin::signed(1), 42));
         // Read pallet storage and assert an expected result.
-        assert_eq!(TemplateModule::something(), Some(42));
+        assert_eq!(SharedStorage::something(), Some(42));
         // Assert that the correct event was deposited
         System::assert_last_event(
             Event::SomethingStored {
@@ -26,8 +26,132 @@ fn correct_error_for_none_value() {
     new_test_ext().execute_with(|| {
         // Ensure the expected error is thrown when no value is present.
         assert_noop!(
-            TemplateModule::cause_error(RuntimeOrigin::signed(1)),
+            SharedStorage::cause_error(RuntimeOrigin::signed(1)),
             Error::<Test>::NoneValue
         );
     });
 }
+
+
+#[test]
+fn test_set_department_reputation_score() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department = vec![1, 2, 3];
+        let score = 10;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department.clone(), score));
+
+        let reputation_score = SharedStorage::reputation_score(address);
+        assert!(reputation_score.is_some());
+        let reputation_score = reputation_score.unwrap();
+        assert_eq!(reputation_score.get_department_score(department.clone()), Some(score));
+    });
+}
+
+#[test]
+fn test_update_department_reputation_score() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department = vec![1, 2, 3];
+        let score = 10;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department.clone(), score));
+
+        let new_score = 20;
+        assert_ok!(SharedStorage::update_department_reputation_score(address, department.clone(), new_score));
+
+        let reputation_score = SharedStorage::reputation_score(address);
+        assert!(reputation_score.is_some());
+        let reputation_score = reputation_score.unwrap();
+        assert_eq!(reputation_score.get_department_score(department.clone()), Some(new_score));
+    });
+}
+
+#[test]
+fn test_add_score_to_department() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department = vec![1, 2, 3];
+        let score = 10;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department.clone(), score));
+
+        let amount = 5;
+        assert_ok!(SharedStorage::add_reputation_score_to_department(address, department.clone(), amount));
+
+        let reputation_score = SharedStorage::reputation_score(address);
+        assert!(reputation_score.is_some());
+        let reputation_score = reputation_score.unwrap();
+        assert_eq!(reputation_score.get_department_score(department.clone()), Some(score + amount));
+    });
+}
+
+#[test]
+fn test_subtract_score_from_department() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department = vec![1, 2, 3];
+        let score = 10;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department.clone(), score));
+
+        let amount = 5;
+        assert_ok!(SharedStorage::subtract_reputation_score_from_department(address, department.clone(), amount));
+
+        let reputation_score = SharedStorage::reputation_score(address);
+        assert!(reputation_score.is_some());
+        let reputation_score = reputation_score.unwrap();
+        assert_eq!(reputation_score.get_department_score(department.clone()), Some(score - amount));
+    });
+}
+
+#[test]
+fn test_get_department_reputation_score() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department = vec![1, 2, 3];
+        let score = 10;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department.clone(), score));
+
+        let department_score = SharedStorage::get_department_reputation_score(address, department.clone());
+        assert_eq!(department_score, Some(score));
+    });
+}
+
+#[test]
+fn test_get_all_department_reputation_scores() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department1 = vec![1, 2, 3];
+        let department2 = vec![4, 5, 6];
+        let score1 = 10;
+        let score2 = 20;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department1.clone(), score1));
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department2.clone(), score2));
+
+        let all_departments = SharedStorage::get_all_department_reputation_scores(address);
+        assert_eq!(all_departments.len(), 2);
+        assert!(all_departments.contains(&(department1.clone(), score1)));
+        assert!(all_departments.contains(&(department2.clone(), score2)));
+    });
+}
+
+#[test]
+fn test_get_total_reputation_score() {
+    new_test_ext().execute_with(|| {
+        let address = 1;
+        let department1 = vec![1, 2, 3];
+        let department2 = vec![4, 5, 6];
+        let score1 = 10;
+        let score2 = 20;
+
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department1.clone(), score1));
+        assert_ok!(SharedStorage::set_department_reputation_score(address, department2.clone(), score2));
+
+        let total_score = SharedStorage::get_total_reputation_score(address);
+        assert_eq!(total_score, score1 + score2);
+    });
+}