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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
import { GlAvatarLabeled, GlLink, GlTableLite } from '@gitlab/ui';
import { shallowMountExtended } from 'helpers/vue_test_utils_helper';
import CandidateDetail from '~/ml/model_registry/components/candidate_detail.vue';
import DetailRow from '~/ml/model_registry/components/candidate_detail_row.vue';
import {
NO_PARAMETERS_MESSAGE,
NO_METRICS_MESSAGE,
NO_METADATA_MESSAGE,
NO_CI_MESSAGE,
} from '~/ml/model_registry/translations';
import { stubComponent } from 'helpers/stub_component';
import { newCandidate } from '../mock_data';
describe('ml/model_registry/components/candidate_detail.vue', () => {
let wrapper;
const CANDIDATE = newCandidate();
const USER_ROW = 1;
const INFO_SECTION = 0;
const CI_SECTION = 1;
const PARAMETER_SECTION = 2;
const METADATA_SECTION = 3;
const createWrapper = (createCandidate = () => CANDIDATE, showInfoSection = true) => {
wrapper = shallowMountExtended(CandidateDetail, {
propsData: { candidate: createCandidate(), showInfoSection },
stubs: {
GlTableLite: { ...stubComponent(GlTableLite), props: ['items', 'fields'] },
},
});
};
const findSection = (section) => wrapper.findAll('section').at(section);
const findRowInSection = (section, row) =>
findSection(section).findAllComponents(DetailRow).at(row);
const findLinkAtRow = (section, rowIndex) =>
findRowInSection(section, rowIndex).findComponent(GlLink);
const findNoDataMessage = (label) => wrapper.findByText(label);
const findLabel = (label) => wrapper.find(`[label='${label}']`);
const findCiUserDetailRow = () => findRowInSection(CI_SECTION, USER_ROW);
const findCiUserAvatar = () => findCiUserDetailRow().findComponent(GlAvatarLabeled);
const findCiUserAvatarNameLink = () => findCiUserAvatar().findComponent(GlLink);
const findMetricsTable = () => wrapper.findComponent(GlTableLite);
describe('All info available', () => {
beforeEach(() => createWrapper());
const mrText = `!${CANDIDATE.info.ciJob.mergeRequest.iid} ${CANDIDATE.info.ciJob.mergeRequest.title}`;
const expectedTable = [
[INFO_SECTION, 0, 'ID', CANDIDATE.info.iid],
[INFO_SECTION, 1, 'MLflow run ID', CANDIDATE.info.eid],
[INFO_SECTION, 2, 'Status', CANDIDATE.info.status],
[INFO_SECTION, 3, 'Experiment', CANDIDATE.info.experimentName],
[INFO_SECTION, 4, 'Artifacts', 'Artifacts'],
[CI_SECTION, 0, 'Job', CANDIDATE.info.ciJob.name],
[CI_SECTION, 1, 'Triggered by', 'CI User'],
[CI_SECTION, 2, 'Merge request', mrText],
[PARAMETER_SECTION, 0, CANDIDATE.params[0].name, CANDIDATE.params[0].value],
[PARAMETER_SECTION, 1, CANDIDATE.params[1].name, CANDIDATE.params[1].value],
[METADATA_SECTION, 0, CANDIDATE.metadata[0].name, CANDIDATE.metadata[0].value],
[METADATA_SECTION, 1, CANDIDATE.metadata[1].name, CANDIDATE.metadata[1].value],
];
it.each(expectedTable)('row %s is created correctly', (section, rowIndex, label, text) => {
const row = findRowInSection(section, rowIndex);
expect(row.props()).toMatchObject({ label });
expect(row.text()).toBe(text);
});
describe('Table links', () => {
const linkRows = [
[INFO_SECTION, 3, CANDIDATE.info.pathToExperiment],
[INFO_SECTION, 4, CANDIDATE.info.pathToArtifact],
[CI_SECTION, 0, CANDIDATE.info.ciJob.path],
[CI_SECTION, 2, CANDIDATE.info.ciJob.mergeRequest.path],
];
it.each(linkRows)('row %s is created correctly', (section, rowIndex, href) => {
expect(findLinkAtRow(section, rowIndex).attributes().href).toBe(href);
});
});
describe('Metrics table', () => {
it('computes metrics table items correctly', () => {
expect(findMetricsTable().props('items')).toEqual([
{ name: 'AUC', 0: '.55' },
{ name: 'Accuracy', 1: '.99', 2: '.98', 3: '.97' },
{ name: 'F1', 3: '.1' },
]);
});
it('computes metrics table fields correctly', () => {
expect(findMetricsTable().props('fields')).toEqual([
expect.objectContaining({ key: 'name', label: 'Metric' }),
expect.objectContaining({ key: '0', label: 'Step 0' }),
expect.objectContaining({ key: '1', label: 'Step 1' }),
expect.objectContaining({ key: '2', label: 'Step 2' }),
expect.objectContaining({ key: '3', label: 'Step 3' }),
]);
});
});
describe('CI triggerer', () => {
it('renders user row', () => {
const avatar = findCiUserAvatar();
expect(avatar.props()).toMatchObject({
label: '',
});
expect(avatar.attributes().src).toEqual('/img.png');
});
it('renders user name', () => {
const nameLink = findCiUserAvatarNameLink();
expect(nameLink.attributes().href).toEqual('path/to/ci/user');
expect(nameLink.text()).toEqual('CI User');
});
});
});
describe('No artifact path', () => {
beforeEach(() =>
createWrapper(() => {
const candidate = newCandidate();
delete candidate.info.pathToArtifact;
return candidate;
}),
);
it('does not render artifact row', () => {
expect(findLabel('Artifacts').exists()).toBe(false);
});
});
describe('No params, metrics, ci or metadata available', () => {
beforeEach(() =>
createWrapper(() => {
const candidate = newCandidate();
delete candidate.params;
delete candidate.metrics;
delete candidate.metadata;
delete candidate.info.ciJob;
return candidate;
}),
);
it('does not render params', () => {
expect(findNoDataMessage(NO_PARAMETERS_MESSAGE).exists()).toBe(true);
});
it('does not render metadata', () => {
expect(findNoDataMessage(NO_METADATA_MESSAGE).exists()).toBe(true);
});
it('does not render metrics', () => {
expect(findNoDataMessage(NO_METRICS_MESSAGE).exists()).toBe(true);
});
it('does not render CI info', () => {
expect(findNoDataMessage(NO_CI_MESSAGE).exists()).toBe(true);
});
});
describe('Has CI, but no user or mr', () => {
beforeEach(() =>
createWrapper(() => {
const candidate = newCandidate();
delete candidate.info.ciJob.user;
delete candidate.info.ciJob.mergeRequest;
return candidate;
}),
);
it('does not render MR info', () => {
expect(findLabel('Merge request').exists()).toBe(false);
});
it('does not render CI user info', () => {
expect(findLabel('Triggered by').exists()).toBe(false);
});
});
describe('showInfoSection is set to false', () => {
beforeEach(() => createWrapper(() => CANDIDATE, false));
it('does not render the info section', () => {
expect(findLabel('MLflow run ID').exists()).toBe(false);
});
});
});
|
