codetether-rlm 0.1.0

Recursive Language Model processing for CodeTether
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13

use super::sample;
use crate::oracle::tree_sitter_oracle::TreeSitterOracle;

#[test]
fn get_enums_finds_all() {
    let mut oracle = TreeSitterOracle::new(sample::rust_code());
    let enums = oracle.get_enums().unwrap();
    assert!(!enums.is_empty());

    let status = enums.iter().find(|e| e.name == "Status").unwrap();
    assert!(status.variants.contains(&"Active".to_string()));
    assert!(status.variants.contains(&"Inactive".to_string()));
}