-
Notifications
You must be signed in to change notification settings - Fork 50
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Della - Pet Park Assignment #19
base: main
Are you sure you want to change the base?
Changes from 4 commits
70ca263
ceef751
dbbfe1b
1597a1d
eeaaf60
979503d
19a1686
b3ef630
87d2526
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,7 +1,126 @@ | ||
//SPDX-License-Identifier: Unlicense | ||
pragma solidity ^0.8.0; | ||
|
||
|
||
contract PetPark { | ||
address owner; | ||
|
||
enum AnimalType { | ||
None, | ||
Fish, | ||
Cat, | ||
Dog, | ||
Rabbit, | ||
Parrot | ||
} | ||
enum Gender { | ||
Male, | ||
Female | ||
} | ||
|
||
struct User { | ||
AnimalType animal; | ||
Gender gender; | ||
uint age; | ||
} | ||
|
||
mapping(AnimalType => uint) public animalCounts; | ||
mapping(address => User) public userInfo; | ||
|
||
event Added(AnimalType animal, uint count); | ||
event Borrowed(AnimalType animal); | ||
event Returned(AnimalType animal); | ||
|
||
constructor() { | ||
owner = msg.sender; | ||
} | ||
|
||
modifier onlyOwner() { | ||
require(owner == msg.sender, "Not an owner"); | ||
_; | ||
} | ||
|
||
modifier isOldEnough(uint age) { | ||
require(age != 0, "Invalid Age"); | ||
_; | ||
} | ||
|
||
modifier isValidAnimal(AnimalType animal) { | ||
require(animal != AnimalType.None, "Invalid animal"); | ||
_; | ||
} | ||
|
||
modifier isValidAnimalType(AnimalType animal) { | ||
require(animal != AnimalType.None, "Invalid animal type"); | ||
_; | ||
} | ||
|
||
modifier isAvailableAnimal(AnimalType animal) { | ||
require(animalCounts[animal] > 0, "Selected animal not available"); | ||
_; | ||
} | ||
|
||
function add(AnimalType _animal, uint _count) | ||
external | ||
onlyOwner | ||
isValidAnimal(_animal) | ||
{ | ||
animalCounts[_animal] = animalCounts[_animal] + _count; | ||
emit Added(_animal, animalCounts[_animal]); | ||
} | ||
|
||
function borrow( | ||
uint _age, | ||
Gender _gender, | ||
AnimalType _animal | ||
) | ||
external | ||
isOldEnough(_age) | ||
isValidAnimalType(_animal) | ||
isAvailableAnimal(_animal) | ||
{ | ||
User storage u = userInfo[msg.sender]; | ||
|
||
if (u.age != _age && u.age != 0) { | ||
revert("Invalid Age"); | ||
} | ||
|
||
if (u.gender != _gender && _gender == Gender.Female) { | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. This is what is unclear to me. Zero value for enums is the first value, so the "default" gender is Male. I'm struggling to work that logic into the check for when a user hasn't borrowed any animals yet. This makes sense to me but it leads to the women <40 cat test failing. There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Instead of thinking of a negative test case, how would you write a condition for women under the age of 40 inside the if clause? And then lets if they are requesting a cat, if so then revert. |
||
revert("Invalid Gender"); | ||
} | ||
|
||
u.age = _age; | ||
u.gender = _gender; | ||
|
||
if (u.animal != AnimalType.None) { | ||
revert("Already adopted a pet"); | ||
blackqueentech marked this conversation as resolved.
Show resolved
Hide resolved
|
||
} | ||
|
||
if ( | ||
u.gender == Gender.Female && _animal == AnimalType.Cat && u.age < 40 | ||
) { | ||
revert("Invalid animal for women under 40"); | ||
} | ||
|
||
if (u.gender == Gender.Male) { | ||
if (_animal != AnimalType.Fish && _animal != AnimalType.Dog) { | ||
revert("Invalid animal for men"); | ||
} | ||
} | ||
|
||
u.animal = _animal; | ||
uint count = animalCounts[_animal]; | ||
animalCounts[_animal] = count - 1; | ||
emit Borrowed(_animal); | ||
} | ||
|
||
} | ||
function giveBackAnimal() external { | ||
User storage u = userInfo[msg.sender]; | ||
if (u.animal == AnimalType.None) { | ||
revert("No borrowed pets"); | ||
} | ||
uint count = animalCounts[u.animal]; | ||
animalCounts[u.animal] = count + 1; | ||
emit Returned(u.animal); | ||
u.animal = AnimalType.None; | ||
} | ||
} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The age, gender and animal checks for the case when the pet was already borrowed, should be in a single if clause. Once you move them, all your tests will pass.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
How can you add them into one clause while keeping all the separate revert statements? ("Invalid Age", "Invalid Gender", "Already borrowed a pet?")
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
My bad.I meant
if block
notif clause
. Something like this