def test_update_experiment_and_get(self):
exp = Experiment('expid8', 'test_experiment', 'unittest', 10, None, 1,
'running')
self.db.save_experiment(exp)
self.assertEqual(self.db.get_experiment('expid8').status, 'running')
self.assertEqual(self.db.get_experiment('expid8').end, None)
exp.status = 'finished'
exp.end = 100
self.db.update_experiment(exp)
self.assertEqual(self.db.get_experiment('expid8').status, 'finished')
self.assertEqual(self.db.get_experiment('expid8').end, 100)
def test_get_experiments(self):
self.maxDiff = None
experiments = [
Experiment('id1', 'name1', 'creator1', 123, 456, 789, 'FINISH'),
Experiment('id2', 'name2', 'creator2', 1, 4, 7, 'QUEUED'),
Experiment('id3', 'name3', 'creator3', 3, 6, 9, 'PROGRESS')
]
self.db_resource.db.save_experiment(experiments[0])
self.db_resource.db.save_experiment(experiments[1])
self.db_resource.db.save_experiment(experiments[2])
result = self.simulate_get('/api/experiments')
dicts = []
for exp in experiments:
dicts.append(exp.__dict__)
self.assertCountEqual(result.json, dicts)
def test_get_running_experiment_on_empty_set(self):
entity1 = Experiment('exp1',
'exp1-name',
'exp1-creator',
1.1,
2.2,
status='FINISHED')
entity2 = Experiment('exp2',
'exp2-name',
'exp2-creator',
3.3,
4.4,
status='FINISHED')
self.db.save_experiment(entity1)
self.db.save_experiment(entity2)
running = self.db.get_running_experiment()
self.assertIsNone(running)
def test_get_running_experiment(self):
entity1 = Experiment('exp1',
'exp1-name',
'exp1-creator',
1.1,
2.2,
status='FINISHED')
entity2 = Experiment('exp2',
'exp2-name',
'exp2-creator',
3.3,
4.4,
status='RUNNING')
self.db.save_experiment(entity1)
self.db.save_experiment(entity2)
running = self.db.get_running_experiment()
self.assertEqual('exp2', running.id)
def test_save_experiment_and_get(self):
entity = Experiment('exp1',
'exp1-name',
'exp1-creator',
1.1,
2.2,
status='FINISHED')
self.db.save_experiment(entity)
actual = self.db.get_experiment('exp1')
self.assertEqual(entity.__dict__, actual.__dict__)
def test_delete_experiment(self):
exp_id = 'expid10'
exp_id_control = 'expid11'
self.db.save_experiment(
Experiment(exp_id, 'test_experiment', 'unittest', 10, 100, 1,
'finished'))
self.db.save_telemetry([Telemetry(1, 'cpu', 'n1', 32, exp_id)])
self.db.save_experiment(
Experiment(exp_id_control, 'test_experiment', 'unittest', 10, 100,
1, 'finished'))
self.db.save_telemetry([Telemetry(1, 'cpu', 'n1', 32, exp_id_control)])
self.assertIsNotNone(self.db.get_experiment(exp_id))
self.db.delete_experiment(exp_id)
self.assertIsNone(self.db.get_experiment(exp_id))
self.assertIsNotNone(self.db.get_experiment(exp_id_control))
def test_save_experiment_and_find_all(self):
entity1 = Experiment('exp1',
'exp1-name',
'exp1-creator',
1.1,
2.2,
status='FINISHED')
entity2 = Experiment('exp2',
'exp2-name',
'exp2-creator',
3.3,
4.4,
status='RUNNING')
self.db.save_experiment(entity1)
self.db.save_experiment(entity2)
experiments = self.db.find_all()
self.assertEqual(2, len(experiments))
self.assertEqual(entity1.__dict__, experiments[0].__dict__)
self.assertEqual(entity2.__dict__, experiments[1].__dict__)
def test_save_and_get_experiment(self):
expected = Experiment('expid1', 'test_experiment', 'unittest', 10, 100,
1, 'running')
self.db.save_experiment(expected)
actual = self.db.get_experiment('expid1')
self.assertEqual('expid1', actual.id)
self.assertEqual('test_experiment', actual.name)
self.assertEqual('unittest', actual.creator)
self.assertEqual(10., actual.start)
self.assertEqual(100., actual.end)
self.assertEqual(1., actual.created)
self.assertEqual('running', actual.status)
def finalize_experiment(self, exp: Experiment, status):
exp.status = status
exp.end = time.time()
self.repository.update_experiment(exp)
self.repository.touch_traces(exp)
def test_save_and_touch_and_get_traces(self):
traces = [
RequestTrace('req1',
'c1',
's1',
1.1,
1.2,
1.3,
server='h1',
status=200),
RequestTrace('req2',
'c2',
's2',
2.1,
2.2,
2.3,
server='h2',
status=200),
RequestTrace('req3',
'c3',
's1',
3.1,
3.2,
3.3,
server='h1',
status=200,
response='time=123'),
RequestTrace('req4',
'c1',
's1',
4.1,
4.2,
4.3,
server='h1',
status=200),
]
self.db.save_experiment(
Experiment('exp1', start=1, end=3.5, status='FINISHED'))
self.db.save_traces(traces)
self.db.touch_traces(
self.db.get_experiment('exp1')) # should touch the first 3
actual = self.db.get_traces('exp1')
expected = [
RequestTrace('req1',
'c1',
's1',
1.1,
1.2,
1.3,
server='h1',
status=200,
exp_id='exp1'),
RequestTrace('req2',
'c2',
's2',
2.1,
2.2,
2.3,
server='h2',
status=200,
exp_id='exp1'),
RequestTrace('req3',
'c3',
's1',
3.1,
3.2,
3.3,
server='h1',
status=200,
response='time=123',
exp_id='exp1')
]
self.assertEqual(expected, actual)