code-gen 0.10.0

This library aids in code generation.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::{CodeBuffer, Expression};

/// A function receiver.
#[derive(Copy, Clone, Ord, PartialOrd, Eq, PartialEq, Hash, Debug, Default)]
pub enum Receiver {
    #[default]
    Borrowed,
    BorrowedMut,
    Owned,
    OwnedMut,
}

impl Expression for Receiver {
    fn write(&self, b: &mut CodeBuffer) {
        let s: &str = match self {
            Self::Borrowed => "&self",
            Self::BorrowedMut => "&mut self",
            Self::Owned => "self",
            Self::OwnedMut => "mut self",
        };
        b.write(s);
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn borrowed() {
        assert_eq!(Receiver::Borrowed.to_code(), "&self");
    }

    #[test]
    fn borrowed_mut() {
        assert_eq!(Receiver::BorrowedMut.to_code(), "&mut self");
    }

    #[test]
    fn owned() {
        assert_eq!(Receiver::Owned.to_code(), "self");
    }

    #[test]
    fn owned_mut() {
        assert_eq!(Receiver::OwnedMut.to_code(), "mut self");
    }

    #[test]
    fn default_is_borrowed() {
        assert_eq!(Receiver::default(), Receiver::Borrowed);
    }
}