Add project list

ilanolkies · Mar 19, 2019 · 7904b90 · 7904b90
1 parent ebf0bfc
commit 7904b90
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/contracts/DirectDonate.sol b/contracts/DirectDonate.sol
@@ -4,7 +4,7 @@ pragma solidity ^0.4.22;
 contract DirectDonate {
     address public owner;
 
-    uint public projectIndex;
+    string[] public projects;
 
     modifier onlyOwner () {
         require(msg.sender == owner);
@@ -13,10 +13,13 @@ contract DirectDonate {
 
     constructor() public {
         owner = msg.sender;
-        projectIndex = 0;
     }
 
-    function addProject () public {
+    function projectIndex () public view returns (uint) {
-        projectIndex++;
+        return projects.length;
+    }
+
+    function addProject (string projectName) public {
+        projects.push(projectName);
     }
 }
diff --git a/test/direct_donate.js b/test/direct_donate.js
@@ -27,10 +27,20 @@ contract('DirectDonate', async (accounts) => {
   it('should add a project', async () => {
     const previousQuantity = await directDonate.projectIndex();
 
-    await directDonate.addProject();
+    await directDonate.addProject('');
 
     const projectQuantity = await directDonate.projectIndex();
 
     assert.equal(projectQuantity, previousQuantity.toNumber() + 1);
   });
+
+  it('should add and return the project', async () => {
+    const projectName = 'NewProject';
+
+    await directDonate.addProject(projectName);
+
+    const project = await directDonate.projects(0);
+
+    assert.equal(project, projectName);
+  });
 });