extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-08-16 17:51:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

nu-table: Add a few tests (#6074)

Browse Source 
Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com>
This commit is contained in:
Maxim Zhiburt
2022-07-19 20:35:13 +03:00
committed by GitHub
parent 41669e60c8
commit b4a7e7e6e9
2 changed files with 727 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

464
crates/nu-table/tests/style.rs
View File

@ -1,34 +1,454 @@
use std::collections::HashMap;
use nu_protocol::Config;
use nu_table::{Alignments, StyledString, Table, TableTheme, TextStyle};
use nu_table::{Alignments, StyledString, Table, TableTheme as theme, TextStyle};
#[test]
fn test_rounded_style() {
let headers = vec![no_style_str("Hello"), no_style_str("World")];
let data = vec![vec![no_style_str("1"), no_style_str("2")]];
let table = Table::new(headers, data.clone(), TableTheme::rounded());
let table = table.draw_table(
&Config::default(),
&HashMap::default(),
Alignments::default(),
std::usize::MAX,
fn test_rounded() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::rounded())),
"╭───┬───┬───┬───╮\n\
│ 0 │ 1 │ 2 │ 3 │\n\
├───┼───┼───┼───┤\n\
│ 0 │ 1 │ 2 │ 3 │\n\
│ 0 │ 1 │ 2 │ 3 │\n\
╰───┴───┴───┴───╯"
);
assert_eq!(table.as_deref(), Some("╭───────┬───────╮\n│ Hello │ World │\n├───────┼───────┤\n│ 1 │ 2 │\n╰───────┴───────╯"));
let table = Table::new(Vec::new(), data, TableTheme::rounded());
let table = table.draw_table(
&Config::default(),
&HashMap::default(),
Alignments::default(),
std::usize::MAX,
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::rounded())),
"╭───┬───┬───┬───╮\n\
│ 0 │ 1 │ 2 │ 3 │\n\
├───┼───┼───┼───┤\n\
│ 0 │ 1 │ 2 │ 3 │\n\
╰───┴───┴───┴───╯"
);
assert_eq!(table.as_deref(), Some("╭───┬───╮\n│ 1 │ 2 │\n╰───┴───╯"));
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::rounded())),
"╭───┬───┬───┬───╮\n\
│ 0 │ 1 │ 2 │ 3 │\n\
╰───┴───┴───┴───╯"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::rounded())),
"╭───┬───┬───┬───╮\n\
│ 0 │ 1 │ 2 │ 3 │\n\
│ 0 │ 1 │ 2 │ 3 │\n\
╰───┴───┴───┴───╯"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::rounded())),
""
);
}
fn no_style_str(text: &str) -> StyledString {
StyledString::new(text.to_owned(), TextStyle::default())
#[test]
fn test_basic() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::basic())),
"+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::basic())),
"+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::basic())),
"+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::basic())),
"+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+\n\
| 0 | 1 | 2 | 3 |\n\
+---+---+---+---+"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::basic())),
""
);
}
#[test]
fn test_reinforced() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::reinforced())),
"┏───┬───┬───┬───┓\n\
│ 0 │ 1 │ 2 │ 3 │\n\
│ 0 │ 1 │ 2 │ 3 │\n\
│ 0 │ 1 │ 2 │ 3 │\n\
┗───┴───┴───┴───┛"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::reinforced())),
"┏───┬───┬───┬───┓\n\
│ 0 │ 1 │ 2 │ 3 │\n\
│ 0 │ 1 │ 2 │ 3 │\n\
┗───┴───┴───┴───┛"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::reinforced())),
"┏───┬───┬───┬───┓\n\
│ 0 │ 1 │ 2 │ 3 │\n\
┗───┴───┴───┴───┛"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::reinforced())),
"┏───┬───┬───┬───┓\n\
│ 0 │ 1 │ 2 │ 3 │\n\
│ 0 │ 1 │ 2 │ 3 │\n\
┗───┴───┴───┴───┛"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::reinforced())),
""
);
}
#[test]
fn test_compact() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::compact())),
concat!(
"───┬───┬───┬───\n",
" 0 │ 1 │ 2 │ 3 \n",
"───┼───┼───┼───\n",
" 0 │ 1 │ 2 │ 3 \n",
" 0 │ 1 │ 2 │ 3 \n",
"───┴───┴───┴───",
)
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::compact())),
concat!(
"───┬───┬───┬───\n",
" 0 │ 1 │ 2 │ 3 \n",
"───┼───┼───┼───\n",
" 0 │ 1 │ 2 │ 3 \n",
"───┴───┴───┴───",
)
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::compact())),
concat!("───┬───┬───┬───\n", " 0 │ 1 │ 2 │ 3 \n", "───┴───┴───┴───",)
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::compact())),
concat!(
"───┬───┬───┬───\n",
" 0 │ 1 │ 2 │ 3 \n",
" 0 │ 1 │ 2 │ 3 \n",
"───┴───┴───┴───",
)
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::compact())),
""
);
}
#[test]
fn test_compact_double() {
assert_eq!(
draw_table(&Table::new(
row(4),
vec![row(4); 2],
theme::compact_double()
)),
concat!(
"═══╦═══╦═══╦═══\n",
" 0 ║ 1 ║ 2 ║ 3 \n",
"═══╬═══╬═══╬═══\n",
" 0 ║ 1 ║ 2 ║ 3 \n",
" 0 ║ 1 ║ 2 ║ 3 \n",
"═══╩═══╩═══╩═══",
)
);
assert_eq!(
draw_table(&Table::new(
row(4),
vec![row(4); 1],
theme::compact_double()
)),
concat!(
"═══╦═══╦═══╦═══\n",
" 0 ║ 1 ║ 2 ║ 3 \n",
"═══╬═══╬═══╬═══\n",
" 0 ║ 1 ║ 2 ║ 3 \n",
"═══╩═══╩═══╩═══",
)
);
assert_eq!(
draw_table(&Table::new(
row(4),
vec![row(4); 0],
theme::compact_double()
)),
concat!("═══╦═══╦═══╦═══\n", " 0 ║ 1 ║ 2 ║ 3 \n", "═══╩═══╩═══╩═══",)
);
assert_eq!(
draw_table(&Table::new(
row(0),
vec![row(4); 2],
theme::compact_double()
)),
concat!(
"═══╦═══╦═══╦═══\n",
" 0 ║ 1 ║ 2 ║ 3 \n",
" 0 ║ 1 ║ 2 ║ 3 \n",
"═══╩═══╩═══╩═══",
)
);
assert_eq!(
draw_table(&Table::new(
row(0),
vec![row(0); 0],
theme::compact_double()
)),
""
);
}
#[test]
fn test_heavy() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::heavy())),
"┏━━━┳━━━┳━━━┳━━━┓\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┣━━━╋━━━╋━━━╋━━━┫\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┗━━━┻━━━┻━━━┻━━━┛"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::heavy())),
"┏━━━┳━━━┳━━━┳━━━┓\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┣━━━╋━━━╋━━━╋━━━┫\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┗━━━┻━━━┻━━━┻━━━┛"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::heavy())),
"┏━━━┳━━━┳━━━┳━━━┓\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┗━━━┻━━━┻━━━┻━━━┛"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::heavy())),
"┏━━━┳━━━┳━━━┳━━━┓\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┃ 0 ┃ 1 ┃ 2 ┃ 3 ┃\n\
┗━━━┻━━━┻━━━┻━━━┛"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::heavy())),
""
);
}
#[test]
fn test_light() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::light())),
concat!(
" 0 1 2 3 \n",
"───────────────\n",
" 0 1 2 3 \n",
" 0 1 2 3 ",
)
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::light())),
concat!(" 0 1 2 3 \n", "───────────────\n", " 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::light())),
concat!(" 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::light())),
concat!(" 0 1 2 3 \n", " 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::light())),
""
);
}
#[test]
fn test_none() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::none())),
concat!(" 0 1 2 3 \n", " 0 1 2 3 \n", " 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::none())),
concat!(" 0 1 2 3 \n", " 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::none())),
concat!(" 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::none())),
concat!(" 0 1 2 3 \n", " 0 1 2 3 ")
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::none())),
""
);
}
#[test]
fn test_thin() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::thin())),
"┌───┬───┬───┬───┐\n\
│ 0 │ 1 │ 2 │ 3 │\n\
├───┼───┼───┼───┤\n\
│ 0 │ 1 │ 2 │ 3 │\n\
├───┼───┼───┼───┤\n\
│ 0 │ 1 │ 2 │ 3 │\n\
└───┴───┴───┴───┘"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::thin())),
"┌───┬───┬───┬───┐\n\
│ 0 │ 1 │ 2 │ 3 │\n\
├───┼───┼───┼───┤\n\
│ 0 │ 1 │ 2 │ 3 │\n\
└───┴───┴───┴───┘"
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::thin())),
"┌───┬───┬───┬───┐\n\
│ 0 │ 1 │ 2 │ 3 │\n\
└───┴───┴───┴───┘"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::thin())),
"┌───┬───┬───┬───┐\n\
│ 0 │ 1 │ 2 │ 3 │\n\
├───┼───┼───┼───┤\n\
│ 0 │ 1 │ 2 │ 3 │\n\
└───┴───┴───┴───┘"
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::thin())),
""
);
}
#[test]
fn test_with_love() {
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 2], theme::with_love())),
concat!(
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤\n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤\n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤",
)
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 1], theme::with_love())),
concat!(
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤\n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤\n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤",
)
);
assert_eq!(
draw_table(&Table::new(row(4), vec![row(4); 0], theme::with_love())),
concat!("❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤\n", " 0 ❤ 1 ❤ 2 ❤ 3 \n", "❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤",)
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(4); 2], theme::with_love())),
concat!(
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤\n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
" 0 ❤ 1 ❤ 2 ❤ 3 \n",
"❤❤❤❤❤❤❤❤❤❤❤❤❤❤❤",
)
);
assert_eq!(
draw_table(&Table::new(row(0), vec![row(0); 0], theme::with_love())),
""
);
}
fn draw_table(table: &Table) -> String {
let cfg = Config::default();
let styles = HashMap::default();
let alignments = Alignments::default();
table
.draw_table(&cfg, &styles, alignments, std::usize::MAX)
.expect("Unexpectdly got no table")
}
fn row(count_columns: usize) -> Vec<StyledString> {
let mut row = Vec::with_capacity(count_columns);
for i in 0..count_columns {
row.push(StyledString::new(i.to_string(), TextStyle::default()));
}
row
}