List donors for porjects

ilanolkies · Mar 19, 2019 · 3acdcea · 3acdcea
1 parent 0c10ccd
commit 3acdcea
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/contracts/DirectDonate.sol b/contracts/DirectDonate.sol
@@ -10,6 +10,7 @@ contract DirectDonate {
         string name;
         address receiver;
         string url;
+        address[] donors;
     }
 
     modifier onlyOwner () {
@@ -30,6 +31,8 @@ contract DirectDonate {
         project.name = name;
         project.receiver = receiver;
         project.url = url;
+        address[] memory donors;
+        project.donors = donors;
 
         projects.push(project);
     }
@@ -39,8 +42,15 @@ contract DirectDonate {
     }
 
     function donate (uint index) public payable {
-        address receiver = projects[index].receiver;
+        Project storage project = projects[index];
+        address receiver = project.receiver;
+        project.donors.push(msg.sender);
 
         receiver.transfer(msg.value);
     }
+
+    function donors (uint index) public view returns (address[] memory) {
+        Project storage project = projects[index];
+        return project.donors;
+    }
 }
diff --git a/test/direct_donate.js b/test/direct_donate.js
@@ -118,4 +118,23 @@ contract('DirectDonate', async (accounts) => {
       assert.equal(balance, previousBalance.toNumber() + value);
     }
   });
+
+  it('should return project\'s donors', async () => {
+    await addProject();
+
+    await directDonate.donate(0, { from: accounts[2], value: 1e18 });
+
+    var donors = await directDonate.donors(0);
+
+    assert.equal(donors.length, 1);
+    assert.equal(donors[0], accounts[2]);
+
+    await directDonate.donate(0, { from: accounts[3], value: 1e18 });
+
+    donors = await directDonate.donors(0);
+
+    assert.equal(donors.length, 2);
+    assert.equal(donors[0], accounts[2]);
+    assert.equal(donors[1], accounts[3]);
+  });
 });