List donations for projects

ilanolkies · Mar 19, 2019 · 93d2707 · 93d2707
1 parent 3acdcea
commit 93d2707
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/contracts/DirectDonate.sol b/contracts/DirectDonate.sol
@@ -11,6 +11,7 @@ contract DirectDonate {
         address receiver;
         string url;
         address[] donors;
+        uint[] donations;
     }
 
     modifier onlyOwner () {
@@ -33,6 +34,8 @@ contract DirectDonate {
         project.url = url;
         address[] memory donors;
         project.donors = donors;
+        uint[] memory donations;
+        project.donations = donations;
 
         projects.push(project);
     }
@@ -45,6 +48,7 @@ contract DirectDonate {
         Project storage project = projects[index];
         address receiver = project.receiver;
         project.donors.push(msg.sender);
+        project.donations.push(msg.value);
 
         receiver.transfer(msg.value);
     }
@@ -53,4 +57,9 @@ contract DirectDonate {
         Project storage project = projects[index];
         return project.donors;
     }
+
+    function donations (uint index) public view returns (uint[] memory) {
+        Project storage project = projects[index];
+        return project.donations;
+    }
 }
diff --git a/test/direct_donate.js b/test/direct_donate.js
@@ -137,4 +137,16 @@ contract('DirectDonate', async (accounts) => {
     assert.equal(donors[0], accounts[2]);
     assert.equal(donors[1], accounts[3]);
   });
+
+  it('should return project\'s donations', async () => {
+    await addProject();
+
+    const value = 1e18;
+    await directDonate.donate(0, { from: accounts[2], value });
+
+    var donations = await directDonate.donations(0);
+
+    assert.equal(donations.length, 1);
+    assert.equal(donations[0], value);
+  });
 });