t-rust-less-lib 0.2.23

Password manager library
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use crate::api::{SecretType, SecretVersion};
use crate::block_store::{Change, ChangeLog, Operation};
use crate::secrets_store::index::Index;
use chrono::prelude::*;
use data_encoding::HEXLOWER;
use sha2::{Digest, Sha256};
use spectral::prelude::*;
use std::collections::HashMap;

#[derive(Default)]
struct TestStore {
  versions: HashMap<String, SecretVersion>,
  changes: Vec<Change>,
}

impl TestStore {
  fn add_secret_version(&mut self, secret_id: &str, version_id: i64) {
    let block_id = Self::generate_block_id(secret_id, version_id);
    let version = Self::generate_secret_version(secret_id, version_id);

    self.versions.insert(block_id.clone(), version);
    self.changes.push(Change {
      op: Operation::Add,
      block: block_id,
    });
  }

  fn make_changelog(&self, node: &str) -> ChangeLog {
    ChangeLog {
      node: node.to_string(),
      changes: self.changes.clone(),
    }
  }

  fn generate_secret_version(secret_id: &str, version_id: i64) -> SecretVersion {
    SecretVersion {
      secret_id: secret_id.to_string(),
      secret_type: SecretType::Login,
      timestamp: Utc.timestamp_opt(1000 + 1000 * version_id, 0).unwrap().into(),
      name: format!("{}_{}", secret_id, version_id),
      properties: Default::default(),
      tags: vec![],
      urls: vec![],
      deleted: false,
      recipients: vec![],
      attachments: vec![],
    }
  }

  fn generate_block_id(secret_id: &str, version_id: i64) -> String {
    let mut hasher = Sha256::new();

    hasher.update(secret_id);
    hasher.update(version_id.to_string());

    HEXLOWER.encode(&hasher.finalize())
  }
}

#[test]
fn test_process_change_logs() {
  let mut test_store: TestStore = Default::default();
  let mut index: Index = Default::default();

  for i in 0..10 {
    for j in 0..5 {
      test_store.add_secret_version(&format!("Secret_{}", i), j)
    }
  }

  assert_that(
    &index.process_change_logs(&[test_store.make_changelog("test_node")], |block_id| {
      Ok(test_store.versions.get(block_id).cloned())
    }),
  )
  .is_ok();

  let filter = Default::default();
  let mut all_matches = index.filter_entries(&filter).unwrap();

  assert_that(&all_matches.entries).has_length(10);

  test_store.changes.clear();

  for i in 10..15 {
    for j in 0..2 {
      test_store.add_secret_version(&format!("Secret_{}", i), j)
    }
  }

  assert_that(
    &index.process_change_logs(&[test_store.make_changelog("test_node")], |block_id| {
      Ok(test_store.versions.get(block_id).cloned())
    }),
  )
  .is_ok();

  all_matches = index.filter_entries(&filter).unwrap();

  assert_that(&all_matches.entries).has_length(15);
}